【问题标题】:Convert for loop to list comprehension将 for 循环转换为列表理解
【发布时间】:2020-03-29 12:26:04
【问题描述】:

我正在尝试将我的代码转换为列表理解,但无法弄清楚

我的代码:

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']

【问题讨论】:

标签: python list dictionary for-loop list-comprehension


【解决方案1】:

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

>>> 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
     ]

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

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 2021-09-06
    相关资源
    最近更新 更多