wyodrębnij wartość kolumny na podstawie innej ramki danych pandy kolumn

Utknąłem na wyodrębnianiu wartości jednej zmiennej warunkującej inną zmienną. Na przykład następująca ramka danych:

A  B
p1 1
p1 2
p3 3
p2 4

Jak mogę sprawdzić, Akiedy B=3? Za każdym razem, gdy wyodrębniłem wartość A, otrzymałem obiekt, a nie ciąg.

Możesz użyć, locaby otrzymać serię spełniającą twój warunek, a następnie ilocuzyskać pierwszy element:

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'

Możesz spróbować query, co oznacza mniej pisania:

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

df[df['B']==3]['A'], zakładając, że df to twoje pandy.

Użyj, df[df['B']==3]['A'].valuesjeśli chcesz tylko element bez nawiasów

male_avgtip=(tips_data.loc[tips_data['sex'] == 'Male', 'tip']).mean()

Pracowałem również nad tymi operacjami klauzulowania i ekstrakcji dla mojego zadania.