有一个NumPy函数可以根据条件从两个数组中删除元素吗？

huangapple
go评论156阅读模式
英文:

is there a numpy function to delete elements in two arrays given a condition?

问题

我想删除两个数组中等于0.0的元素，以及在一个数组中删除0.0元素后，在另一个数组中相同位置的非零元素。我期望的结果如下：

  1. a = np.array([5.0, 10.0])
  2. b = np.array([1.0, 1.0])
英文:

I have two arrays:

  1. a = np.array([0.0, 3.0, 5.0, 0.0, 10.0])
  2. b = np.array([0.0, 0.0, 1.0, 4.0, 1.0 ])

I want to delete elements equal to 0.0 from both lists and the non-zero elements which are in the same position of a deleted 0.0 element in the other list. Exactly, what I expect is:

  1. a = np.array([5.0, 10.0])
  2. b = np.array([1.0, 1.0 ])

thanks!

答案1

得分: 2

  1. 尝试：
  3. ```py
  4. a = np.array([0.0, 3.0, 5.0, 0.0, 10.0])
  5. b = np.array([0.0, 0.0, 1.0, 4.0, 1.0])
  7. mask = ~((a == 0) | (b == 0))
  9. a = a[mask]
  10. b = b[mask]
  12. print(a)
  13. print(b)

输出：

  1. [ 5. 10.]
  2. [1. 1.]
  2. <details>
  3. <summary>英文:</summary>
  5. Try:
  7. ```py
  8. a = np.array([0.0, 3.0, 5.0, 0.0, 10.0])
  9. b = np.array([0.0, 0.0, 1.0, 4.0, 1.0 ])
  11. mask = ~((a == 0) | (b == 0))
  13. a = a[mask]
  14. b = b[mask]
  16. print(a)
  17. print(b)

Prints:

  1. [ 5. 10.]
  2. [1. 1.]

huangapple
  • 本文由 发表于 2023年4月7日 04:00:45
  • 转载请务必保留本文链接：https://go.coder-hub.com/75953326.html
  • arrays
  • numpy
  • python-3.x
匿名

发表评论

匿名网友

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen:

确定