Python筛选列表指定区间数值触发NumPy索引报错的解决方案咨询

• 循环逻辑错误：for i in x的遍历逻辑中，i已经是x容器内存储的具体浮点数值，不是元素对应的索引下标。后续用x[i]取值相当于拿浮点数作为索引去访问元素，Python和NumPy均不支持浮点数作为合法索引，这是触发报错的核心原因。
• 语法不规范：if判断语句下方的执行代码没有做缩进处理，Python要求缩进标记代码块的归属，即使索引问题修复，这段代码也会触发缩进错误。

你的需求是筛选(-50, 50)开区间内的元素，如果需要包含端点-50、50，把下文判断条件中的</>替换为<=/>=即可。

通用写法（兼容普通Python列表、NumPy数组） #

修正原有循环逻辑，直接判断遍历得到的元素值即可：

y = []
for i in x:
    if -50 < i < 50:
        y.append(i)

也可以用更简洁的列表推导式实现：

y = [i for i in x if -50 < i < 50]

NumPy专属高效写法（从报错信息判断你使用的是NumPy数组） #

利用NumPy布尔索引可以无需写循环，实现高性能的批量筛选：

import numpy as np
# 如果x本身已经是NumPy数组可以跳过下一行
x = np.array(x)
y = x[(x > -50) & (x < 50)]
# 如果你需要得到普通Python列表而非NumPy数组，末尾追加.tolist()即可：
# y = y.tolist()

内容的提问来源于stack exchange，提问作者Nikita Agarwal