根据列表动态选择数据框中的列

Question

我有一个动态变化的列名列表，基于我需要从数据框中选择这些列并将其连接起来的列名

例如

column_names = ['Year', 'State', 'Pol_no']

数据帧：

Year State  Pol_no           Data Source1(sum)
2010     A       1                 90
2010     B       2                180
2010     C       6                270
2010     D       4                360
2010     F       5                450
2011     A       1                 90
2011     B       2                180
2011     C       3                100
2011     D       4                360
2011     E       5                450
2012     A       1                 90
2012     B       2                180
2012     C       3                270
2012     D       4                360
2013     E       5                450
2014     A       5                900

在这种情况下，我必须从数据框中选择year，state，pol_no列并连接它。

在其他情况下，它可能只是列表中的两列[＆＃39;状态＆＃39; ＆＃39; Pol_no＆＃39;] 那么我需要从dataframe中选择上面的两列并连接。

任何可以帮助我做到这一点的通用函数

Answer 1

使用：

df1 = df[['Year', 'State', 'Pol_no']]