Python工具开发实践-csv2excel
Python学习有一段时间了，今天来尝试编写一个程序来实现csv文件转换为excel文件的功能。
首先分析需求，将需求分解为如下几个步骤：
1、获取文件名称；
2、打开csv，可以使用pandas；
3、保存为excel，可以使用pandas
4、可以对程序处理时间进行计时，会用到time模块
对于第一个步骤，如果直接告诉程序文件的名称的话，程序的可移植性太差了，不够灵活，每次都要修改程序，不如升华一下，告诉程序一个目录，程序自动获取目录下的所有csv文件，这样就方便多了。
修改需求如下：
1、告知一个目录，程序自动获取目录下所有的csv文件名称，会用到os模块；
2、打开csv，可以使用pandas；
3、保存为excel，可以使用pandas
4、可以对程序处理时间进行计时，会用到time模块
每一个小功能可以编写成一个函数，然后在主函数main里调用。
第一个函数：
def get_allfile(): """ 获取文件夹下所有csv文件 """ # 用户交互界面，提示用户输出 path = input('Please input your file path: ') # 列出当前目录下的所有文件，返回一个列表 files = os.listdir(path) # 初始化一个空列表，用来存储csv文件列表 file_list = [] for f in files: # 遍历文件列表中的文件，判断是否为csv文件 if os.path.splitext(f)[1] == '.csv': # 如果是，则将文件的绝对目录放到csv文件列表中，否则跳过 file_list.append(path + '\\' + f) else: pass # 返回csv文件列表 return file_list第二个和第三个功能都用pandas，可以放到一个函数：
def csv2excel(f): """ 打开csv,另存为excel文件 """ # 打开csv df = pd.DataFrame(pd.read_csv(f, header=0, encoding='gbk')) # 另存为excel，文件绝对目录+csv文件名称+后缀.xlsx，去掉索引列 df.to_excel(os.path.splitext(os.path.dirname(f) + '\\' + os.path.basename(f))[0] + '.xlsx', index=None)用time模块计时比较简单，放到主函数main()就好了
def main(): # 获取文件列表 file_list = get_allfile() # 计时开始 start_time = time.time() # csv转excel处理 for file in file_list: csv2excel(file) # 计时结束 end_time = time.time() # 计算程序处理耗时 cost_time = end_time - start_time # 用户交互界面，输出运行时间，并保留2为小数 print('程序运行时间：{}秒'.format(float('%.2f' % cost_time))) # 为防止程序运行完自动跳出，将页面保留60秒 time.sleep(60)至此，各个函数都写好了，将各个函数放到一个py文件就可以了，记着首先要导入使用到的模块
import pandas as pdimport osimport time另外，下面的最关键的两行
if __name__ == '__main__': main()
本文参与 腾讯云自媒体分享计划，分享自微信公众号。
原始发表：2019-04-21，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读