比较一个numpy数组和另一个数组的每个元素

Question

A = np.array([5,1,5,8])
B = np.array([2,5])

我想将A数组与B的每个元素进行比较。换句话说，我正在寻找一个执行以下计算的函数：

A>2
A>5
(array([ True, False,  True,  True]), array([False, False, False,  True]))

Answer 1

并不是特别花哨，但列表理解会起作用：

[A > b for b in B]

[array([ True, False,  True,  True], dtype=bool),
 array([False, False, False,  True], dtype=bool)]

您还可以使用np.greater() ，它需要Brenlla在注释中使用的添加维技巧：

np.greater(A, B[:,np.newaxis])

array([[ True, False,  True,  True],
       [False, False, False,  True]], dtype=bool)