在 Python 2 中将多个列表和字典解包为函数参数

NoN*_*ble 5 python python-2.x python-3.x

可能的相关问题(我搜索了相同的问题,但没有找到)


https://docs.python.org/3/tutorial/controlflow.html#unpacking-argument-lists 中所述,Python 提供了一种使用星号将参数解包为函数的便捷方法

>>> list(range(3, 6))            # normal call with separate arguments
[3, 4, 5]
>>> args = [3, 6]
>>> list(range(*args))            # call with arguments unpacked from a list
[3, 4, 5]
Run Code Online (Sandbox Code Playgroud)

在我的代码中,我正在调用这样的函数:

def func(*args):
    for arg in args:
        print(arg)
Run Code Online (Sandbox Code Playgroud)

在 Python 3 中,我这样称呼它:

a = [1, 2, 3]
b = [4, 5, 6]
c = [7, 8, 9]

func(*a, *b, *c)
Run Code Online (Sandbox Code Playgroud)

哪些输出

1 2 3 4 5 6 7 8 9
Run Code Online (Sandbox Code Playgroud)

然而,在 Python 2 中,我遇到了一个异常:

>>> func(*a, *b, *c)
  File "<stdin>", line 1
    func(*a, *b, *c)
             ^
SyntaxError: invalid syntax
>>> 
Run Code Online (Sandbox Code Playgroud)

似乎 Python 2 无法处理多个列表的解包。有没有比这更好、更干净的方法来做到这一点

func(a[0], a[1], a[2], b[0], b[1], b[2], ...)
Run Code Online (Sandbox Code Playgroud)

我的第一个想法是我可以将列表连接成一个列表并解压缩它,但我想知道是否有更好的解决方案(或我不明白的东西)。

d = a + b + c
func(*d)
Run Code Online (Sandbox Code Playgroud)

Chr*_*ert 11

建议:迁移到 Python 3
自 2020 年 1 月 1 日起,Python 软件基金会不再支持 Python 编程语言的 2.x 分支。

解包列表并传递给 *args

Python 3 解决方案

def func(*args):
    for arg in args:
        print(arg)

a = [1, 2, 3]
b = [4, 5, 6]
c = [7, 8, 9]

func(*a, *b, *c)
Run Code Online (Sandbox Code Playgroud)

Python 2 解决方案

如果需要 Python 2,itertools.chain则将提供解决方法:

import itertools


def func(*args):
    for arg in args:
        print(arg)

a = [1, 2, 3]
b = [4, 5, 6]
c = [7, 8, 9]

func(*itertools.chain(a, b, c))
Run Code Online (Sandbox Code Playgroud)

输出

1
2
3
4
5
6
7
8
9
Run Code Online (Sandbox Code Playgroud)

解压字典并传递给 **kwargs

Python 3 解决方案

1
2
3
4
5
6
7
8
9
Run Code Online (Sandbox Code Playgroud)

Python 2 解决方案

正如Elliot在评论中提到的,如果您需要解压多个字典并传递给kwargs,您可以使用以下内容:

def func(**args):
    for k, v in args.items():
        print(f"key: {k}, value: {v}")


a = {"1": "one", "2": "two", "3": "three"}
b = {"4": "four", "5": "five", "6": "six"}
c = {"7": "seven", "8": "eight", "9": "nine"}

func(**a, **b, **c)
Run Code Online (Sandbox Code Playgroud)