import codecs
|
import csv
|
from openpyxl import load_workbook
|
|
|
def read_csv(file_name):
|
data_list = list()
|
with codecs.open(file_name, encoding="utf-8") as f:
|
for row in csv.DictReader(f, skipinitialspace=True):
|
data_list.append(row)
|
|
return data_list
|
|
|
def read_excel(file_name, sheet_name):
|
# 打开Excel文件，加载workbook
|
workbook = load_workbook(filename=file_name)
|
# 根据名称获取worksheet
|
worksheet = workbook.get_sheet_by_name(sheet_name)
|
# 获取所有数据
|
rows = worksheet.rows
|
|
# 将第一行数据作为字典的键
|
header = []
|
for cell in next(rows):
|
header.append(cell.value)
|
|
# 从第二行起遍历每一行数据，并存储到字典中
|
data_list = []
|
for row in rows:
|
data = {}
|
for i, cell in enumerate(row):
|
data[header[i]] = cell.value
|
data_list.append(data)
|
|
return data_list
|
