如何将数据按行分割成小数据集？

Question

我有一个 60000x32 的数据集。我想这样分割它 第一次分割= (0:126,:) 第二=(126:252,:) ` 第三= (252:378,:) .. ..直到最后...

应该按这个顺序。每个分割的大小需要为 126x32。我怎样才能做到这一点？

Answer 1

尝试这样的事情

def segment_data(data, n_rows):
    """
    :param data: dataframe with 60000 rows and 32 features
    :param n_rows: number of rows in each segment
    :return: list of dataframes with 126 rows and 32 features
    """
    segments = []
    for i in range(0, len(data), n_rows):
        segment = data.iloc[i:i + n_rows, :]
        segments.append(segment)
    return segments


segments = segment_data(data, 126)