根据另一个列熊猫数据框提取列值

一尘不染

根据另一个列熊猫数据框提取列值

python

我有点被困在提取另一个变量的条件变量的值上。例如，以下数据框：

A  B
p1 1
p1 2
p3 3
p2 4

如何获得Awhen的价值B=3？每当我提取的值时A，我都会得到一个对象，而不是字符串。

阅读 159

2020-12-20

共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'

2020-12-20