[python] 다른 열 pandas 데이터 프레임을 기반으로 열 값 추출

나는 다른 변수에서 한 변수 조건의 값을 추출하는 데 붙어 있습니다. 예를 들어, 다음 데이터 프레임 :

A  B
p1 1
p1 2
p3 3
p2 4

A언제 가치를 얻을 수 B=3있습니까? 의 값을 추출 할 때마다 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'


답변

query입력이 적은을 시도해 볼 수 있습니다 .

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


답변

df[df['B']==3]['A'], df가 pandas.DataFrame이라고 가정합니다.


답변

df[df['B']==3]['A'].values괄호없이 항목 자체를 원하는 경우 사용


답변

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

나는 또한 내 과제를 위해이 클로징 및 추출 작업을 수행했습니다.


답변