优化下面代码: def get_case_data(self): self.get_keys() # self.keys # 键值对的所有键 while True: row = self.get_next_row() # 键值对的所有值 # print("row", row) if not row: break null_list = [] # 所有空数据的下标集合 for i, v in enumerate(row): # print(i,v) # i 是index v 是value if not v: null_list.append(i) keys = [] values = [] for i in range(len(row)): if i not in null_list: keys.append(self.keys[i]) values.append(row[i]) # print(keys) # print(values) # 快捷方式 组合两个列表成为 dict的方式 caps = dict(zip(keys, values)) # print("kwargs", caps) action_value, *kwargs_values = caps.values() action_key, *kwargs_keys = caps.keys() # print("kwargs_keys:", kwargs_keys) # print("kwargs_values:", kwargs_values) kwargs = dict(zip(kwargs_keys, kwargs_values))
时间: 2024-03-17 10:44:25 浏览: 56
这段代码可以优化的部分有:
1. 使用列表推导式代替for循环
2. 简化获取非空键值对的逻辑,使用zip()函数直接过滤
3. 将获取非空键值对的逻辑放到一个单独的方法中,提高代码的可读性
下面是优化后的代码:
```
def get_case_data(self):
self.get_keys()
while True:
row = self.get_next_row()
if not row:
break
null_list = [i for i, v in enumerate(row) if not v]
keys, values = self.get_non_null_items(self.keys, row, null_list)
caps = dict(zip(keys, values))
action_value, *kwargs_values = caps.values()
action_key, *kwargs_keys = caps.keys()
kwargs = dict(zip(kwargs_keys, kwargs_values))
```
同时,需要添加一个新的方法get_non_null_items:
```
def get_non_null_items(self, keys, values, null_list):
return [k for i, k in enumerate(keys) if i not in null_list], [v for i, v in enumerate(values) if i not in null_list]
```
阅读全文