读写csv文件

记录维维的每一篇Hello World

数据单行写入

import csv
import random
# 文件名
file_name = "file.csv"
# 生成随机数据
data = [random.random() for i in range(5)]
# 打开文件
file = open(file_name, 'w', newline="", encoding='u-f-8-sig')
# 写入数据
with file:
    writer = csv.writer(file)
    writer.writerow(data)

数据多行写入

import csv
import random
# 文件名
file_name = "file.csv"
# 设置行列数
row, col = 7, 5
# 生产随机数据
data = [[random.random() for c in range(col)] for r in range(row)]
# 打开文件
file = open(file_name, 'w', newline="", encoding='utf-8-sig')
# 写入数据
with file:
    writer = csv.writer(file)
    writer.writerows(data)

字典数据单行写入

import csv
# 文件名
file_name = "one.csv"
# 数据
data = {
    'id': '12345',
    'name': '小米'
}
# 打开文件
file = open(file_name, 'w', newline='', encoding='utf-8-sig')
# 写入数据
with file:
    writeDict = csv.DictWriter(file, fieldnames=['id', 'name'])
    writeDict.writeheader()
    writeDict.writerow(data)

字典数据多行写入

import csv
# 文件名
file_name = "one.csv"
# 数据
data1 = {
    'id': '12345',
    'name': '小米'
}
data2 = {
    'id': '67890',
    'name': '小花'
}
# 打开文件
file = open(file_name, 'w', newline='', encoding='utf-8-sig')
# 写入数据
with file:
    writeDict = csv.DictWriter(file, fieldnames=['id', 'name'])
    writeDict.writeheader()
    writeDict.writerows([data1, data2])

读取数据

import csv
# 文件名
file_name = "one.csv"
# 打开文件
file = open(file_name, 'r', newline='', encoding='utf-8-sig')
# 数据读取
with file:
    reader = csv.reader(file)
    for row in reader:
        print(','.join(row))

字典数据读取

import csv
#文件名
file_name = "one.csv"
# 打开文件
file = open(file_name, 'r', newline='', encoding='utf-8-sig')
# 读取数据
with file:
    reader = csv.DictReader(file)
    for row in reader:
        print(row)

注意事项

1、编码必须为utf-8-sig，不然中文会乱码

2、newline默认为换行，改成空字符避免隔一行

3、字典数据写入，字典中有多余键值会报错

python环境

电脑：Windows10

python：3.10.4