我想知道是否有一个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)
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
点击生成二维码
我来说两句