我有点卡在以另一个变量为条件提取一个变量的值。例如,以下数据框:
A B p1 1 p1 2 p3 3 p2 4
我怎样才能得到Awhen的价值B=3?每次提取 的值时A,我都会得到一个对象,而不是字符串。
A
B=3
您可以使用loc获取满足您条件的系列,然后iloc获取第一个元素:
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'