在 DF 中输入列作为参数

Question

我有一个名为 table 的 Pandas DataFrame，我想查看其中的一些列，例如： 表[['col_A', 'col_B', 'col_C', 'col_D', 'col_E']]

但是，我希望将“col_C”、“col_D”、“col_E”存储在名为 var 的变量中，以便看到我的结果执行类似的操作： 表[['col_A', 'col_B', var]]

我不知道如何定义我的变量。我尝试设置 var = 'col_C' + ', ' + 'col_D' + ', ' + 'col_E' 为了连接字符串，但失败了。

如果它只是一列，它就可以工作。我的意思是，执行 var2 = 'col_C'，然后我可以访问我的表 表[['col_A', 'col_B', var2]]

Answer 1

使用列表串联。

var = ['col_C', 'col_D', 'col_E']
table2 = table[['col_A', 'col_B'] + var]