python - Numpy:过滤并行数组

Question

我有两个并行的 numpy 数组，我只想保留那些 a-value 大于 3 的元素:

import numpy as np

a = np.array([1, 2, 3, 4, 5])
b = np.array([0.1, 0.15, -0.3, 23.4, 112.5])

c = np.array(list(zip(a, b)))

c = c[c[:, 0] > 3].T

a = c[0]
b = c[1]

如何以 numpy 方式进行操作？

Answer 1

您可以只使用 bool 索引:

a[a > 3]
# array([ 4.,  5.])

b[a > 3]
# array([  23.4,  112.5])

<小时/>

a, b = a[a > 3], b[a > 3]