如何解决如何找到两个不同大小的数组的相同行?
我有两个大小不同的数组
a = np.array([[5,0],[2,4],[0,1],[3,[1,5],[5,6],[7,9]])
b = np.array([[0,3],4]])
我需要
c = np.array([False,True,False,False])
即数组“ b”在数组“ a”中具有行[5,6]和[2,4]。目前,我是通过
logical = np.zeros(a.shape[0]).astype(bool)
for i in range(b.shape[0]):
logical += np.all(a == b[i],axis=1)
是否有用于执行此操作的numpy代码?
解决方法
让我们尝试广播:
(a[None,:] == b[:,None]).all(-1).any(0)
输出:
array([False,True,False,False])
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。