Python 读取DataFrame前几行的某一列的具体值

矫情吗；* 2023-10-01 10:34 81阅读 0赞

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

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

但是其输出为:

  1. 0    <=50K
  2. Name: income, dtype: object

再进行下面的判断

  1. la='<=50K'
  2.         lb='>50K'
  3.         la_result=la in b
  4.         lb_result=lb in b

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

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

输出如下:

  1. print(b)
  2. ['<=50K']

发表评论

表情：
评论列表 （有 0 条评论，81人围观）

还没有评论，来说两句吧...

相关阅读