zip()函数 & * zip(*iterables)函数
1.定义
多个元组、列表、字典等旧迭代器
组合成新迭代器
- 从参数中的多个迭代器取元素组合成一个新的迭代器
- 返回:一个zip对象,其内部元素为元组;可以转化成列表或元组
- 传入参数:元组、列表、字典等迭代器
2.zip()函数
1.压缩:zip()函数
zip函数有一个参数:从迭代器中依次取一个元组,组成一个元组。
zip函数有两个参数:zip(a,b)函数分别从a和b中取一个元素组成元组,再次将组成的元组组合成一个新的迭代器。
(1)维数相同:正常组合对应位置的元素。
(2)维数不同:取两者中的最小的行列数
应用:a.矩阵点乘 & 相加(对应位置相乘/加)
b.列表元素依次相连
c.⚠️zip反转字典,使用dict()函数
print("-" * 20 + "zip()函数的用法" + "-" * 20)
# 1.当zip()函数中只有一个参数时,zip(iterable)从迭代器中依次取一个元组,组成一个元组。
mylist = [1, 2, 3, 4]
print(zip(mylist)) # <zip object at 0x10351d9c8>
print(type(zip(mylist))) # <class 'zip'>
print(list(zip(mylist))) # [(1,), (2,), (3,), (4,)]
# 2.当zip()函数有两个参数时,zip(a,b)函数分别从a和b中取一个元素组成元组,再次将组成的元组组合成一个新的迭代器。
# a与b的维数相同时,正常组合对应位置的元素。
# 当a与b行或列数不同时,取两者中的最小的行列数。
m = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
n = [[1, 1, 1], [2, 2, 3], [3, 3, 3]]
p = [[1, 1, 1], [2, 2, 2]]
# 维数相同
print(list(zip(m, n))) # [([1, 2, 3], [1, 1, 1]), ([4, 5, 6], [2, 2, 3]), ([7, 8, 9], [3, 3, 3])]
# 维数不同
print(list(zip(m, p))) # [([1, 2, 3], [1, 1, 1]), ([4, 5, 6], [2, 2, 2])]
print("-" * 20 + "zip()函数的应用" + "-" * 20)
# 矩阵相加减、点乘(for循环 + 列表生成式)
# 矩阵的相乘(点乘/点积)
matrix_multi = [x * y for a, b in zip(m, n) for x, y in zip(a, b)]
print("矩阵点乘:\n", matrix_multi) # [1, 2, 3, 8, 10, 18, 21, 24, 27]
# 矩阵相加
matrix_add = [x + y for a, b in zip(m, n) for x, y in zip(a, b)]
print("矩阵相加:\n", matrix_add) # [2, 3, 4, 6, 7, 9, 10, 11, 12]
# 列表元素依次相连
l = ['a', 'b', 'c', 'd', 'e', 'f']
print("列表元素依次相连结果:", list(zip(l[:-1], l[1:])))
3.、*zip(*iterables)函数
2.解压缩:*zip(*iterables)函数
*zip()函数是zip()函数的逆过程,将zip对象变成原先组合前的数据。
print("-" * 20 + "*zip(*iterables)函数的应用" + "-" * 20)
# print(list(zip(*(m, n)))) # ([1, 2, 3], [1, 1, 1]) ([4, 5, 6], [2, 2, 3]) ([7, 8, 9], [3, 3, 3])
print(*zip(m, n))
m2, n2 = zip(*(zip(m, n)))
print('m2是否是m', m == list(m2)) # true
print('n2是否是n', n == list(n2)) # true