[Python知识库]Python 读取DataFrame前几行的某一列的具体值

在做实验时候需要读取DataFrame的前一行的某一列的值，我想要得到的是’>=50k’这个结果，原来的代码为

b = str(data['income'].head(1))

但是其输出为:

0    <=50K
Name: income, dtype: object

再进行下面的判断

        la='<=50K'
        lb='>50K'
        la_result=la in b
        lb_result=lb in b

显然la_result 和 rb_result都为False，这不是我想要的结果，正确的代码是取data[str2].head(1)里面的值’<=50k’或’>50k’，修改如下；

b = str(data[str2].head(1).values)  # str2为你想要取的列的列名

输出如下:

print(b)
['<=50K']