numpy：找到范围内元素的索引

Question

例如，我有一个数字数组

a = np.array([1, 3, 5, 6, 9, 10, 14, 15, 56])

我想找到特定范围内元素的所有索引。例如，如果范围是（6，10），则答案应该是（3，4，5）。有内置的功能可以做到这一点吗？

Answer 1

您可以np.where用来获取索引并np.logical_and设置两个条件：

import numpy as np
a = np.array([1, 3, 5, 6, 9, 10, 14, 15, 56])

np.where(np.logical_and(a>=6, a<=10))
# returns (array([3, 4, 5]),)