Python读写csv文件

头文件

import csv
import pandas as pd

建一个csv文件

# 设置csv文件的路径（包括csv文件名称及后缀）
csv_path = result_path + file_name + '.csv'
# 设置表头，表头可以为空
dataDic={'video_name':[],'imgname':[]}
# 将表头写入csv文件
df = pd.DataFrame(dataDic)
df.to_csv(csv_path)

写入每列对应的数据

dataDic={'video_name':[video_name],'imgname':[imgname]}
df = pd.DataFrame(dataDic)
# 多次写入，且不希望覆盖之前的数据用mode='a'
df.to_csv(csv_path,mode='a',header=False)
# 多次写入，且希望覆盖之前的数据用mode='w'
df.to_csv(csv_path,mode='w',header=False)

读取csv文件内的数据

方法1：

with open(csv_path, "r") as f:
    reader = csv.DictReader(f)
    # 按行读取csv文件内容
    for row in reader:
    	# 读取该行img_name列的内容，row为字典
        print(row['img_name'])
    f.close()

方法2：

with open(csv_path, "r") as f:
    reader = csv.reader(f)
    # 按行读取csv文件内容，row为列表
    for row in reader:
    	# 读取该行img_name列的内容
        print(row[0])
    f.close()