假设我有一个形状为(2,4)的numpy ndarray,如下所示
>>> array1 = numpy.random.rand(2,4)
array([[ 0.87791012, 0.84566058, 0.73877908, 0.40377929],
[ 0.9669688 , 0.15913901, 0.70374509, 0.95776427]])
我有第二个形状(2,)的数组,如下所示
>>> array2 = numpy.random.rand(2)
array([ 0.57126204, 0.67938752])
我想沿着列维度比较这两个数组,以找到array1中大于array2的元素(按元素)。所需的结果是
array([[ 1., 1., 1., 0.],
[ 1., 0., 1., 1.]])
如果两者的维度相同,我可以直接使用(array1
results = np.zeros_like(array1)
for each in range(array1.shape[1]):
results[:,each] = array1[:,each] > array2
有没有更Pythonic/numpy的方法?
将array2
重塑为形状为(2,1)
的2d数组,然后由于numpy广播,比较应该可以工作:
(array1 > array2[:,None]).astype(int)
#array([[1, 1, 1, 0],
# [1, 0, 1, 1]])