如何在Python中删除含0值的数组元素及对应时间元素？

现有两个各含300个元素的列表time_list和EL1_list，需删除EL1_list中值为0的元素，同时删除time_list中对应的元素，保证两者元素数量一致，方便后续绘图。

原实现代码存在逻辑错误，无法达成目标：

EL1_array = list(zip(time_list, EL1_list))
print(EL1_array)
for (i,j) in EL1_array:
    if j == 0:
        EL1_array.remove(j)
        print(EL1_array)

原代码问题分析 #

• 遍历列表时直接修改原列表，会导致迭代器跳过部分元素，无法遍历所有元素
• EL1_array.remove(j)逻辑错误：j是元组的第二个值（如0.0），但列表中的元素是(i,j)格式的元组，remove无法找到匹配元素

正确实现方法 #

方法1：列表推导式（推荐，简洁高效） #

直接筛选出EL1值不为0的配对，再拆分回两个列表：

# 筛选符合条件的(time, EL1)配对
filtered_pairs = [(t, el) for t, el in zip(time_list, EL1_list) if el != 0]
# 拆分回两个单独列表
time_filtered, EL1_filtered = zip(*filtered_pairs)
# 若需要列表类型（zip返回元组），可转换
time_filtered = list(time_filtered)
EL1_filtered = list(EL1_filtered)

方法2：遍历收集新列表（直观易懂） #

逐个检查元素，将符合条件的元素分别加入新列表：

time_filtered = []
EL1_filtered = []
for t, el in zip(time_list, EL1_list):
    if el != 0:
        time_filtered.append(t)
        EL1_filtered.append(el)

方法3：反向遍历删除（不推荐，效率较低） #

如果必须修改原列表而非创建新列表，可通过反向遍历索引避免迭代混乱：

pairs = list(zip(time_list, EL1_list))
# 从后往前遍历索引
for idx in range(len(pairs)-1, -1, -1):
    if pairs[idx][1] == 0:
        del pairs[idx]
# 将拆分结果赋值回原列表
time_list[:], EL1_list[:] = zip(*pairs)

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