从dataframe获取指定的列集

Question

我使用手动选择数据框中的列

 df_final = df[['column1','column2'.......'column90']]

相反，我提供了一个列表中的列名列表

 dp_col =  [col for col in df if col.startswith('column')]

但不确定如何使用此列表仅从源数据帧获取这些列集

Answer 1

您可以将其用作要选择的列列表，因此：

df_final = df[[col for col in df if col.startswith('column')]]

字符串列表的“来源”并不重要，只要您将字符串列表传递给下标，这将正常工作

Answer 2

将loc访问与布尔屏蔽一起使用：

df.loc[:, df.columns.str.startswith('column')]