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

发布时间：2020-12-16 23:07:54 所属栏目：Python 来源：网络整理

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

我在某个变量上提取一个变量条件的值,例如,以下数据框：

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'

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!