获取数组的特定索引

Question

import numpy as np
X = [-10000, -1000, -100, -10, -1, 0, 1, 10, 100, 1000, 10000]
l = np.where(np.array(X) > 100)[0]

所以我有

l = array([ 9, 10], dtype=int64)

现在我想得到一个新的数组 X，其中的元素 l 作为索引。 我想得到：

X = [1000, 10000]

我想到了：

X = X[l]

但它不起作用。 在这种情况下，正确的 function 是什么？ 我不想使用 for 循环。

Answer 1

您需要将列表X转换为 numpy 数组，然后才能使用索引列表对其进行索引：

>>> X = np.array(X)[l]
>>> X
array([ 1000, 10000])

Answer 2

您应该将列表 X 转换为 numpy 数组

x=np.array(X)

然后使用索引

x=x[x>100]

这将在不使用 where() function 的情况下获得您需要的结果

“x>100”表达式创建一个boolean数组，其中True元素对应于x数组中大于100的元素。这个数组可以作为索引来提取满足条件的元素。