csv、excel等文件相关操作

分享自己关于学习文件操作的相关内容。

open函数

之前一直是这样写的：

f = open('./data.txt',mode='a',encoding='utf8')
f.write("hello")
f.writelines(['xiaoming','xiaowang'])
f.write('小明与小汪')
# 文件读取
lines = f.read()
print(lines)
lines = f.readlines()
print(lines)
f.close()

这样写每次都要关闭，不是很简洁，所以我就改成了下面一种方式：

with open('./data.txt',mode='a',encoding='utf8') as f:
    # 文件写入
    f.write("helloworld123")
    f.writelines(['xiaoming','xiaowang'])
    f.write('小明与小汪')
    lines = f.read()
    lines = f.readlines()
    print(lines)

这样写每次操作完会自动关闭，也省了自己写代码关闭，也可以避免忘记关闭的烦恼，看着也更简洁。

csv 操作

写入文件：
其中的newline=’’，是新起一行的意思。

import csv

with open('./data.csv',mode='w',newline='',encoding='utf8') as file:
    # 写入文件类
    fw = csv.writer(file)
    fw.writerow(["aaaaaaa",'username'])
    fw.writerow(['helloworld','xiaoming'])

    fw.writerows([('heiheihei','wawawa'),('xiaoming','xiaowang')])

读取文件：

import csv

with open('./data.csv',mode='r',newline='',encoding='utf8') as file:
    rd = csv.reader(file)
    #一行一行的读取
    for line in rd:
        print(line)

dict方式写入数据:

import csv

with open('./data.csv',mode='w',newline='',encoding='utf8') as file:
    field_names = ['username','passwd']
     # 制定表头
    cdw = csv.DictWriter(file,fieldnames=field_names)
     # 写入表头
    cdw.writeheader()
    # 写入内容
    cdw.writerow({'username':'xiaoming','passwd':'123456'})
    cdw.writerow({'username': 'xiaowang', 'passwd': '654321'})

dict读取数据

import csv

with open('./data.csv',mode='r',newline='',encoding='utf8') as file:
    dr = csv.DictReader(file)
    for line in dr:
        print(line)

Excel文件解析

写入：

from openpyxl import Workbook

wb = Workbook()
worksheet = wb.create_sheet('data')
worksheet.append(['username','password'])
for i in range(10):
    worksheet.append([f'testuser{i}','123456'])
# 保存文件
wb.save('./data.xlsx')

读取：

from openpyxl import load_workbook
from openpyxl.worksheet.worksheet import Worksheet

wb = load_workbook('data.xlsx')
# print(wb.sheetnames)

ws:Worksheet = wb['data']
test_data = []
for row in ws.values:
    test_data.append(row)

print(test_data)

json模块

转字符串：

import  json

jstr = json.dumps(['hi',{'username':'xiaowang'}])
# 转换为字符串
print(jstr, type(jstr))
# ["hi", {"username": "xiaowang"}] <class 'str'>

with open('./data.txt',mode='w') as file:
    file.write(jstr)

解析为Python格式：

import  json

ls = json.loads('["hi", {"username": "xiaowang"}]')
print(ls,type(ls))

写入json文件:

import  json

test_data = {
    "id":1,
    "test_steps":[
        "1.打开",
        "2.输入信息"
    ]
}

with open('./data.json',mode='w',encoding='utf8') as file:
    json.dump(test_data, file)

读取json文件：

import  json
with open('./data.json',encoding='utf8') as file:
    data = json.load(file)
    print(data,type(data)) 

不得不感叹互联网上的资源是真的多，啥资源都有，真的是很不错很不错！
继续学习，继续努力，冲就对啦！