在做实验时候需要读取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']