欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

Python -- CSV文件读写

程序员文章站 2022-03-25 17:24:51
非字典类型数据读写(reader、writer) delimiter 用于分隔字段的单字符,默认为 ',' 字典类型数据读写 fieldnames 参数 指定字典中值的顺序,csv文件中,第一行数据为键 extrasaction 参数 如果传递给 writerow() 方法的字典的某些键在 fiel ......

非字典类型数据读写(reader、writer)

  • delimiter
    • 用于分隔字段的单字符,默认为 ','
import csv

datas = [[1, 2, 3], ['纪', '宇'], ['xxx', 'yyy', 'zzz'], '字符串']

# 写
with open('test.csv', 'w+', newline='') as fr:
    writer = csv.writer(fr, delimiter='|')
    for data in datas:
        writer.writerow(data)
# 读
with open('test.csv', 'r') as fw:
    reader = csv.reader(fw)
    for row in reader:
        print(row)

Python -- CSV文件读写



字典类型数据读写

  • fieldnames 参数
    • 指定字典中值的顺序,csv文件中,第一行数据为键
  • extrasaction 参数
    • 如果传递给 writerow() 方法的字典的某些键在 fieldnames 中找不到
    • 其设置默认值 'raise',会引发 valueerror
    • 如果将其设置为 'ignore',则字典中的其他键值将被忽略
  • writeheader()
    • 写入一行字段名称,fieldnames 参数的值
    • 不写入键,则会以第一行的数据作为默认字段
import csv

datas = [
    {'title': '标题1', 'url': 'xxx1.com', 'pubtime': 'time1', 'author': '作者1'},
    {'title': '标题2', 'url': 'xxx2.com', 'pubtime': 'time2', 'author': '作者2'},
    {'title': '标题3', 'url': 'xxx3.com', 'pubtime': 'time3', 'author': '作者3'},
]

# 写
with open('test.csv', 'w+', newline='', encoding='utf-8') as fr:
    # 设置字段
    fieldnames = ['title', 'pubtime', 'url']
    writer = csv.dictwriter(fr, fieldnames=fieldnames, extrasaction='ignore')
    # 写入第一行的字段
    writer.writeheader()
    for data in datas:
        writer.writerow(data)

# 读
with open('test.csv', 'r', encoding='utf-8') as fw:
    reader = csv.dictreader(fw)
    for row in reader:
        # 返回的行是dict类型
        print(row['title'], row['url'], '\t--\t', row)

Python -- CSV文件读写