列表数据写入csv和读取csv

列表数据写入csv

列表数据案例

• 数据header：test_header = ['num1','num2','num3']
• 数据内容：test_data = [[1,2,3],[3,2,1]]
• 写入csv文件

with open(basePath + "test.csv",'w',encoding='utf-8',newline="") as f: # newline=""表示写入行不要空行
    writer = csv.writer(f)
    writer.writerow(test_header)
    writer.writerows(test_data)

打开csv文件，可以看到

写入csv完成！

读取csv数据，形成列表

主要目标：

• 以列表形式获取列名
• 以列表形式获取数据内容

import csv
test_header = []
test_data = []
basePath = "F:\\test2\\"
with open(basePath + "test.csv") as f:
    reader = csv.reader(f)
    t = 0
    for row in reader:
        if t== 0:
            test_header.extend(row)
            t =1
        else:
            test_data.append(row)

结果显示：