python – 查找包含任何列表的NumPy数组行

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了python – 查找包含任何列表的NumPy数组行，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含452字，纯文字阅读大概需要1分钟。

内容图文

我有一个2D NumPy数组a和一个list / set / 1D NumPy数组b.我想找到那些包含任何b的行,即

import numpy as np

a = np.array([
    [1, 2, 3],
    [4, 5, 3],
    [0, 1, 0]
    ])

b = np.array([1, 2])

# result: [True, False, True]

任何提示？

解决方法:

您可以使用np.in1d查找a中每个元素中b中任何元素的匹配项.现在,np.in1d会使数组变平,所以我们需要重新整形.最后,因为我们想要在a中的每一行找到任何匹配项,所以沿着每一行使用np.any.因此,我们会有这样的实现 –

np.in1d(a,b).reshape(a.shape).any(axis=1)

内容总结

以上是互联网集市为您收集整理的python – 查找包含任何列表的NumPy数组行全部内容，希望文章能够帮你解决python – 查找包含任何列表的NumPy数组行所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端