lstrip(), rstrip() cho danh sách

Question

Tôi có một loạt các danh sách lớn với các số nguyên. Các danh sách này có thể bắt đầu hoặc kết thúc bằng một vài số không.

Có cách nào dễ dàng để loại bỏ các số không ở bên trái hoặc bên phải từ danh sách không? Tương tự như lstrip() hoặc rstrip() cho chuỗi?

Dữ liệu trông giống như

[0,0,0,1,2,3,4] 

hoặc

[1,2,3,4,0,0,0] 

tôi phải có khả năng riêng lstrip() hoặc rstrip(). Tôi không cần một dải từ cả hai mặt của danh sách.

Answer 1

Bạn có thể sử dụng itertools.dropwhile():

>>> L = [0, 0, 1, 1, 2, 2, 0] 
>>> list(itertools.dropwhile(lambda x: x == 0, L)) 
[1, 1, 2, 2, 0] 

Answer 2

l = ['10000', '000001'] 
map(lambda x: x.strip('0'), l) 

>>> ['1', '1'] 

Answer 3

Tôi đoán danh sách của bạn có chứa chuỗi các số nguyên? Giống như ['001','100'] thay vì [001,100]?

Hãy thử [x.strip('0') for x in bigList]. Xem str.split trong python docs.

Answer 4

Có giải pháp hiệu quả hơn tích hợp sẵn trong itertools.dropwhile(). Bạn có thể sử dụng toàn năng collections.deque, đây sẽ là cấu trúc dữ liệu lý tưởng cho tác vụ này, bởi vì bên trái hoặc bên phải pop là O(1). Đây là trường hợp trái dải, và phải dải sẽ là chỉ là hình ảnh phản chiếu của nó:

from collections import deque 

def noLeadingZero(l): 
    d = deque(l) 
    for e in l: 
     if e == 0: 
      d.popleft() 
     else: 
      break 
    return list(d) 

l = [0, 0, 1, 1, 2, 2, 0] 
print(noLeadingZero(l)) 
# Result: 
# [1, 1, 2, 2, 0] 

Hãy kiểm tra hiệu quả của nó đối với đoạn mã sau đó sử dụng được xây dựng trong itertools.dropwhile():

from itertools import dropwhile 
print(list(dropwhile(lambda x: x == 0, l))) 

đây là bài kiểm tra hiệu suất:

import timeit 

print timeit.timeit(
setup= """from itertools import dropwhile 
l = [0, 0, 1, 1, 2, 2, 0]""", 
stmt="""list(dropwhile(lambda x: x == 0, l))""") #2.308 

print timeit.timeit(
setup= """from collections import deque 
l = [0, 0, 1, 1, 2, 2, 0] 
def noLeadingZero(l): 
    d = deque(l) 
    for e in l: 
     if e == 0: 
      d.popleft() 
     else: 
      break 
    return list(d)""", 
stmt="""noLeadingZero(l)""") #1.684 -> Win!