Pandas: як додати новий стовпець із номерами рядків

Є два способи додати новий стовпець, що містить номери рядків, до pandas DataFrame:

Спосіб 1: Використовуйте assign()

 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: використовуйте assign(), щоб додати стовпець з номером рядка

Наступний код показує, як використовувати функцію assign() для додавання нового стовпця під назвою row_number , який відображає номер рядка кожного рядка в DataFrame:

 #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() , щоб додати новий стовпець під назвою row_number , який відображає номер рядка кожного рядка в DataFrame:

 #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:

Панди: як знайти різницю між двома колонками
Панди: як знайти різницю між двома лініями
Панди: як відняти два стовпці