Python:读写CSV文件

简介:CSV (Comma Separated Values) 格式是电子表格和数据库中最常见的输入、输出文件格式。又称逗号分隔值(Comma-Separated Values,CSV,有时也称为字符分隔值,因为分隔字符也可以不是逗号),其文件以纯文本形式存储表格数据(数字和文本)。纯文本意味着该文件是一个字符序列,不含必须像二进制数字那样被解读的数据。

数据较复杂的场景更推荐读写excel。

参考:Python:xlrd和xlwt模块操作Excel表格

特定规则:

开头是不留空,以行为单位。
可含或不含列名,含列名则居文件第一行。
一行数据不跨行,无空行。
以半角逗号(即,)作分隔符,列为空也要表达其存在。
列内容如存在半角引号(即"),替换成半角双引号("")转义,即用半角引号(即"")将该字段值包含起来。
文件读写时引号,逗号操作规则互逆。
内码格式不限,可为 ASCII、Unicode 或者其他。
不支持数字
不支持特殊字符

读取范例1:使用csv模块

# -*- coding: UTF-8 -*-
import csv
file = "demo.csv"
with open(file, "r")as f:
    f_csv = csv.reader(f)
    for row in f_csv:
        print(row)

读取范例2:使用pandas模块,若内容存在中文需要增加encoding=“gbk”

import pandas as pd

file = "demo.csv"
data = pd.read_csv(file, encoding="gbk")
print(data.head(2))  # 打印前2行
print("*" * 100)
print(data.columns)  # 返回全部列名
print("*" * 100)
print(data.shape)  # 返回csv文件形状  例如(4, 3) => 4行3列
print("*" * 100)
print(data.loc[1:2])  # 打印第1到2行

写入范例1:请注意,会造成换行。

# -*- coding: UTF-8 -*-
import csv

csvFile = open("test.csv", "w")
writer = csv.writer(csvFile)

# 写入列名。
writer.writerow(["id", "name", "age"])

row1 = ["1001", "Tom", 30]
row2 = ["1002", "David", 25]
row3 = ["1003", "Lily", 20]
row4 = ["1004", "韩梅梅", 15]

writer.writerows([row1, row2, row3, row4])
csvFile.close()

Python:读写CSV文件_第1张图片

写入范例2:使用newline=“”来解决换行问题

# -*- coding: UTF-8 -*-
import csv

import csv

headers = ['id', 'name', 'age']

row1 = {"id": "1001", "name": "Tom", "age": 30}
row2 = {"id": "1002", "name": "Lily", "age": 35}
row3 = {"id": "1003", "name": "李雷", "age": 40}
row4 = {"id": "1004", "name": "韩梅梅", "age": 45}

rows = [row1, row2, row3, row4]

with open('test2.csv', 'w', newline='')as f:
    f_csv = csv.DictWriter(f, headers)
    f_csv.writeheader()
    f_csv.writerows(rows)

Python:读写CSV文件_第2张图片
微信公众号:玩转测试开发
欢迎关注,共同进步,谢谢!

在这里插入图片描述

你可能感兴趣的:(Python学习,python,开发语言,后端)