如何基于指定条件从Python数组列表中移除不符合要求的数组

首先完成需求仅需两步：筛选符合条件的重复实验数组、对符合要求的数组按时间步计算均值，完整可运行代码如下：

import numpy as np

# 1. 读取5次重复实验的结果
repetitions = []
for count in range(5):
    rank = np.loadtxt(f"rank_{count}_prev")
    repetitions.append(rank)

# 2. 筛选所有元素都不低于0.0001的数组
valid_reps = []
for rep in repetitions:
    # 用all()判断数组所有元素是否都满足阈值要求，返回单个布尔值
    if (rep >= 0.0001).all():
        valid_reps.append(rep)

# 3. 按时间步计算均值，axis=0表示按数组维度聚合计算
if valid_reps: # 避免无有效数组时报错
    step_mean = np.array(valid_reps).mean(axis=0)
else:
    step_mean = np.array([])
    print("无符合条件的重复实验数组")

# 输出结果，step_mean就是长度为60的每个时间步均值数组
print(step_mean)

原代码问题说明 #

• 第一次报错NameError: name 'rank' is not defined是因为你在循环中直接del rank删除了迭代变量，后续循环访问时变量不存在
• 第二次报错ValueError是因为你直接将numpy数组作为if判断条件，numpy不支持直接对多元素数组做布尔判断，需要用all()/any()明确判断规则
• 第三次结果不符合预期是因为你循环遍历了数组内的每个元素，每遇到一个符合要求的元素就添加一次完整数组，导致数组被重复添加多次

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