首先引用
import csv
import codecs
import random
写入函数
def write_csv(file_name, datas): # file_name为写入CSV文件的路径,datas为要写入数据列表
file_csv = codecs.open(file_name,'w+','utf-8') # 追加
writer = csv.writer(file_csv, delimiter=' ', quotechar=' ', quoting=csv.QUOTE_MINIMAL)
writer.writerow(datas)
print("保存文件成功,处理结束")
读入函数:
def read_csv(file_name):
f = open(file_name, 'r')
csvreader = csv.reader(f)
final_list = list(csvreader)
data = final_list[0][0].split(' ') # 分割为列表
data = [int(x) for x in data] # 转换为数字
return data
测试:
order_list_1w = 'quickSortData\Order_List.csv'
disorder_list_1w = 'quickSortData\Disorder_List.csv' # 数据量为1w的无序数列
datas = [i for i in range(10000)]
for i in datas:
print(i, end=" ")
print('\n' + "---------")
write_csv(order_list_1w, datas)
random.shuffle(datas) # 打乱列表
# write_csv(disorder_list_1w, datas) # 存储无序的1w列表
for i in datas:
print(i, end=" ")
print('\n' + "---------")
结果: