1
"""
2
===========================
3
@Time : 2023/2/1 11:19
4
@File : csv_to_excel.py
5
@Software: PyCharm
6
@Platform: Win10
7
@Author : DataShare
8
===========================
9
"""
10
import pandas as pd
11
import argparse
12

13
if __name__ == '__main__':
14
    parser = argparse.ArgumentParser(description='csv_to_excel')
15
    parser.add_argument('--input_file', '-in', type=str, required=True, help='csv文件')
16
    parser.add_argument('--output_file', '-out', type=str, required=False, default=None, help='excel文件')
17
    args = parser.parse_args()
18

19
    data = pd.read_csv(args.input_file, sep=',', dtype='str', quotechar='"', header=0)
20

21
    print('csv文件行数为：', len(data))  # 判断数据行数是否一致，防止不可见字符，例如：回车 等
22

23
    if args.output_file is not None:
24
        if args.output_file.endswith('.xlsx'):
25
            output_file_converted = args.output_file
26
        else:
27
            output_file_converted = args.output_file + '.xlsx'
28
    else:
29
        output_file_converted = args.input_file.split('.csv')[0] + '.xlsx'
30

31
    # 这是由于Excel单个工作表限制URL类型数据量为65530,超出的部分会被舍弃
32
    # 只要将strings_to_urls自动转换功能关闭就好了
33
    writer = pd.ExcelWriter(output_file_converted, engine='xlsxwriter',
34
                            engine_kwargs={'options': {'strings_to_urls': False}})
35
    data.to_excel(writer, index=False)
36
    writer.close()
37

38
    print('数据转换完成')