python操作CSV文件

python操作CSV文件总结

   总结一下python操作数据库，每次用都要百度查，还不如看自己总结哒~ 还能给自己增加访问量，嘿嘿嘿

读取csv

with open("test_write.csv", 'r') as r:
    csv_reader = csv.reader(r)
    next(csv_reader)   # 跳过，不操作首行标题
    for i in csv_reader:
        print(i)

上面是基本的读取csv流程，下面来说说关于读取的其他方法

csv.DictReader()

    以字典格式读取csv文件，得到的结果是带着每列标题的数据，方便后续使用这些数据，可以根据他的key得到他的value值

with open("test_write.csv", 'r') as r:
    csv_reader = csv.DictReader(r)
    next(csv_reader)   # 跳过，不操作首行标题
    for i in csv_reader:
        print(i)

写入CSV

import csv
from faker import Faker  # 一个生成随机值的库

fake = Faker("zh_CN")

"""
主要部分
打开test_write.csv文件，往里面写入两列数据("姓名"、"学号")
"""
import csv
from faker import Faker

fake = Faker("zh_CN")

"""
主要部分
打开"test_write.csv"文件，往里面写入两列数据“姓名”、“学号”
"""
with open("test_write.csv", 'w', newline='')  as r:
    csv_writer =csv.writer(r)
    csv_writer.writerow(["姓名", "学号"])
    for i in range(10):
        csv_writer.writerow([fake.name(), i])

生成的csv文件如下图：

打开文件一定要记得加 newline 参数，否则就会是下面这个样子，每写一行都会有一个空行