我在某个变量上提取一个变量条件的值,例如,以下数据框:
A B
p1 1
p1 2
p3 3
p2 4
当B = 3时,如何得到A的值?每次当我提取A的值时,我得到一个对象,而不是一个字符串.
解决方法
您可以使用
loc
获得满足您的条件,然后
iloc
获得第一个元素的系列:
In [2]: df Out[2]: A B 0 p1 1 1 p1 2 2 p3 3 3 p2 4 In [3]: df.loc[df['B'] == 3,'A'] Out[3]: 2 p3 Name: A,dtype: object In [4]: df.loc[df['B'] == 3,'A'].iloc[0] Out[4]: 'p3'