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)]