Python 当我们有20到50列数据时，如何在熊猫中创建包含3列或4列数据帧的列表？

当我们使用下面的代码创建列表时，我们会得到所有列的列表，但我只想得到3到5个特定列的列表

---

col_list=list（df）

使用

列表的切片：

df[df.columns[2:5]]

样本：

df = pd.DataFrame({'A':[1,2,3],
                   'B':[4,5,6],
                   'C':[7,8,9],
                   'D':[1,3,5],
                   'E':[5,3,6],
                   'F':[7,4,3]})

print (df)
   A  B  C  D  E  F
0  1  4  7  1  5  7
1  2  5  8  3  3  4
2  3  6  9  5  6  3

print (df.columns[2:5])
Index(['C', 'D', 'E'], dtype='object')

print (df[df.columns[2:5]])
   C  D  E
0  7  1  5
1  8  3  3
2  9  5  6

另一个解决方案：

col_list= list(df)
print (col_list[2:5])
['C', 'D', 'E']

print (df[col_list[2:5]])
   C  D  E
0  7  1  5
1  8  3  3
2  9  5  6