Python帮助-如何从一维数组中提取特定范围的值？ [重复]

Posted 2021-03-27

此问题已经在这里有了答案：

Numpy: find index of the elements within range（12个答案）

已关闭1分钟前。

在python中，有命令可以让我提取给定数组的特定子集吗？

例如，假设我具有以下一维数组：

x = np.array([1,0,35,9,1,23,10,2,4,8,3])

我如何只获得2到10之间的元素？因此，我希望输出为：

subx = [9,2,4,8,3]

答案

其中一种方法是使用list comprehension注意：包括2，下限，但不包括10，上限

x = [1,0,35,9,1,23,10,2,4,8,3]
y = [c for c in x if c >=2 and c<10]

由于您使用的是numpy您也可以执行以下操作：

import numpy as np
y = np.where(np.logical_and(x>=2, x<10))

另一答案

希望这会有所帮助

a=[1,0,35,9,1,23,10,2,4,8,3]
li=[]
for ele in a:
    if ele in range(2,10):
        li.append(ele)
print(li)

输出：

[9, 2, 4, 8, 3]

另一答案

由于您使用的是numpy数组，因此可以：

x[(x >= 2) & (x < 10)]