想要在1D和2D阵列之间执行元素明智的比较.需要将1D阵列的每个元素与相应的2D行进行比较(例如,更大),并且将创建掩模.这是一个例子:
A = np.random.choice(np.arange(0, 10), (4,100)).astype(np.float)
B = np.array([5., 4., 8., 2. ])
我想要做
A<B
这样A的第一行将与B [0]进行比较,后者为5.结果将是一个布尔数组.
如果我试试这个我得到:
operands could not be broadcast together with shapes (4,100) (4,)
有任何想法吗？