Панди: як отримати перший рядок кожної групи

Ви можете використовувати наступний базовий синтаксис, щоб отримати перший рядок кожної групи в pandas DataFrame:

 df. groupby (' column_name '). nth ( 0 )

У наступному прикладі показано, як використовувати цей синтаксис на практиці.

Приклад: отримати перший рядок кожної групи в Pandas

Припустімо, що у нас є наступні pandas DataFrame:

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' team ': ['A', 'A', 'B', 'B', 'B', 'C', 'C', 'C'],
                   ' points ': [18, 22, 19, 14, 14, 11, 20, 29],
                   ' assists ': [5, 19, 14, 8, 9, 12, 13, 8]})

#view DataFrame
df

	team points assists
0 to 18 5
1 To 22 19
2 B 19 14
3 B 14 8
4 B 14 9
5 C 11 12
6 C 20 13
7 C 29 8

Ми можемо використати наступний код, щоб отримати перший рядок для кожної команди:

 #get first row for each team
df. groupby (' team '). nth ( 0 )

	assist points
team		
At 18 5
B 19 14
C 11 12

Ми також можемо вказати as_index=False , щоб зберегти початкові значення індексу:

 #get first row for each team, keep original index values
df. groupby (' team ', as_index= False ). nth ( 0 )

        team points assists
0 to 18 5
2 B 19 14
5 C 11 12

Також зауважте, що ви можете передати список значень функції nth() , якщо хочете отримати перші n рядків кожної групи.

Наприклад, наступний код показує, як отримати перші два рядки кожної групи:

 #get first two rows for each team, keep original index values
df. groupby (' team ', as_index= False ). nth (( 0,1 ) )

team points assists
0 to 18 5
1 To 22 19
2 B 19 14
3 B 14 8
5 C 11 12
6 C 20 13

Примітка . Повну документацію щодо функції nth() можна знайти тут .

Додаткові ресурси

У наступних посібниках пояснюється, як виконувати інші типові операції в pandas:

Як отримати перший рядок Pandas DataFrame
Як видалити перший рядок у Pandas DataFrame
Як вставити рядок у Pandas DataFrame