python – 根据另一列pandas数据框提取列值

前端之家收集整理的这篇文章主要介绍了python – 根据另一列pandas数据框提取列值前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。
我在某个变量上提取一个变量条件的值,例如,以下数据框:

A B
p1 1
p1 2
p3 3
p2 4

当B = 3时,如何得到A的值?每次当我提取A的值时,我得到一个对象,而不是一个字符串.

解决方法

您可以使用 loc获得满足您的条件,然后 iloc获得第一个元素的系列:
  1. In [2]: df
  2. Out[2]:
  3. A B
  4. 0 p1 1
  5. 1 p1 2
  6. 2 p3 3
  7. 3 p2 4
  8.  
  9. In [3]: df.loc[df['B'] == 3,'A']
  10. Out[3]:
  11. 2 p3
  12. Name: A,dtype: object
  13.  
  14. In [4]: df.loc[df['B'] == 3,'A'].iloc[0]
  15. Out[4]: 'p3'

猜你在找的Python相关文章