Split dataframe in blocchi relativamente pari in base alla lunghezza

Question

Devo creare una funzione che divida il dataframe diviso in blocchi della dimensione necessaria. Ad esempio se dataframe contiene 1111 righe, voglio essere in grado di specificare una dimensione del blocco di 400 righe e ottenere tre dataframe più piccoli con dimensioni di 400, 400 e 311. Esiste una funzione di convenienza per eseguire il lavoro? Quale sarebbe il modo migliore per archiviare e iterare su dataframe affettato?

Esempio dataframe

import numpy as np 
import pandas as pd 

test = pd.concat([pd.Series(np.random.rand(1111)), pd.Series(np.random.rand(1111))], axis = 1) 

Answer 1

È possibile utilizzare .groupby come di seguito.

for g, df in test.groupby(np.arange(len(test)) // 400): 
    print(df.shape) 
# (400, 2) 
# (400, 2) 
# (311, 2)