今天看到一个比较有意思的代码:
# 代码1
list(zip(*([1,2,3],[1,2,3],[1,2,3])))
# 输出
[(1, 1, 1), (2, 2, 2), (3, 3, 3)]
看这个的时候可能觉得没什么奇怪的,我们试着将上面代码中的*
去掉试试:
# 代码2
list(zip(([1,2,3],[1,2,3],[1,2,3])))
# 输出
[([1, 2, 3],), ([1, 2, 3],), ([1, 2, 3],)]
看出上面代码1和代码2的区别了吗?代码1和代码2的输出都是列表,列表中的元素也都是元组,只是代码1中的各个元组分别是由原始的3个列表[1,2,3], [1,2,3], [1,2,3]
中对应位置的元素组成的输出列表中的单个元组元素。而代码2输出中的各个元素确是个嵌套了原始输入列表的元组。这是怎么回事呢?让我们抽丝剥茧一步步来看。
首先,我们来复习一下zip的用法:
zip(iter1, iter2, iter3, ...)
a = [1,2,3]
b = [4,5,6]
for i in zip(a,b):
print(i)
# 输出
(1, 4)
(2, 5)
(3, 6)
如上所示,zip接受多个可迭代容器作为参数,并同时对多个容器进行迭代,返回各个容器的相同位置的元素组成一个元素返回。zip的返回就是类似一个包含这多个组成后的元素的生成器。如上所示,使用for循环可打印各个迭代输出结果。
现在我们知道zip的用法了,zip是接受多个可迭代容器作为参数并同时迭代多个容器,将多个容器的迭代的元素组合在一起返回。那么当zip的输入只有一个可迭代容器的时候会怎么样呢?下面我们试试:
a = [1,2,3]
for i in zip(a):
print(i)
# 输出
(1,)
(2,)
(3,)
当zip输入只有一个容器时候,zip便只对这一个容器进行迭代,输出也是由这一个元素组成的元组。让我们将这一个容器稍微改变一下,将他换成一个嵌套元组试试?
a = [[1,2,3],[4,5,6],[7,8,9]]
for i in zip(a):
print(i)
# 输出
([1, 2, 3],)
([4, 5, 6],)
([7, 8, 9],)
有没有感觉有点熟悉?这个代码其实跟最开始的代码2已经一样了,这时候我们就不难理解为什么代码2的输出是那样的了。那么代码1的输出又该怎么解释呢?
要理解代码1,首先,我们得要知道*
的用法,在python中*
可以表示一个类似合并的意思,一般用在参数个数不确定的情况,将这些不确定个数的参数组合在一起形成一个类似可迭代容器的东西。通过下面的代码可以更好的理解:
def f(*l):
for i in l:
print(i)
a = [1,2,3]
b = [6,7,8]
f(a, b)
在上面的代码中,函数f的参数虽然只有一个l,但是由于其前面加了*
号,所以其可以表示任意多个参数。
再回到上面的代码1,我们通过在嵌套输入参数元组([1,2,3],[1,2,3],[1,2,3])
之前加上*
号,相当于将元组([1,2,3],[1,2,3],[1,2,3])
解包并将其中的各个元素(这里就是元组中的各个列表)传递给zip,所以代码1中zip的结果是分别同时迭代3个子列表的结果。