Python/NumPy | Pandas

[Python] Pandas - DataFrame column 추가

비번변경 2023. 1. 2. 17:44

개요

pandas DataFrame에 column 추가하는 방법을 정리한다.

 

 

일반적인 방법

Dictionary와 비슷하게 대괄호에 column 이름을 지정하여 값을 초기화한다. 추가되는 열은 가장 오른쪽에 위치한다.

df1 = pd.DataFrame({
    '고객번호': [1001, 1002, 1003, 1004, 1005, 1006, 1007],
    '이름': ['둘리', '도우너', '또치', '길동', '희동', '마이콜', '영희']
}, columns=['고객번호', '이름'])
df1
df1['금액'] = 0
df1

# 실행 결과
고객번호	이름
0	1001	둘리
1	1002	도우너
2	1003	또치
3	1004	길동
4	1005	희동
5	1006	마이콜
6	1007	영희

고객번호	이름	금액
0	1001	둘리	0
1	1002	도우너	0
2	1003	또치	0
3	1004	길동	0
4	1005	희동	0
5	1006	마이콜	0
6	1007	영희	0

 

 

DataFrame.insert

DataFrame.insert 함수를 사용하여 열을 추가할 수 있다. 매개변수로 열을 삽입할 위치와 column 이름과 값을 전달한다.

df1.insert(loc=0, column='날짜', value=datetime.datetime.now())
df1

# 실행 결과
날짜	고객번호	이름	금액
0	2022-12-22 07:37:28.185321	1001	둘리	0
1	2022-12-22 07:37:28.185321	1002	도우너	0
2	2022-12-22 07:37:28.185321	1003	또치	0
3	2022-12-22 07:37:28.185321	1004	길동	0
4	2022-12-22 07:37:28.185321	1005	희동	0
5	2022-12-22 07:37:28.185321	1006	마이콜	0
6	2022-12-22 07:37:28.185321	1007	영희	0

 

 

참고 문서

https://bio-info.tistory.com/16

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