python - manejo - extraer el valor de la columna en base a otro marco de datos de pandas de columna

manejo de data frame python

Question

Estoy atascado en extraer el valor de una variable condicionante en otra variable. Por ejemplo, el siguiente marco de datos:

A B
p1 1
p1 2
p3 3
p2 4

¿Cómo puedo obtener el valor de A cuando B=3 ? Cada vez que extraía el valor de A , obtenía un objeto, no una cadena.

Answer 1

Puede probar la query , que es menos tipeada:

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

Answer 2

Puede usar loc para obtener series que satisfagan su condición y luego iloc para obtener el primer elemento:

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 3

Utilice df[df[''B'']==3][''A''].values si solo desea el elemento en sí sin los corchetes

Answer 4

df[df[''B'']==3][''A''] , suponiendo que df es su pandas.DataFrame.