python – 查找NumPy数组与值列表中的任何值相等的位置

内容导读

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

内容图文

我有一个整数数组,想要找到该数组在多个值列表中的任何值的位置.

这可以通过单独处理每个值,或者在循环中使用多个“或”语句来轻松完成,但我觉得必须有更好/更快的方法来完成它.我实际上正在处理大小为4000 x 2000的数组,但这里是问题的简化版：

fake = arange(9).reshape((3,3))

array([[0, 1, 2],
       [3, 4, 5],
       [6, 7, 8]])

want = (fake==0) + (fake==2) + (fake==6) + (fake==8)

print want 

array([[ True, False,  True],
       [False, False, False],
       [ True, False,  True]], dtype=bool)

我想要的是从涉及假的单个命令和值列表[0,2,6,8]获得所需的方法.

我假设有一个已经包含这个包的包将比我在Python中用循环编写一个函数要快得多.

解决方法:

功能numpy.in1d似乎做你想要的.唯一的问题是它只适用于1d数组,所以你应该像这样使用它：

In [9]: np.in1d(fake, [0,2,6,8]).reshape(fake.shape)
Out[9]: 
array([[ True, False,  True],
       [False, False, False],
       [ True, False,  True]], dtype=bool)

我不知道为什么这仅限于1d数组.看看它的source code,它首先似乎压平了两个阵列,之后它做了一些巧妙的排序技巧.但是没有什么可以阻止它再次在结束时取消结果,就像我必须在这里手工做的那样.

内容总结

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

内容备注

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

内容手机端