python - 快速切片数据集并存储在新的数据框中？

Question

我是 python 新手，非常感谢您的帮助。

我将 4 个具有相同标题的数据集导入到 python 中。现在我想创建 4 个数据框，其中仅包含 4 个数据集中选定的列。我知道如何以丑陋的方式完成此任务，但是执行此任务最有效的方法是什么？

我尝试了 for 循环，但无法使其工作:D

数据集导入为 df1,df2,df3,df4

dataset_list = (df1,df2,df3,df4)
new_dataframes= (df_1,df_2,df_3,df_4)
for i in dataset_list:
    for e in new_dataframes:
        e = i.loc[0:,['column1','column2','column3','column4']]

Answer 1

您可以使用字典理解:

cols = ['column1','column2','column3','column4']
dfs = {k: df[cols] for k, df in enumerate([df1, df2, df3, df4], 1)}

此方法的好处是它可以满足任意数量的项目，而无需手动增加变量名称。