Pandas에서 할당() 메서드를 사용하는 방법(예제 포함)

insert() 메소드를 사용하여 pandas DataFrame에 새 열을 추가할 수 있습니다.

이 방법은 다음 기본 구문을 사용합니다.

 df. assign (new_column = values)

이 메서드는 콘솔에 새 DataFrame만 표시할 뿐 원본 DataFrame을 실제로 수정하지는 않는다는 점에 유의하는 것이 중요합니다.

원본 DataFrame을 수정하려면 할당() 메서드의 결과를 새 변수에 저장해야 합니다.

다음 예에서는 다음 pandas DataFrame에서 할당() 메서드를 다양한 방식으로 사용하는 방법을 보여줍니다.

 import pandas as pd

#define DataFrame
df = pd. DataFrame ({' points ': [25, 12, 15, 14, 19, 23, 25, 29],
                   ' assists ': [5, 7, 7, 9, 12, 9, 9, 4],
                   ' rebounds ': [11, 8, 10, 6, 6, 5, 9, 12]})

#view DataFrame
print (df)

   points assists rebounds
0 25 5 11
1 12 7 8
2 15 7 10
3 14 9 6
4 19 12 6
5 23 9 5
6 25 9 9
7 29 4 12

예시 1: DataFrame에 새 변수 할당

다음 코드는 할당() 메서드를 사용하여 포인트 열의 값에 2를 곱한 값과 동일한 points2 라는 DataFrame에 새 변수를 추가하는 방법을 보여줍니다.

 #add new variable called points2
df. assign (points2 = df.points * 2 )

    points assists rebounds points2
0 25 5 11 50
1 12 7 8 24
2 15 7 10 30
3 14 9 6 28
4 19 12 6 38
5 23 9 5 46
6 25 9 9 50
7 29 4 12 58

이 할당() 메서드는 원본 DataFrame을 수정하지 않습니다.

원본 DataFrame을 인쇄하면 변경되지 않은 상태로 유지되는 것을 볼 수 있습니다.

 #print original DataFrame
print (df)

   points assists rebounds
0 25 5 11
1 12 7 8
2 15 7 10
3 14 9 6
4 19 12 6
5 23 9 5
6 25 9 9
7 29 4 12

Assign() 메소드의 결과를 저장하기 위해 결과를 새 DataFrame에 저장할 수 있습니다.

 #add new variable called points2 and save results in new DataFrame
df. assign (points2 = df.points * 2 )

#view new DataFrame
print (df_new)

   points assists rebounds points2
0 25 5 11 50
1 12 7 8 24
2 15 7 10 30
3 14 9 6 28
4 19 12 6 38
5 23 9 5 46
6 25 9 9 50
7 29 4 12 58

이제 df_new 라는 새 DataFrame에는 우리가 만든 points2 열이 포함됩니다.

예시 2: DataFrame에 여러 개의 새 변수 할당

다음 코드는 할당() 메서드를 사용하여 DataFrame에 세 개의 새 변수를 추가하는 방법을 보여줍니다.

 #add three new variables to DataFrame and store results in new DataFrame
df_new = df. assign (points2 = df. points * 2 ,
                   assists_rebs = df. assists + df. rebounds ,
                   conference = ' Western ')

#view new DataFrame
print (df_new)

   points assists rebounds points2 assists_rebs conference
0 25 5 11 50 16 Western
1 12 7 8 24 15 Western
2 15 7 10 30 17 Western
3 14 9 6 28 15 Western
4 19 12 6 38 18 Western
5 23 9 5 46 14 Western
6 25 9 9 50 18 Western
7 29 4 12 58 16 Western

DataFrame에 세 개의 새로운 열이 추가되었습니다.

참고 : 여기에서 pandas 할당() 메서드에 대한 전체 문서를 찾을 수 있습니다.

추가 리소스

다음 튜토리얼에서는 Pandas에서 다른 일반적인 기능을 사용하는 방법을 설명합니다.

Pandas에서 explain() 함수를 사용하는 방법
Pandas에서 idxmax() 함수를 사용하는 방법
Pandas에서 선택한 열에 함수를 적용하는 방법