小编典典

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

all

我有点卡在以另一个变量为条件提取一个变量的值。例如,以下数据框:

A  B
p1 1
p1 2
p3 3
p2 4

我怎样才能得到Awhen的价值B=3?每次提取 的值时A,我都会得到一个对象,而不是字符串。


阅读 57

收藏
2022-07-09

共1个答案

小编典典

您可以使用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'
2022-07-09