python - 获得二维数组的列计数

Question

我有一个二维数组数据。并且希望每次第 j 次迭代为 1 时都获得计数。其中 i = 行，j = 列。如果没有 for 循环，我该如何去做呢？

概念上是这样的:

for r in range(row):
    if(data[r][j] == 1)
        amount += 1

Answer 1

您可以执行以下操作:

import numpy as np
a = np.array([[0, 1], [1, 1]])
j = 1
np.sum(a[:, j] == 1)

结果将为您提供 2，而 np.sum(a[:, 0] == 1) 将给出 1

如果如评论中所述，您想在多个数组上使用条件，则可以使用np.ological_and(condition1, condition2):

np.sum(np.logical_and(a[:, 0] == 1, b[:, 0] == 2))