我有两个numpy数组

  • 时间:
  • 分类: Python

长度不相等。如何找到数组1和数组2中的元素？谢谢

# 回答1

听起来像是一份布景工作。在此处查看文档

选择 | 换行 | 行号
  1. In [21]: a
  2. Out[21]: array([1, 2, 3, 4])
  3.  
  4. In [22]: b
  5. Out[22]: array([2, 4, 4, 5, 5])
  6.  
  7. In [23]: a2=set(a)
  8.  
  9. In [24]: b2=set(b)
  10.  
  11. In [25]: a2
  12. Out[25]: set([1, 2, 3, 4])
  13.  
  14. In [26]: b2
  15. Out[26]: set([2, 4, 5])
  16.  
  17. In [27]: result = a2 & b2 #or a2.intersection(b2)
  18.  
  19. In [28]: result
  20. Out[28]: set([2, 4])
  21.  

或者，您可以循环访问其中一个数组，并检查它是否在另一个数组中。

选择 | 换行 | 行号
  1. In [29]: for i in a:
  2.    ....:     if i in b:
  3.    ....:         print i,
  4.    ....:         
  5.    ....:         
  6. 2 4
  7.  
  8.  

标签: python

添加新评论