在Python中使用函数式编程列表到字典

问题描述 投票:0回答:1

我想使用 python 中的函数式编程,使用规则

x if x odd; (x*2, x*3) if x is even
将列表转换为字典。

示例

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.x list dictionary lambda functional-programming
1个回答
1
投票

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

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}
© www.soinside.com 2019 - 2024. All rights reserved.