使用Python将Excel文件内容写入YAML文件的过程包括以下几个步骤:
- 使用
openpyxl
模块读取Excel文件; - 将Excel表格数据转换成Python对象;
- 将Python对象转换成YAML字符串;
- 将YAML字符串写入YAML文件。
以下为代码示例:
import openpyxl
import yaml
xlsx_name = "data.xlsx"
# 打开Excel文件
wb = openpyxl.load_workbook(xlsx_name)
sheet = wb.active
# 读取Excel表格数据并存储为Python对象
data = {'data': []}
for row in sheet.iter_rows(min_row=2, values_only=True):
item = {'name': row[0], 'age': row[1], 'email': row[2]}
data['data'].append(item)
for i in data['data']:
print(i)
# 将Python对象转换成YAML字符串
yaml_data = yaml.dump(data)
# 将YAML字符串写入YAML文件
with open('data.yaml', 'w') as file:
file.write(yaml_data)
# 读取yaml文件内容
with open('data.yaml', 'r') as file:
dataR = yaml.load(file, Loader=yaml.FullLoader)
print(dataR)
'''
{'name': 's1', 'age': 22, 'email': '[email protected]'}
{'name': 's2', 'age': 23, 'email': '[email protected]'}
{'name': '宋', 'age': 24, 'email': '[email protected]'}
{'data': [{'age': 22, 'email': '[email protected]', 'name': 's1'}, {'age': 23, 'email': '[email protected]', 'name': 's2'}, {'age': 24, 'email': '[email protected]', 'name': '宋'}]}
'''
在以上代码中,首先使用openpyxl
模块打开Excel文件,并读取其中的表格数据。然后,将表格数据存储为Python对象,并转换成YAML格式的字符串。最后,将YAML字符串写入到YAML文件中。
需要注意的是,本例中使用了iter_rows()
函数来迭代读取Excel表格数据,其中参数min_row=2
表示从第二行开始读取,因为第一行通常是表头而非数据。