前言
查阅pandas发现pandas只提供了根据列名获取数据的方法,没有直接根据索引获得列的方法,所以自己绕了个弯先获取列名再根据索引获取列名,最终达到根据索引获取数据的效果
演示
df = pd.DataFrame({'A': np.random.randint(1, 100, 10),
'B': np.random.randint(1, 100, 10),
'C': np.random.randint(1, 100, 10)})
df_title = df.columns.values.tolist() # 获得列名
print(df_title)
print(df[df_title[0:2]]) # 打印前两列
这样子就实现了根据索引获取数据,主要解决想看连续的几列,但又不想一一输入列名的问题。此外,还能用于循环来解决一下特殊的需求。