优化以下代码: def data_type_handle(data): """判断数据类型。0--三层嵌套列表,1--双层嵌套列表, 2--列表, 3--字典列表(MySQL类查询结果), 4--非列表""" if isinstance(data, list): if all(isinstance(d, list) for d in data): if all(isinstance(k, list) for d in data for k in d): return 0 elif all(isinstance(k, dict) for d in data for k in d): return 3 return 1 elif all(isinstance(d, dict) for d in data): return 3 return 2 return
时间: 2024-02-14 11:25:48 浏览: 52
DE-KELM的差分算法优化核极限学习机故障诊断,优化前后对比(Matlab完整源码和数据)
以下是对代码进行优化的改进:
```python
def data_type_handle(data):
"""判断数据类型。0--三层嵌套列表,1--双层嵌套列表, 2--列表, 3--字典列表(MySQL类查询结果), 4--非列表"""
if isinstance(data, list):
if all(isinstance(d, (list, dict)) for d in data):
if all(isinstance(k, list) for d in data for k in d):
return 0
return 1
elif all(isinstance(d, dict) for d in data):
return 3
return 2
return 4
```
改进的主要点如下:
- 使用多个`isinstance()`函数的参数为元组,可以同时检查多种类型,减少代码重复。
- 将判断非列表的情况归为类型4,增加代码的可读性。
请注意,这只是一种优化方法,根据具体的需求和场景,可能还有其他的优化方案。
阅读全文