给定两个长度相同的NumPy数组x和y，如何选择所有对(x，y)，使x在x_min和x_max之间有界？

Question

我有两个数组，x和y，长度相同。我想选择所有的(喜，易)这样的x_min

Answer 1

idx = (x_min < x) & (x < x_max)

x[idx]
y[idx]

并在必要时堆叠

np.vstack((x[idx], y[idx])).T

Answer 2

您可以使用内置的zip()在相同长度的列表中迭代相应的对。从这里，只需找到您的x列表的最小和最大，并做一个列表理解：

>>> x = random.choices(range(100), k=10)  # [19, 16, 43, 89, 64, 7, 42, 9, 19, 78]
>>> y = random.choices(range(100), k=10)  # [80, 69, 37, 50, 15, 0, 56, 63, 18, 93]
>>> x_min = min(x)  # 7
>>> x_max = max(x)  # 89
>>> couples = [tup for tup in zip(x, y) if x_min < tup[0] < x_max]
>>> print(couples)
[(19, 80), (16, 69), (43, 37), (64, 15), (42, 56), (9, 63), (19, 18), (78, 93)]