pandas 水平拆分 dataframe
def numpy_split_pd(df, split_num):
# 使用numpy拆分DataFrame 把索引均分 均分后再用索引拆分DataFrame
lst_index = list(map(lambda a: a.tolist(), numpy.array_split(df.index.tolist(), split_num)))
for idx in lst_index:
df_split = df.iloc[idx[0]: idx[-1] + 1]
yield df_split
传入dataframe,和需要拆分的个数 可以平均拆分dataframe
使用:
res = numpy_split_pd(df_data, 10)
for i in res:
print(i)
为者常成,行者常至
自由转载-非商用-非衍生-保持署名(创意共享3.0许可证)