如何在numpy中比较数组的每个元素与另一个整个数组的循环

Question

设数组A和数组B为:

import numpy as np
A = np.array([1, 5, 2, 6])
B = np.array([4, 2, 1, 1])

如何比较数组A和数组B的每个元素，使结果如下所示

results = np.array([
A[0] > B,
A[1] > B,
A[2] > B,
A[3] > B
])
# resulting in a 2d array like so:
>>> results
[[False False False False] [True True True True] [False False True True] [True True True True]]

Answer 1

最简单的方法是将A转换为列向量，并将其与B进行比较，从而触发两者的自动广播:

>>> A[:, None]
array([[1],
[5],
[2],
[6]])
>>> A[:, None] > B
array([[False, False, False, False],
[ True,  True,  True,  True],
[False, False,  True,  True],
[ True,  True,  True,  True]])