列表理解多个列表

list comprehension for multiple lists

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

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]

注意:输出必须是三个单独的列表,就像循环中的列表一样。

试试 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)

使用列表理解的最简单方法

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)

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