Python如何按行比较2个数组中的元素

Question

我是Python初学者，这是一个例子:

a = array ([[4,7,8],
            [8,8,8],
            [5,8,4]]) 
b = array([[1,1,7],
           [2,6,9],
           [2,3,4]])
the output would be [1,1,0]

我想比较b第一行中有多少元素比a第一行第一个元素大。并对每一行应用相同的比较。比较必须是2个数组中的同一行。原始数据形状为(297,6940)。有什么简单的方法可以做到这一点吗？谢谢!!!

Answer 1

使用直接数组比较并将它们沿行求和如下:

(a<b).sum(axis=1)

array([0, 1, 0])

a<b
array([[False, False, False],
       [False, False,  True],
       [False, False, False]])