Как разделить dataframe pandas на части

Вы можете использовать следующий базовый синтаксис, чтобы разбить DataFrame pandas на более мелкие части:

 #specify number of rows in each chunk
n= 3

#split DataFrame into chunks
list_df = [df[i:i+n] for i in range(0, len (df),n)]

Затем вы можете получить доступ к каждому фрагменту, используя следующий синтаксис:

 #access first chunk
list_df[0]

В следующем примере показано, как использовать этот синтаксис на практике.

Пример: разделить DataFrame Pandas на части

Предположим, у нас есть следующий DataFrame pandas с девятью строками, содержащими информацию о различных баскетболистах:

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' team ': ['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I'],
                   ' points ': [18, 22, 19, 14, 14, 11, 20, 28, 23],
                   ' assists ': [5, 7, 7, 9, 12, 9, 9, 4, 11],
                   ' rebounds ': [11, 8, 10, 6, 6, 5, 9, 12, 10]})

#view DataFrame
print (df)

  team points assists rebounds
0 A 18 5 11
1 B 22 7 8
2 C 19 7 10
3 D 14 9 6
4 E 14 12 6
5 F 11 9 5
6 G 20 9 9
7:28 4 12
8 I 23 11 10

Мы можем использовать следующий синтаксис, чтобы разделить DataFrame на фрагменты, каждый из которых состоит из трех строк:

 #specify number of rows in each chunk
n= 3

#split DataFrame into chunks
list_df = [df[i:i+n] for i in range(0, len (df),n)]

Затем мы можем использовать следующий синтаксис для доступа к каждой части:

 #view first chunk
print (list_df[0])

  team points assists rebounds
0 A 18 5 11
1 B 22 7 8
2 C 19 7 10

#view second chunk
print (list_df[1])

  team points assists rebounds
3 D 14 9 6
4 E 14 12 6
5 F 11 9 5

#view third chunk
print (list_df[2])

  team points assists rebounds
6 G 20 9 9
7:28 4 12
8 I 23 11 10

Обратите внимание, что каждый фрагмент содержит три строки, как мы указали.

Обратите внимание, что в этом примере мы использовали DataFrame всего с девятью строками в качестве простого примера.

На практике вы, скорее всего, будете работать с DataFrame, содержащим сотни тысяч или даже миллионы строк.

Вы можете использовать тот же синтаксис, что и в этом примере, чтобы разделить DataFrame на фрагменты определенного размера.

Дополнительные ресурсы

В следующих руководствах объясняется, как выполнять другие распространенные задачи в pandas:

Pandas: как разделить DataFrame по значению столбца
Pandas: как разделить строковый столбец на несколько столбцов
Pandas: как разделить столбец списков на несколько столбцов