Python生成CSV文件操作方法

较近在做批量上传的功能，批量上传用到的是以逗号作为分隔符的csv文件。在测试阶段为了测试批量上传的功能，需要大量的数据进行测试。

这个时候我就想到可以利用Python生成csv进行测试。

首先，在PyCharm中选择File -> New -> Python File

指定文件名比如 generate_CSV_file.py

接下来就是写代码了：

import csv

import osdef generate_csv():

current_dir = os.path.abspath('.')

file_name = os.path.join(current_dir, "generated_800rows.csv") # open file in binary model to avoid generate empty line in Windows

csvfile = open(file_name, 'ab+') writer = csv.writer(csvfile, delimiter=',') header = ['Scheduled KPI Plan name', 'Description', 'Plan Type', 'Visibility', 'Location Filter',

'Resource Filter', 'Time Filter', 'Calculation Model', 'Alert_Rule_Group',

'Multi_Attribute_Filter', 'RECURRENCE_TYPE', 'Execute Every', 'Executed On',

'Start Scheduling Date/Time', 'End Scheduling Date/Time'] writer.writerow(header)

rows = []

one_row = [] index_of_rows = 0 while (index_of_rows < 400):

index_of_rows = index_of_rows + 1

one_row = ['KPITST_MAR5_' + str(index_of_rows), 'Test 20190619', 'Scheduled KPI Plan', 'Global', 'BUSER2_GLOBAL',

'ATP_RF_1', 'PF_HAN', 'ATP_KPI_CALMODEL', 'ATP_KPI_ALERTS_GP', '', 'Minutes', 5, '',

'2019-07-05T08:00:00.000Z',

'2019-12-13T08:00:00.000Z']

rows.append(one_row) while (index_of_rows < 800):

index_of_rows = index_of_rows + 1

one_row = ['KPITST_MAR5_' + str(index_of_rows), 'Test 20190619', 'Scheduled KPI Plan', 'Global', 'BU_CREATED_GLOBAL',

'BUSER2_GLOBAL', 'ATP_PAST1', 'ATP_KPI_CALMODEL', 'ATP_KPI_ALERTS_GP', '', 'Days', 1, '',

'2019-07-12T08:00:00.000Z',

'2019-10-13T08:00:00.000Z']

rows.append(one_row) writer.writerows(rows);

csvfile.close()

if __name__ == '__main__':

generate_csv()