将Python代码转换为列表理解

时间:2019-12-04 16:23:30

标签: python list-comprehension

我尝试将我的代码转换为列表理解,但无法解决

我的代码:

dicts = [
    {"p": 1, "p": 2, "olp": 3, "r": 4}, 
    {"s": 1, "o": 2, "e": 3, "cs": 4}, 
    {"vc":1, "txc":2, "faw":3, "e": 4}
]

l = []

for d in dicts:
    for k, v in sorted(d.items(), key=lambda kv: kv[1]):
        if v % 4 == 0:
            if len(k) > 1:
                l.append(k[1])
            else:
                l.append(k)
        elif v % 2 == 0:
            if len(k) > 2:
                l.append(k[0])
            else:
                l.append(k)
print(''.join(l))

我已经尝试过了,但是没有用,您能帮我吗?

l = [
    k[0] if len(k)>2 else k elif v % 2 == 0 
    k[1] if len(k)>1 else k if k % 4 == 0  
    for k,v in sorted(d.items(), key=lambda kv: kv[1]) for d in dicts
]

l的预期输出:

['p', 'r', 'o', 's', 't', 'e']

1 个答案:

答案 0 :(得分:4)

重新格式化为单行列表理解,代码如下:

>>> l = [k[1] if v%4 == 0 and len(k) > 1 else k[0] if len(k) > 2 else k for d in dicts for k,v in sorted(d.items(), key=lambda kv: kv[1]) if v%2 == 0]
>>> l
['p', 'r', 'o', 's', 't', 'e']

或者格式化得更好:

l = [
     k[1] if v%4 == 0 and len(k) > 1
     else k[0] if len(k) > 2 
     else k
     for d in dicts
     for k,v in sorted(d.items(), key=lambda kv: kv[1])
     if v%2 == 0
     ]

据我所知,您也不需要在sorted()上致电d.items(),因此除非您的MWE外部不需要,否则您可以忽略它。