多个列表的列表理解

Question

是否可以将这样的循环转换为列表理解？

lst_a = []
lst_b = []
lst_c = []
    
test_dict = {'a':1, 'b':2, 'c':3, 'd':4} 

for k, v in test_dict.items():
    lst_a.append(k)
    lst_b.append(k.upper())
    lst_c.append((k+k))

例如，对于这样的事情：

lst_a, lst_b, lst_c = [list comprehension logic here for the above loop]

注意： output 必须是三个单独的列表，就像循环中的列表一样。

Answer 1

尝试使用zip() ：

test_dict = {'a': 1, 'b': 2, 'c': 3, 'd': 4}

lst_a, lst_b, lst_c = zip(*[(k, k.upper(), k+k) for k in test_dict])

# Convert them to list
# lst_a, lst_b, lst_c = map(list, zip(*[(k, k.upper(), k+k) for k in test_dict)])) 

# Iterate this dict three times.
# lst_a, lst_b, lst_c = ([k for k in test_dict], [k.upper() for k in test_dict], [k + k for k in test_dict])
print(lst_a,lst_b,lst_c)

Answer 2

使用列表推导的最简单方法

lst_a = []
lst_b = []
lst_c = []
    
test_dict = {'a':1, 'b':2, 'c':3, 'd':4} 

[[lst_a.append(k), lst_b.append(k.upper()), lst_c.append(k+k)] for k in test_dict]

print(lst_a, lst_b, lst_c)

output - ['a', 'b', 'c', 'd'] ['A', 'B', 'C', 'D'] ['aa', 'bb', 'cc', 'dd']