操作很簡單,我都是官方文檔找的,寫到CSDN上算是筆記, 順便說一句csdn的markdown比印象筆記友好多了,寫着也舒服一些,以後筆記就寫在CSDN吧。 也給大家一個參考,具體的操作還是移步官方文檔,文檔地址自己找?
Excel文件操作 openpyxl庫的使用
稍微提一下,操作excel的庫很多,有說xlwt的, 有說pandas的, 我找的openpyxl,操作簡單方便些
import os
import openpyxl # 需要安裝 pip install openpyxl
def write_to_xlsx(file_name, sheet_name, write_data):
"""
追加保存文件到 excel文件
:param file_name: 文件名 str(不用加後綴)
:param sheet_name: 操作頁名稱
:param write_data: 保存數據 list類型 追加保存
:return:
"""
if not os.path.exists("./{}.xlsx".format(file_name)):
# 判斷是否由此文件 沒有就建立Workbook對象
wb = openpyxl.Workbook()
else:
# 有此文件 加載源文件
wb = openpyxl.load_workbook("./{}.xlsx".format(file_name))
# 獲取所有的sheet名稱
sheet_names = wb.sheetnames
if sheet_name in sheet_names:
# 判斷傳入 sheet_name名稱是否已存在,存在就切換
ws = wb[sheet_name]
else:
# 不存在就新建
ws = wb.create_sheet(sheet_name)
# 添加數據 list類型
ws.append(write_data)
# 保存數據
wb.save("./{}.xlsx".format(file_name))
if __name__ == '__main__':
# 測試
write_to_xlsx("abc", "bbc", ["123", "aaaa", "zzzz"])
write_to_xlsx("abc", "bbc", ["222", "ccca", "zzzz"])
write_to_xlsx("abc", "bbc", ["333", "zzza", "zzzz"])
CSV文件操作
csv模塊是python內置的,但是數據以逗號形式存儲, 據我所查不支持sheet分頁,但是我存數文件一般都用csv文件,方便,簡單
import csv
def save_file(file_name, save_data):
"""
保存文件
:param save_data: 保存數據 list類型
:return:
"""
with open("./{}.csv".format(file_name), "a+", newline="", encoding="utf-8") as data_csv:
csv_writer = csv.writer(data_csv, dialect=("excel",))
csv_writer.writerow(save_data)