Pandas: 행 번호가 있는 새 열을 추가하는 방법

Pandas DataFrame에 행 번호가 포함된 새 열을 추가하는 방법에는 두 가지가 있습니다.

방법 1: 할당() 사용

 df = df. assign (row_number= range ( len (df)))

방법 2: Reset_index() 사용

 df[' row_number '] = df. reset_index (). index

두 방법 모두 동일한 결과를 생성합니다.

다음 예에서는 다음 pandas DataFrame에서 실제로 각 메서드를 사용하는 방법을 보여줍니다.

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' team ': ['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H'],
                   ' points ': [5, 17, 7, 19, 12, 13, 9, 24],
                   ' assists ': [4, 7, 7, 6, 8, 7, 10, 11]})

#view DataFrame
print (df)

  team points assists
0 to 5 4
1 B 17 7
2 C 7 7
3 D 19 6
4 E 12 8
5 F 13 7
6 G 9 10
7:24 a.m. 11

예 1: 행 번호 열을 추가하려면 할당()을 사용하세요.

다음 코드는 할당() 함수를 사용하여 DataFrame의 각 행의 행 번호를 표시하는 row_number 라는 새 열을 추가하는 방법을 보여줍니다.

 #add column that contains row numbers
df = df. assign (row_number= range ( len (df)))

#view updated DataFrame
print (df)

  team points assists row_number
0 to 5 4 0
1 B 17 7 1
2 C 7 7 2
3 D 19 6 3
4 E 12 8 4
5 F 13 7 5
6 G 9 10 6
7:24 11 7

row_number 열의 값 범위는 0에서 7까지입니다.

예 2: Reset_index()를 사용하여 행 번호 열 추가

다음 코드는 Reset_index() 함수를 사용하여 DataFrame의 각 행의 행 번호를 표시하는 row_number 라는 새 열을 추가하는 방법을 보여줍니다.

 #add column that contains row numbers
df[' row_number '] = df. reset_index (). index

#view updated DataFrame
print (df)

  team points assists row_number
0 to 5 4 0
1 B 17 7 1
2 C 7 7 2
3 D 19 6 3
4 E 12 8 4
5 F 13 7 5
6 G 9 10 6
7:24 11 7

row_number 열의 값 범위는 0에서 7까지입니다.

이는 이전 예제의 결과와 일치합니다.

추가 리소스

다음 튜토리얼에서는 Pandas에서 다른 일반적인 작업을 수행하는 방법을 설명합니다.

Pandas: 두 열의 차이점을 찾는 방법
Pandas: 두 줄의 차이점을 찾는 방법
팬더:두 열을 빼는 방법