pandas dataframe에서 column으로 dataframe 조회하기

[준비] dataframe 생성

import pandas as pd

list_val = [
    ['x1', 'y1', 1, 2, 3],
    ['x2', 'y2', 1, 5, 6],
    ['x3', 'y3', 7, 8, 9]
]

list_col = ['col1', 'col2', 'col3', 'col4', 'col5']
list_idx = ['a', 'b', 'c']

df_a = pd.DataFrame(list_val, columns=list_col, index=list_idx)
df_a

---

1개의 column의 구성된 dataframe을 조회한다.

loc를 사용하는 대신 대괄호 2개를 사용한다.

df_a[['col1']]

※ df_a.loc[:, ['col1']] 표현으로도 동일한 결과를 조회할 수 있다. (row 생략)

※ ['col1'] 대신 'col1'을 입력하면 dataframe이 아니라 series가 조회된다.

 

---

2개 이상의 column으로 구성된 dataframe을 조회한다.

df_a[['col1', 'col4']]

 

---

[참고]

https://pandas.pydata.org/docs/reference/api/pandas.DataFrame.loc.html