优化代码def lifecycle_visual(esn, data_pack_module, params, switches): """ 数据可视化调用函数 Life cycle visualization :param esn: :param data_pack_module: :param params: :param switches: :return: """ module_path = params['module_paths'][esn] n_volt_probe = data_pack_module['n_volt_probe'] n_temp_probe = data_pack_module['n_temp_probe'] if switches['lifecycle_visual']: # 路径创建 visual_path = os.path.join(module_path, 'visualization') if not os.path.exists(visual_path): os.makedirs(visual_path) fig_save_name = os.path.join(visual_path, "%s.png" % esn) lifecycle(data_pack_module['data'], esn, n_volt_probe, n_temp_probe, [], [], fig_save_name, dpi=100) print("\033[0;31;42m SUCCESS: module全生命周期可视化 Done. \033[0m") # 重点信息【模块分析完成】:红色字体绿色背景
时间: 2024-01-20 08:04:08 浏览: 81
这段代码看起来已经很简洁了,但是还有一些可以优化的地方。比如可以使用 `os.makedirs(visual_path, exist_ok=True)` 来代替 `if not os.path.exists(visual_path): os.makedirs(visual_path)`,这样可以减少代码行数。同时,可以将 `fig_save_name` 的赋值放到 `lifecycle` 函数的参数列表中,这样可以更直观地看出该变量的作用。最后,可以使用 f-string 来格式化字符串,使代码更简洁易读。
重构后的代码如下:
```python
def lifecycle_visual(esn, data_pack_module, params, switches):
"""
数据可视化调用函数
Life cycle visualization
:param esn:
:param data_pack_module:
:param params:
:param switches:
:return:
"""
module_path = params['module_paths'][esn]
n_volt_probe = data_pack_module['n_volt_probe']
n_temp_probe = data_pack_module['n_temp_probe']
if switches['lifecycle_visual']:
# 路径创建
visual_path = os.path.join(module_path, 'visualization')
os.makedirs(visual_path, exist_ok=True)
fig_save_name = f"{visual_path}/{esn}.png"
lifecycle(data_pack_module['data'], esn, n_volt_probe, n_temp_probe, [], [], fig_save_name, dpi=100)
print("\033[0;31;42m SUCCESS: module全生命周期可视化 Done. \033[0m") # 重点信息【模块分析完成】:红色字体绿色背景
```
阅读全文