如何根据数组中的元素对python多维数组中的值求和？

Question

说我有一个array名为我的阵列

my_array= [[1,2],[1,3],[2,3]]

我想添加从1开始的每个子列表中的所有第二个元素,以便我的输出为5

使用NumPy也不是一个选择......

有谁知道如何做到这一点 ？

Answer 1

您可以使用条件列表推导.

my_array=[[1,2],[1,3],[2,3]]
my_sum=sum(v[1] for v in my_array if v[0]==1)

print(my_sum)

输出:

5