dataframe 열(column, 세로줄) 여러줄 multi-column 선택 in python

 

내용은 엄청 간단합니다. 그런데 말입니다. 막생 해보면 동시에 두개 열 선택이 잘 안됩니다.

자, 이제 예제 나갑니다.

>>> from pandas import Series, DataFrame
>>> raw_data = {'col0': [1, 2, 3, 4, 5],
...             'col1': [10, 20, 30, 40, 50],
...             'col2': [100, 200, 300, 400, 500]}
>>> data = DataFrame(raw_data)
>>> print(data)
   col0  col1  col2
0     1    10   100
1     2    20   200
2     3    30   300
3     4    40   400
4     5    50   500

간단하게 위와 같은 데이터를 만들었습니다.

즉 열 하나를 선택할때는 data['col1'] 이런식으로 하면됩니다.

>>> print(data['col1'])
0    10
1    20
2    30
3    40
4    50

잘 선택 되었습니다. 응용해보면, col1,col2를 선택하고 싶으면 data['col1','col2'] 넣으면 동작 하지 않습니다.

그러면 ?? 정답은 list로 묶어 주는것입니다.

>>> print(data[['col1','col2']])
   col1  col2
0    10   100
1    20   200
2    30   300
3    40   400
4    50   500

한개일때는 list로 안묶고, 두개이상일때는 list로 묶어주어야합니다. 

잘모를때는 한참 해메게 되는 tip이였습니다.

이쯤되면 궁금해지는데 한개일때도 리스트를 두번 사용하면 어떨까요? 이런 질문을 가지고 계신 분들을 위해서 시험해봤습니다. 정상 동작합니다.

>>> print(data[['col1']])
   col1
0    10
1    20
2    30
3    40
4    50