首页 > 解决方案 > 在 python 中使用函数式编程列出字典

问题描述

我想x if x odd; (x*2, x*3) if x is even使用 python 中的函数式编程使用规则将列表转换为字典。

示例list([1, 2, 3])_dict({1:1, 4:2, 6:2, 3:3})

我有一个将列表转换为的代码{1: 1, (4, 2): (6, 2), 3: 3}

from itertools import chain
print(
    dict(
        map(
            lambda elem:
            (lambda x: (x, x))(elem) if elem % 2 == 1
            else (lambda x: chain(
                ((x*2, x), (x*3, x))
            ))(elem),
            list([1, 2, 3])
        )
    )
)

有人能弄清楚吗?

结果应该与此代码产生的相同

d = dict()
for x in [1, 2, 3]:
    if x % 2 == 1:
        d[x] = x
    else:
        d[x*2] = x
        d[x*3] = x

标签: python-3.xlistdictionarylambdafunctional-programming

解决方案


您可以简单地使用字典理解:

lst = [1, 2, 3]

out = {k: v for v in lst for k in ((v,) if v % 2 else (v * 2, v * 3))}
print(out)

印刷:

{1: 1, 4: 2, 6: 2, 3: 3}

编辑:没有for循环或任何库:

d = (
    dict(map(lambda x: (x, x), filter(lambda x: x % 2, lst)))
    | dict(map(lambda x: (x * 2, x), filter(lambda x: x % 2 == 0, lst)))
    | dict(map(lambda x: (x * 3, x), filter(lambda x: x % 2 == 0, lst)))
)
print(d)

印刷:

{1: 1, 3: 3, 4: 2, 6: 2}

推荐阅读