python使用pandas按照行数分割表格_Python

python使用pandas按照行数分割表格

2021-12-20 00:06羊草 Python

本文主要介绍了python使用pandas按照行数分割表格，文中通过示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下

问题

一张excel表格，大概1万行，需要录入系统
系统每次最多只能录入500行表格数据，一旦超过500行，就会录入失败
需要把1万行的数据按照500行分割，形成20个表格，这样才能录入系统

思路

使用pandas得到总行数，比如10002行，分割表格的时候，要保留一行表头
第一张表，是1-500行，第二张表是 501-1000，以此类推
最后一张表应该是1000-10002行，生成的表格数量是10000/500+1,21张
生成的表格按照顺序保存到一个目录中
写一个函数，可以按照任意指定的分割数量进行分割。

代码实现

				?

									#按行数分割表格函数

									#问题

									#1.如果有有一个十万行表格，要录入系统，但是系统每次最多只能录入500行？

									#解决问题：

									#1.按照指定的行数分割表格

									#2.分割出来的表格按照序号命名

									import pandas as pd

									import os

									def splitexcel(file,num):

									    file_dir='result'   #创建目录

									    if os.path.isdir(file_dir):

									        os.rmdir(file_dir)

									    else:

									        os.mkdir(file_dir)

									    n = 1

									    row_list = []

									    df = pd.dataframe(pd.read_excel(file, sheet_name=0))

									    row_num = int(df.shape[0])  # 获取行数

									    if num >= row_num:  #如果分割行数大于总行数，报错

									        raise exception('too much!!')

									    try:

									        for i in list(range(num,row_num,num)):

									            row_list.append(i)

									        row_list.append(row_num)  # 得到完整列表

									    except exception as e:

									        print (e)

									    (name,ext)=os.path.splitext(file)  #获取文件名

									    for m in row_list:

									        filename=os.path.join(file_dir,name+'-' + str(n) + '.xlsx')

									        if m <row_num:

									            df_handle=df.iloc[m-num:m] #获取n行之前

									            print (df_handle)

									            df_handle.to_excel(filename , sheet_name='sheet1',index=false)

									        elif m == int(row_num):

									            remainder=int(int(row_num)%num) #余数

									            df_handle=df.iloc[m-remainder:m] #获取最后不能整除的行

									            df_handle.to_excel(filename , sheet_name='sheet1', index=false)

									        n = n + 1

									if __name__=='__main__':

									    file= 'result.xls'

									    splitexcel(file,num=10)