Programming
pandas dataframe 조건 조회하기
engyjoon
2021. 1. 22. 13:32
[준비] dataframe 생성
import pandas as pd
list_val = [
['xab1', 'yab1', 1, 2, 3],
['xab2', 'yab2', 1, 5, 6],
['xcd3', 'ycd3', 7, 8, 9],
['xcd4', 'ycd4', 10, 11, 12],
['xef5', 'yef5', 13, 14, 15]
]
list_col = ['col1', 'col2', 'col3', 'col4', 'col5']
list_idx = ['a', 'b', 'c', 'd', 'e']
df_a = pd.DataFrame(list_val, columns=list_col, index=list_idx)
df_a
1개 숫자 column을 비교하여 조회한다.
df_a.loc[df_a['col3'] >= 10]
1개 문자 column을 비교하여 조회한다.
df_a.loc[df_a['col1'] == 'xab1']
여러 개 column을 비교하여 조회한다.
df_a.loc[(df_a['col3'] == 1) & (df_a['col4'] > 3)]
isin 함수를 사용해 조회한다.
df_a.loc[df_a['col1'].isin(['xab1', 'xef5'])]
between 함수를 사용해 조회한다.
df_a.loc[df_a['col4'].between(8, 14)]