修改代码,获取“绩效监测”表:ori_excel = openpyxl.load_workbook(ori_excel_path,data_only=True)#读取数据源表格 ori_sheet = ori_excel.active#获取第一个sheet
时间: 2024-01-22 14:21:08 浏览: 53
修改后的代码如下:
```python
ori_excel = openpyxl.load_workbook(ori_excel_path, data_only=True) # 读取数据源表格
ori_sheet = ori_excel["绩效监测"] # 获取“绩效监测”表
```
这样就可以直接获取到“绩效监测”表了。
相关问题
规范代码:读取数据源表格 ori_excel = openpyxl.load_workbook(ori_excel_path) ori_sheet = ori_excel.active rows = ori_sheet.max_row cols = ori_sheet.max_column 加载公式解析器 from transformers import AutoTokenizer tokenizer = AutoTokenizer.from_pretrained("microsoft/CodeGPT-small-py") 遍历每一行 for r in range(2, rows + 1): info = {} # 定义一个字典用来存放数据 mould_excel = openpyxl.load_workbook(mould_excel_path) mould_sheet = mould_excel.active # 遍历每一列 for c in range(1, cols + 1): # 如果标题栏和单元格内有内容 ori_cell_value = ori_sheet.cell(1, c).value cell_value = ori_sheet.cell(r, c).value if ori_cell_value and cell_value: # 获取单元格数据类型和值 data_type = ori_sheet.cell(r, c).data_type value = cell_value # 如果单元格内是函数公式,则运算后写入 if data_type == 'f': # 公式解析 tokens = [str(token.value) if token.type == Token.NUMBER else repr(token.value) for token in tokenizer.parse(value)] # 运算 result = calculate(tokens) # 写入结果 mould_sheet.cell(row=r, column=c).value = result else: # 直接写入值 mould_sheet.cell(row=r, column=c).value = value # 将计算结果写入到模板表格 if data_type == 'f': mould_sheet.cell(row=r, column=c, value=result) else: mould_sheet.cell(row=r, column=c, value=value) else: # 如果单元格内不是函数公式,则直接写入 info[ori_cell_value] = cell_value mould_sheet.cell(row=r, column=c, value=cell_value)
下面是规范化后的代码:
```python
import openpyxl
from transformers import AutoTokenizer
# 读取数据源表格
ori_excel = openpyxl.load_workbook(ori_excel_path)
ori_sheet = ori_excel.active
rows = ori_sheet.max_row
cols = ori_sheet.max_column
# 加载公式解析器
tokenizer = AutoTokenizer.from_pretrained("microsoft/CodeGPT-small-py")
# 遍历每一行
for r in range(2, rows + 1):
info = {}
# 定义一个字典用来存放数据
mould_excel = openpyxl.load_workbook(mould_excel_path)
mould_sheet = mould_excel.active
# 遍历每一列
for c in range(1, cols + 1):
# 如果标题栏和单元格内有内容
ori_cell_value = ori_sheet.cell(1, c).value
cell_value = ori_sheet.cell(r, c).value
if ori_cell_value and cell_value:
# 获取单元格数据类型和值
data_type = ori_sheet.cell(r, c).data_type
value = cell_value
# 如果单元格内是函数公式,则运算后写入
if data_type == 'f':
# 公式解析
tokens = [str(token.value) if token.type == Token.NUMBER else repr(token.value) for token in tokenizer.parse(value)]
# 运算
result = calculate(tokens)
# 写入结果
mould_sheet.cell(row=r, column=c).value = result
else:
# 直接写入值
mould_sheet.cell(row=r, column=c).value = value
# 将计算结果写入到模板表格
if data_type == 'f':
mould_sheet.cell(row=r, column=c, value=result)
else:
mould_sheet.cell(row=r, column=c, value=value)
else:
# 如果单元格内不是函数公式,则直接写入
info[ori_cell_value] = cell_value
mould_sheet.cell(row=r, column=c, value=cell_value)
```
优化代码: # 读取模板表格 mould_excel = openpyxl.load_workbook(mould_excel_path) mould_sheet = mould_excel.active # 遍历每一列 for c in range(1, cols + 1): # 如果标题栏和单元格内有内容 if ori_sheet.cell(1, c).value and ori_sheet.cell(r, c).value: # 如果单元格内是函数公式,则运算后写入 if ori_sheet.cell(r, c).data_type == 'f': formula = ori_sheet.cell(r, c).value tokens = AutoTokenizer(formula).parse() for token in tokens: if token.type == Token.NUMBER: token.value = str(token.value) elif token.type == Token.STRING: token.value = repr(token.value) result = tokens.to_tree().calculate()
可以尝试对代码进行如下优化:
1. 将数据类型判断和数据读取分离开来,这样可以减少重复的数据读取操作。
2. 使用更加简洁的语法,如使用列表推导式等。
3. 将公式解析和运算分离开来,这样可以更好地控制程序的流程。
下面是优化后的代码示例:
```
# 读取模板表格
mould_excel = openpyxl.load_workbook(mould_excel_path)
mould_sheet = mould_excel.active
# 遍历每一列
for c in range(1, cols + 1):
# 如果标题栏和单元格内有内容
if ori_sheet.cell(1, c).value and ori_sheet.cell(r, c).value:
# 获取单元格数据类型和值
data_type = ori_sheet.cell(r, c).data_type
value = ori_sheet.cell(r, c).value
# 如果单元格内是函数公式,则运算后写入
if data_type == 'f':
# 公式解析
tokens = AutoTokenizer(value).parse()
tokens = [(str(token.value) if token.type == Token.NUMBER else repr(token.value)) for token in tokens]
# 运算
result = calculate(tokens)
# 写入结果
mould_sheet.cell(row=r, column=c).value = result
else:
# 直接写入值
mould_sheet.cell(row=r, column=c).value = value
```
其中,`calculate` 函数用于对公式进行运算。
阅读全文