[Pandas] 데이터 프레임에 컬럼 이름 추가하기

 

배경

---

데이터 프레임을 분석하는데 있어 컬럼 명이 존재하지 않아서 별도로 달아주어야 할 경우가 있다.

 

 

해결 코드

---

column_name = ['column1', 'column2', 'column3', 'column4', 'column5']

dataframe = pd.read_csv(fullpath)

data = frame.values.tolist()

new_dataframe = pd.DataFrame(data, columns=column_name)

 

 

코드 해설

---

1. 먼저 생성할 컬럼명을 컬럼의 개수에 맞게 작성을 해준다.
2. pandas를 통해 분석할 데이터 프레임을 읽어들인다.
3. 데이터 프레임의 values를 모두 리스트에 넣으며 이때 리스트에 들어가는 각 요소는 데이터 프레임의 한 행이 된다.
4. pd.DataFrame을 사용하여 columns 인자에 생성한 리스트 변수를 넣어줌으로써 컬럼명이 생성된 데이터 프레임이 생성된다.