计算一个numpy矩阵有多少列包含所有正值

Question

我想检查一个 numpy 数组/矩阵有多少列只有正值。

我拿了我的矩阵并打印了A>0，得到了True和False，然后我尝试了any和all函数，但没有成功。

In [55]: a = np.array([[13, 21, 12],
                       [21, -1,  6],
                       [ 1, 10,  2],
                       [41,  1,  4]])

输出应该是2。

我将矩阵A 保存在B 中并尝试编写：

B.all(axis=1).any()>0

Answer 1

这个函数统计元素都大于0的列数：

def count(mat):
    counter = 0
    tmp = mat > 0
    for col in tmp.T:
        if all(col):
             counter += 1
    return counter

这个功能是如何工作的？

首先它为 tmp 分配一个布尔值矩阵，指示原始矩阵的对应值是否大于 0，然后遍历该矩阵的转置并检查所有值是否为 True，这意味着它们都更大大于 0。

转置包含原始矩阵的列。乳清创建一个 numpy 数组，您将行传递给函数。通过转置，数组将包含列。