Python 기본 이중 배열과 pandas 데이터 프레임 변환

Python에서는 기본 이중 배열과 pandas DataFrame 사이를 자주 변환한다. 표 형태의 데이터를 계산하거나 CSV와 엑셀 파일로 다룰 때는 DataFrame이 편하고, 간단한 반복 처리에는 기본 배열이 가볍다.

이중 배열에서 DataFrame 만들기

import pandas as pd

rows = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
df = pd.DataFrame(rows, columns=["A", "B", "C"])

DataFrame에서 배열 만들기

import pandas as pd

df = pd.DataFrame({"A": [1, 2, 3], "B": [4, 5, 6]})
rows = df.to_numpy().tolist()

df.values도 사용할 수 있지만, 현재 코드에서는 반환 의도가 더 분명한 to_numpy()를 쓰는 편이 낫다. 열 이름과 데이터 타입을 유지해야 한다면 배열로 바꾸기 전에 그 정보가 필요한지 먼저 확인한다.