본문 바로가기

Python

[Python] list를 DataFrame으로 만들기

업무를 하다보면 dataframe을 만들어야 하는 경우가 많다. 여러 방법이 있겠지만 내가 느낀 가장 쉬운 방법은 모든 데이터를 list type으로 만들어 변환시켜주는 것이다. 딕셔너리에서 dataframe으로 만들 때 결국 list comprehension을 사용하여 만들기 때문에 처음부터 list로 만들어 주는게 편할 것이다. 제일 흔한 두가지 경우를 살펴보자

 

1. 각각의 리스트가 하나의 row인 경우

import pandas as pd

project_list = [['a', 1, '가'], ['b', 2, '나'], ['c', 3, '다']]

df = pd.DataFrame(project_list, columns=['영문', '숫자', '한글'])

2. 각각의 리스트가 하나의 col인 경우

import pandas as pd

project_list = [['a','b','c'], [1, 2, 3], ['가','나','다']]

df = pd.DataFrame(project_list).transpose()
df.columns = ['영문', '숫자', '한글']

이렇게 둘 다 실행시키면 같은 결과가 나오는 것을 볼 수 있다.