比较两个向量的numpy运算符
我想知道是否有python numpy操作符来比较相同形状的两个向量。明确地比较两个向量的numpy运算符,numpy,Numpy,我想知道是否有python numpy操作符来比较相同形状的两个向量。明确地 我可以直接通过numpy API获得结果吗?它是什么?非常感谢 如果阵列是a和b: c = ((np.repeat(a, b.shape[0]).reshape(a.shape[0], b.shape[0]) - b) == 0).astype(int) 或者,正如hpaulj和FBruzzesi所说: c = (a[:, None] == b).astype(int) 更简单:a[:,None]==b@hpau
我可以直接通过numpy API获得结果吗?它是什么?非常感谢 如果阵列是a和b:
c = ((np.repeat(a, b.shape[0]).reshape(a.shape[0], b.shape[0]) - b) == 0).astype(int)
或者,正如hpaulj和FBruzzesi所说:
c = (a[:, None] == b).astype(int)
更简单:
a[:,None]==b
@hpaulj那真是天才!作为最后一句话,考虑转换为整数:<代码>(A:[NOT]==B)。astype(int)< /C> >有0和1,而不是布尔值。@ BrunMoLO确实起作用了。这是惊人的简单。谢谢你们的及时回复。浮点还是整数?