提问者:小点点

沿着列查找数组1(多于一列)中大于数组2(单列)中的值的值


假设我有一个形状为(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的方法?


共1个答案

匿名用户

array2重塑为形状为(2,1)的2d数组,然后由于numpy广播,比较应该可以工作:

(array1 > array2[:,None]).astype(int)
#array([[1, 1, 1, 0],
#       [1, 0, 1, 1]])