[Python] 데이터프레임 컬럼 추가하기

안녕하세요! 데이터프레임의 컬럼 추가 방법에 대해 알아보도록 하겠습니다.

데이터프레임의 컬럼 추가

데이터프레임에서 컬럼을 추가하는 방법에는 여러 가지가 있습니다. 여기서는 파이썬의 pandas 모듈을 사용하는 방법에 대해 설명하겠습니다.

1. 새로운 컬럼 생성하기

가장 간단한 방법은 pandas 모듈의 데이터프레임에 새로운 컬럼을 추가하는 것입니다. 다음과 같이 새로운 컬럼을 생성하여 데이터프레임에 추가할 수 있습니다.

 

import pandas as pd

 

df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]})

df['D'] = [10, 11, 12]

print(df)

 

위 코드를 실행하면 다음과 같은 결과가 출력됩니다.

   A  B  C   D
0  1  4  7  10
1  2  5  8  11
2  3  6  9  12

2. insert() 함수 사용하기

insert() 함수를 사용하여 지정한 위치에 새로운 컬럼을 추가할 수도 있습니다. insert() 함수는 데이터프레임에서 지정한 위치에 행이나 열을 추가합니다.

 

import pandas as pd

 

df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]})

df.insert(1, 'D', [10, 11, 12])

print(df)

 

위 코드를 실행하면 다음과 같은 결과가 출력됩니다.

   A   D  B  C
0  1  10  4  7
1  2  11  5  8
2  3  12  6  9

 

위 코드에서 insert() 함수는 1번째 위치에 D라는 새로운 컬럼을 추가하였습니다.

3. assign() 함수 사용하기

assign() 함수를 사용하여 새로운 컬럼을 추가할 수도 있습니다. assign() 함수는 새로운 컬럼을 추가한 새로운 데이터프레임을 반환합니다.

 

import pandas as pd

 

df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]})

df = df.assign(D=[10, 11, 12])

print(df)

 

위 코드를 실행하면 결과는 insert() 함수를 사용한 예제와 동일합니다.

   A  B  C   D
0  1  4  7  10
1  2  5  8  11

2  3  6  9  12  

마무리

이상으로 데이터프레임에서 컬럼을 추가하는 방법에 대해 알아보았습니다. 각각의 방법은 상황에 따라 다르게 사용될 수 있으니 필요에 따라 적절한 방법을 선택하여 사용하시기 바랍니다.