将列作为参数输入到数据框中

Question

我有一个叫做table的Pandas数据表，我想查看它的一些列，比如：

table[['col_A', 'col_B', 'col_C', 'col_D', 'col_E']]

不过，我想把'col_C'、'col_D'和'col_E'这几列存到一个叫做var的变量里，这样我就可以通过类似下面的方式来查看我的结果：

table[['col_A', 'col_B', var]]

我不知道该怎么定义我的变量。我试着设置

var = 'col_C' + ', ' + 'col_D' + ', ' + 'col_E'

来连接字符串，但失败了。

如果只是查看一列是可以的。比如，我可以这样做 var2 = 'col_C'，然后我就可以通过下面的方式访问我的数据表：

table[['col_A', 'col_B', var2]]

Answer 1

索引是一个字符串的列表，所以可以使用列表连接的方法。

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