2022年 11月 3日

python中的 zip函数详解

python中zip()函数用法举例

 定义:zip([iterable, …])
  zip()是Python的一个内建函数,它接受一系列可迭代的对象作为参数,将对象中对应的元素打包成一个个tuple(元组),然后返回由这些tuples组成的list(列表)。若传入参数的长度不等,则返回list的长度和参数中长度最短的对象相同。利用*号操作符,可以将list unzip(解压),看下面的例子就明白了:

  1. a = [1, 2, 3]
  2. b = [2, 3, 4]
  3. c = [4, 5, 6, 7]
  4. # 返回一个对象
  5. zipped = zip(a, b)
  6. # list() 转换为列表
  7. print(list(zipped))
  8. 运行结果:
  9. [(1, 2), (2, 3), (3, 4)]
  10. # 元素个数与最短的列表一致
  11. zipped = zip(a, c)
  12. print(list(zipped))
  13. print(type(zipped), zipped)
  14. 运行结果:
  15. [(1, 4), (2, 5), (3, 6)]
  16. <class 'zip'> <zip object at 0x000001C7C9814180>
  17. # 对zip对象进行解压缩
  18. a1, a2 = zip(*zip(a, c))
  19. print(a1, a2)
  20. 运行结果:
  21. (1, 2, 3) (4, 5, 6)
  22. # zip对象可以转化为字典
  23. name_list = ["bob", "jim", "james", "julie", "june"]
  24. number_list = ["1", "2", "3", "4", "5"]
  25. name_and_number = dict(zip(name_list, number_list))
  26. print(name_and_number)
  27. 运行结果:
  28. {'bob': '1', 'jim': '2', 'james': '3', 'julie': '4', 'june': '5'}
  29. # 搭配for循环,支持并行迭代操作方法 zip()方法用在for循环中,就会支持并行迭代:
  30. l1 = [2, 3, 4]
  31. l2 = [4, 5, 6]
  32. for (x, y) in zip(l1, l2):
  33. print(x, y, '--', x * y)
  34. 运行结果:
  35. 2 4 -- 8
  36. 3 5 -- 15
  37. 4 6 -- 24
  38. 其实它的工作原理就是使用了zip()的结果,在for循环里解包zip结果中的元组,用元组赋值运算。
  39. 就好像(x,y)=(2,6),赋值、序列解包操作。在对文件的操作中我们也会用到遍历,
  40. 例如Python遍历文件夹目录与文件操作,就是很方便实用的。
  41. # 二维矩阵变换(矩阵的行列互换)
  42. a = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
  43. # 方法一,推导式:
  44. print([[row[col] for row in a] for col in range(len(a[0]))])
  45. 运行结果:
  46. [[1, 4, 7], [2, 5, 8], [3, 6, 9]]
  47. # 方法二,利用zip函数:
  48. print(list(zip(*a)))
  49. print(list(map(list, zip(*a))))
  50. 运行结果:
  51. [(1, 4, 7), (2, 5, 8), (3, 6, 9)]
  52. [[1, 4, 7], [2, 5, 8], [3, 6, 9]]
  53. 这种方法速度更快但也更难以理解,将list看成tuple解压,恰好得到我们“行列互换”的效果,
  54. 再通过对每个元素应用list()函数,将tuple转换为list