pandas数据平均切分后，变换数组形状，并加标签

import math
import pandas as pd
import numpy as np
from tqdm import tqdm

data0 = pd.read_csv("filename")
def cut_df(df, n):
    df=df[1:120001]#取前120000行数据
    df_num = len(df)
    every_epoch_num = math.floor((df_num/n))#floor函数向下取整数
    df1=pd.DataFrame()#创建空dataframe对象
    for index in tqdm(range(n)):
        if index < n-1:
            df_tem = df[every_epoch_num * index: every_epoch_num * (index + 1)]
        else:
            df_tem = df[every_epoch_num * index:]        
        df_tem=np.array(df_tem)
        df_tem=df_tem.reshape(1,-1)#array变形状 
        df_tem=pd.DataFrame(df_tem)
        df1 = pd.concat([df1, df_tem],axis=0)#变形状后上下连接，axis=1是左右连接
    return df1
# data.insert(2,'c','')
# 2 ：插入的列的位置
# ‘c'：待插入列的列名
# ‘ ’：插入的值，这里插入的是空值
df0=cut_df(data0,200)
df0.insert(0,'label',0)
df0.to_csv("./df0.csv", index=False)
df0