工作中我们需要对DataFrame进行比较多样化的操作,那么当我们需要取出某一列数据时该如何进行操作呢,下面的代码就是对几种方法的一个总结,希望对各位有所帮助。可以根据自己的需求取出需要的一列或多列。
import pandas as pd
data1 = pd.DataFrame(...) #任意初始化一个列数为3的DataFrame
data1.columns=['a', 'b', 'c']
1.
data1['b']
#这里取到第2列(即b列)的值
2.
data1.b
#效果同1,取第2列(即b列)
#这里b为列名称,但必须是连续字符串,不能有空格。如果列名有空格,则只能采取第1种方法
3.
data1[data1.columns[1:]]
#这里取data1的第2列到最后一列的所有数据
ps:这种方法容易有误解,注意区分
data1[5:10]
#这里取6到11行的所有数据,而不是列数据