làm thế nào để san bằng một danh sách 2D thành 1D mà không cần sử dụng gọn gàng?

Question

tôi có một danh sách như sau:

[[1,2,3],[1,2],[1,4,5,6,7]] 

và tôi muốn san bằng nó vào [1,2,3,1,2,1,4,5,6,7]

là có một hàm trọng lượng nhẹ để làm điều này mà không sử dụng NumPy?

Answer 1

Without NumPy (ndarray.flatten), bạn có thể sử dụng chain.from_iterable đó là một nhà xây dựng thay thế cho itertools.chain:

>>> list(chain.from_iterable([[1,2,3],[1,2],[1,4,5,6,7]])) 
[1, 2, 3, 1, 2, 1, 4, 5, 6, 7] 

bạn cũng có thể sử dụng reduce trong python 2 và functools.reduce trong 3 đó là hiệu quả hơn cho các danh sách ngắn:

>>> reduce(lambda x,y :x+y ,[[1,2,3],[1,2],[1,4,5,6,7]]) 
[1, 2, 3, 1, 2, 1, 4, 5, 6, 7] 

Hoặc sử dụng một danh sách hiểu:

[j for i in [[1,2,3],[1,2],[1,4,5,6,7]] for j in i] 

benchmark:

:~$ python -m timeit "from itertools import chain;chain.from_iterable([[1,2,3],[1,2],[1,4,5,6,7]])" 
1000000 loops, best of 3: 1.58 usec per loop 
:~$ python -m timeit "reduce(lambda x,y :x+y ,[[1,2,3],[1,2],[1,4,5,6,7]])" 
1000000 loops, best of 3: 0.791 usec per loop 
:~$ python -m timeit "[j for i in [[1,2,3],[1,2],[1,4,5,6,7]] for j in i]" 
1000000 loops, best of 3: 0.784 usec per loop 

Một benchmark trên @ câu trả lời của Will rằng sử dụng sum (nhanh của nó đối với danh sách ngắn nhưng không phải cho danh sách dài):

:~$ python -m timeit "sum([[1,2,3],[4,5,6],[7,8,9]], [])" 
1000000 loops, best of 3: 0.575 usec per loop 
:~$ python -m timeit "sum([range(100),range(100)], [])" 
100000 loops, best of 3: 2.27 usec per loop 
:~$ python -m timeit "reduce(lambda x,y :x+y ,[range(100),range(100)])" 
100000 loops, best of 3: 2.1 usec per loop 

Answer 2

này sẽ làm việc trong trường hợp cụ thể của bạn. Một hàm đệ quy sẽ làm việc tốt nhất nếu bạn có nhiều cấp độ của vòng lặp lồng nhau.

def flatten(input): 
    new_list = [] 
    for i in input: 
     for j in i: 
      new_list.append(j) 
    return new_list 

Answer 3

Chỉ với một danh sách như thế này, mẹo nhỏ gọn yêu thích của tôi chỉ sử dụng sum;

sum có một đối số tùy chọn: sum(iterable [, start]), vì vậy bạn có thể làm:

list_of_lists = [[1,2,3], [4,5,6], [7,8,9]] 
print sum(list_of_lists, []) # [1,2,3,4,5,6,7,8,9] 

làm việc này vì các nhà điều hành + sẽ xảy ra là các nhà điều hành nối cho các danh sách, và bạn đã nói với nó rằng giá trị khởi đầu là [] - danh sách trống.

nhưng documentaion cho sum khuyên bạn nên sử dụng itertools.chain thay vào đó, vì nó rõ ràng hơn nhiều.