파이썬 조건부 인덱싱을 사용하여 셀에서 스칼라 값을 얻는 방법

아래에 표시된 데이터 프레임이 있습니다. A (스크립트의 변수) 값에 따라 B 열의 스칼라 값을 가져와야합니다. loc () 함수를 시도하고 있지만 스칼라 값 대신 Series를 반환합니다. 스칼라 값 ()은 어떻게 얻습니까?

>>> x = pd.DataFrame({'A' : [0,1,2], 'B' : [4,5,6]})
>>> x
   A  B
0  0  4
1  1  5
2  2  6

>>> x.loc[x['A'] == 2]['B']
2    6
Name: B, dtype: int64

>>> type(x.loc[x['A'] == 2]['B'])
<class 'pandas.core.series.Series'>

 

해결 방법

 

우선, .loc 에서 행 및 열 인덱스에 모두 액세스하는 것이 좋습니다.

x.loc[x['A'] == 2, 'B']

둘째, 시리즈 또는 데이터 프레임에서 .values ​​를 사용하여 항상 기본 numpy 행렬을 얻을 수 있습니다.

In : x.loc[x['A'] == 2, 'B'].values[0]
Out: 6

 

참조 페이지 https://stackoverflow.com/questions/30813088