python处理csv文件

最新推荐文章于 2024-09-13 09:00:00 发布

原创最新推荐文章于 2024-09-13 09:00:00 发布 · 270 阅读

0 ·

CC 4.0 BY-SA版权

python 同时被 2 个专栏收录

13 篇文章

订阅专栏

爬虫

6 篇文章

订阅专栏

本文介绍了如何使用Python读取和写入CSV文件。对于读取，涵盖了列表和字典两种方式，通过迭代器处理表头。在写入部分，同样展示了列表和字典方法，并强调了newline参数的设置。

Python读取csv文件

1、使用列表的方式读取

import csv

with open('1.csv', 'r', encoding='utf-8') as fp:
    # reader是一个迭代器
    reader = csv.reader(fp)
    next(reader)
    # next(reader)
    for x in reader:
        print(x)

reader是一个迭代器，因此使用next方法，可以去掉表头

2、使用字典的方式读取

with open('1.csv', 'r') as fp:
    # 使用DictReader创建的对象，不会包行标题那行的数据
    reader = csv.DictReader(fp)
    #reader是一个迭代器，遍历这个迭代器，返回的是一个字典
    for x in reader:
        print(x)
        print(x.get('姓名'))
结果
OrderedDict([('姓名', '小刘'), ('年龄', '25'), ('性别', '男')])
小刘
OrderedDict([('姓名', '小罗'), ('年龄', '26'), ('性别', '女')])
小罗

Python写入csv文件

1、使用列表的方式写入

header = ['姓名', '年龄', '性别']
contents = [('小刘', '25', '男'), ('小罗', '26', '女')]
with open('2.csv', 'w', newline='') as fp:
    writer = csv.writer(fp)
    writer.writerow(header)
    writer.writerows(contents)

newline=’’
2、使用字典的方式写入

header = ['姓名', '年龄', '性别']
contents = [('小刘', '25', '男'), ('小罗', '26', '女')]
with open('2.csv', 'w', newline='', encoding='utf-8') as fp:
    writer = csv.writer(fp)
    writer.writerow(header)
    writer.writerow(header)
    writer.writerows(contents)

header = ['姓名', '年龄', '性别']
values = [
    {'姓名': '小刘', '年龄': '25', '性别': '男'},
    {'姓名': '小罗', '年龄': '26', '性别': '女'},
]
with open('3.csv', 'w', encoding='utf-8', newline='') as fp:
    writer = csv.DictWriter(fp, header)
    # 写入表头数据的时候，需要调用writeheader方法
    writer.writeheader()
    writer.writerows(values)