Numpy 元素更大（对于另一个数组中的每个元素）

Question

I struggle to write find the best "question" so please feel free to suggest another title.

Lets say I have a=np.array([5,3,2,4]) and b=np.array([1,2]) - I want to get a list of list (or np.arrays) with the value of a>b[i] ie it can be written as a list comprehension

[a[i]>p for p in b] which returns

[np.array([True,True,True,True]), np.array([True,True,False,True])] . Since I have a rather large data set I hoped that there was a numpy function to do such, or is list comprehension the better way here?

Answer 1

You can use numpy broadcasting for this, you just need to add an extra dimension into each of your arrays:

>>> a[None,:] > b[:,None]
array([[ True,  True,  True,  True],
       [ True,  True, False,  True]])