如何解决ndarray 通过索引从另一个数组中的条件删除行
我有两个ndarrays
,其中X
的第一个维度的长度与y
的大小相同:
X = np.asarray([[1,2,3],[4,5,6],[7,8,9],[3,6,1]])
y = np.asarray([1,3])
l = [0,7]
如果来自 X
的同一索引行的值在 y
中,我想删除 l
中的每一行。
因此,在这种情况下,我将:
X = np.asarray([[1,1]])
那是因为来自 y - 的第二个和第三个元素在 l 中。因此,应从 X 中删除第 2 行和第 3 行。 怎么办?
解决方法
一个简单的单行解决方案是使用 delete
和 argwhere
X = np.delete(X,np.argwhere(np.isin(y,l)).flatten(),axis=0)
输出
array([[1,2,3],[3,6,1]])
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。