Làm cách nào để tổng hợp một mảng đa chiều trong python gọn gàng nhất?

Question

Gần nhất là summing columns.

Vì vậy, tôi sẽ làm điều gì đó tương tự như trong câu hỏi của tôi:

Nói rằng tôi đã là một danh sách Python 2D như sau:

my_list = [ [1,2,3,4], 
      [2,4,5,6] ] 

tôi có thể nhận được tổng số hàng với một sự hiểu biết danh sách:

row_totals = [ sum(x) for x in my_list ] 

Trong một dòng, làm cách nào tôi có thể tổng hợp toàn bộ mảng 2d?

27 

Answer 1

Bạn có thể làm dễ dàng như

sum(map(sum, my_list)) 

hoặc cách khác

sum(sum(x) for x in my_list)) 

và gọi nó là một ngày, nếu bạn không mong đợi hơn 2 chiều. Lưu ý rằng giải pháp đầu tiên có nhiều khả năng không phải là giải pháp nhanh nhất (như trong thời gian thực hiện), do việc sử dụng map(). Điểm chuẩn và so sánh khi cần thiết.

Cuối cùng, nếu bạn thấy mình sử dụng mảng đa chiều, hãy xem xét sử dụng NumPy và các hàm thân thiện với mảng cao cấp của nó. Dưới đây là một đoạn trích ngắn cho vấn đề của bạn:

import numpy as np 

my_list = np.array([[1,2,3,4], [2,4,5,6]]) 
np.sum(my_list) 

Điều này sẽ phù hợp với bất kỳ số thứ nguyên nào mà mảng của bạn có thể có.

Answer 2

>>> sum ([ sum(x) for x in [[1,2,3,4], [2,4,5,6]] ]) 
27 

Answer 3

Một giải pháp sử dụng itertools:

>>> from itertools import chain 
>>> my_list = [ [1,2,3,4], [2,4,5,6] ] 
>>> sum(chain(*my_list)) 
27 

Answer 4

>>> from itertools import chain 
>>> my_list = [[1,2,3,4], [2,4,5,6]] 
>>> sum(chain.from_iterable(my_list)) 
27 

Answer 5

Bạn có thể sử dụng tiền để đầu thêm danh sách bên nhau và sau đó tổng hợp các kết quả danh sách dẹt:

>>> my_list = [ [1,2,3,4], [2,4,5,6] ] 

>>> sum(my_list, []) 
[1, 2, 3, 4, 2, 4, 5, 6] 

>>> sum(sum(my_list, [])) 
27