[python/파이썬] 데이터 전처리 - 조건에 맞는 데이터 추출

조건에 맞는 데이터 추출

• 데이터 생성

import pandas as pd

df = pd.DataFrame({'a' : [i for i in range(1, 11)], 'b' : [i for i in range(11, 21)], 'c' : [i for i in range(21, 31)]})

df

• 문제 : a, c열을 출력

df[['a', 'c']]

• 문제 : a가 3 이상인 데이터 출력하기

df[df['a'] >= 3]

• 문제 : a가 3 이상인 데이터 중 a, c열만 출력하기

df[df['a'] >= 3][['a','c']]

• 문제 a 가 3 이상이고, b 가 16 미만인 데이터를 출력하기

df[(df['a'] >= 3) & (df['b'] < 16)]

a = (df['a'] >= 3) & (df['b'] < 16)

a

0 False
1 False
2 True
3 True
4 True
5 False
6 False
7 False
8 False
9 False
dtype: bool

type(a)

pandas.core.series.Series

df[a]

• 문제 : a가 3이하 이거나 7이상인 데이터 출력하기

df[(df['a'] <= 3) | (df['a'] >= 7)]

• 문제 : a 가 3 이상이고, b 가 16 미만이거나 c 가 30 인 데이터를 출력하기

df[(df['a'] >= 3) & ((df['b'] < 16) | (df['c'] == 30))]