Extraction de la valeur d'une colonne en fonction d'une autre colonne dans un cadre de données pandas

Question

Je suis un peu bloqué sur l'extraction de la valeur d'une variable conditionnant une autre variable. Par exemple, le dataframe suivant :

A  B
p1 1
p1 2
p3 3
p2 4

Comment puis-je obtenir la valeur de A quand B=3 ? Chaque fois que j'ai extrait la valeur de A j'ai obtenu un objet, pas une chaîne.

Answer 1

Vous pourriez utiliser loc pour obtenir des séries qui satisfont à votre condition et ensuite iloc pour obtenir le premier élément :

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'

Answer 2

Vous pouvez essayer query ce qui permet de moins taper :

df.query('B==3')['A']

Answer 3

df[df['B']==3]['A'] en supposant que df est votre pandas.DataFrame.

Answer 4

Utilice df[df['B']==3]['A'].values si vous voulez juste l'article lui-même sans les parenthèses

Answer 5

C'est plus facile pour moi de penser en ces termes, mais en empruntant d'autres réponses. La valeur que vous voulez est située dans la série :

df[*column*][*row*]

donde colonne y rangée pointent sur les valeurs que vous voulez renvoyer. Pour votre exemple, colonne est "A" et pour rangée vous utilisez un masque :

df['B'] == 3

Pour obtenir la première valeur appariée de la série, il existe plusieurs options :

df['A'][df['B'] == 3].values[0]
df['A'][df['B'] == 3].iloc[0]
df['A'][df['B'] == 3].to_numpy()[0]