主页 / 问题 / 1409641
Accepted
bluesky
Asked:2022-07-14 07:21:01 +0800 CST2022-07-14 07:21:01 +0800 CST

如何索引numpy向量的元素

  • 772

如何在一个表达式中快速（例如，元素个数为10**9）选择向量'a'中值大于'a_min'，小于'a_max'的元素，但不显示带有数字“idx”的元素。例如

a = np.arange(20)
a_min = 5
a_max = 15
idx = [6, 7, 9, 12]
res_ind = np.where((a > a_min) & (a < a_max) & (a != a[idx]))

第三个条件 'a != a[idx]' 不正确。

python

1 个回答

  1. Best Answer

    您可以使用掩码数组

    import numpy as np

a = np.arange(20)
a_min = 5
a_max = 15
idx = [6, 7, 9, 12]
masked_a = np.ma.array(a, mask=False)
masked_a.mask[idx] = True
print(np.ma.where(np.logical_and(masked_a > a_min, masked_a < a_max)))

    将打印：

    (array([ 8, 10, 11, 13, 14]),)

    https://www.online-python.com/Q9oyeXGbvD

    • 1

相关问题