Split Datenrahmen in relativ gleichmäßigen Stücke zu Länge nach

Question

Ich habe eine Funktion erstellen, die Datenrahmen in Stücke der benötigten Größe zur Verfügung gestellt spaltete würde. Zum Beispiel, wenn Datenrahmen 1111 Zeilen enthält, mag ich in der Lage sein Chunkgröße von 400 Zeilen zu spezifizieren, und erhält drei kleinere Datenrahmen mit Größen von 400, 400 und 311. Gibt es eine Komfortfunktion die Arbeit zu tun? Was wäre der beste Weg zum Speichern und Iterieren über geschnittene Datenrahmen?

Beispiel Datenrahmen

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

können Sie .groupby wie unten verwenden.

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