Python如何将循环结果保存到变量以自动执行数据删除操作

你可以通过以下方式把循环筛选出的索引自动存入变量，无需手动复制粘贴：

方法1：普通循环收集结果 #

先初始化一个空列表，在循环里把符合条件的索引添加进去：

# 创建空列表存储要删除的索引
to_drop = []
for i in range(1,9):
    if i % 2 == 0:
        to_drop.append(i)

此时to_drop的值就是[2,4,6,8]，直接传入drop方法即可：

mydata.drop(to_drop, axis=0, inplace=True)

方法2：列表推导式（更简洁） #

用Python的列表推导式可以一行完成筛选，代码更紧凑：

to_drop = [i for i in range(1,9) if i % 2 == 0]
mydata.drop(to_drop, axis=0, inplace=True)

这样就能自动完成索引收集和批量删除，不用再手动复制循环输出的内容了。

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