[ Pandas ] 레코드 추출하기

- 레코드 추출

데이터 프레임으로부터 다양한 조건으로 원하는 특정 행들을 추출할 수 있습니다.

df = pd.DataFrame({
    'btype' : ['a','b','o','ab', 'a'],
    'height' : [175, 163, 168, 180, 172],
    'weight' : [78,56,88,73,77]    
})
df

 

 

- 데이터 순서로부터 추출

head 나 tail 로 앞,뒷쪽 일부 데이터를 추출할 수 있습니다.

df.tail(2)

df.head(2)

df[1:3]

df[3:]

 

 

 

- 조건으로 추출 (loc)

loc 는 db 의 셀렉트 문과 비슷한 일을 해줄수있습니다.

이것을 사용하여 다양한 조건으로 데이터를 추출할 수 있습니다.

df.loc[ df['btype'] == 'a']

df.loc[ df['height'] > 170 ]

 

 

 

& (and) 나 |(or) 를 이용해 조건을 추출할 수도 있습니다.

하지만 사용할때 각 조건들은 괄호를 붙여주어야 합니다.

df.loc[ ( df['height'] > 170 ) & ( df['btype'] == 'a' ) ]

 

 

 

df.loc[ ( df['height'] > 170 ) | ( df['btype'] == 'o' ) ]

 

 

 

다음과 같이 특정 컬럼만 추출도 가능합니다.

df.loc[df['btype'] == 'a', ['height','btype']]