Python基礎——zip

0. zip 的特殊情況

• 不等長：

  a = range(3)
  b = range(5)
  >> list(zip(a, b))
  [(0, 0), (1, 1), (2, 2)]
  

• zip(*)

  • *對其作用的物件進行 unpack，拆包；

  l = [[1, 2, 3], [4, 5, 6]]
  >> list(zip(l))
  [([1, 2, 3],), ([4, 5, 6],)]
  >> list(zip(*l))
  [(1, 4), (2, 5), (3, 6)]
  

1. 兩層迴圈與 zip 的一一對應

[(i, j) for i in x for j in y] 也即：

l = []
for i in x:
	for j in y:
		l.append((i, j))	

表達的是一種笛卡爾積的關係。那麼如何實現一一對應呢，這時就需要zip操作了：

>>> [(i, j) for i, j in zip(x, y)]
[(0, 'a'),
 (1, 'b'),
...
 (24, 'y'),
 (25, 'z')]

2. meshgrid

不由得讓人聯想起meshgrid函式，我們要實現`[0, 1, 2, 3, 4, 5]$\times$[0, 1, 2, 3, 4, 5]網格的整數節點位置：

>>> [Y, X] = np.meshgrid(range(0, 6), range(0, 6))
>>> Y
array([[0, 1, 2, 3, 4, 5],
       [0, 1, 2, 3, 4, 5],
       [0, 1, 2, 3, 4, 5],
       [0, 1, 2, 3, 4, 5],
       [0, 1, 2, 3, 4, 5],
       [0, 1, 2, 3, 4, 5]])
>>> X
array([[0, 0, 0, 0, 0, 0],
       [1, 1, 1, 1, 1, 1],
       [2, 2, 2, 2, 2, 2],
       [3, 3, 3, 3, 3, 3],
       [4, 4, 4, 4, 4, 4],
       [5, 5, 5, 5, 5, 5]])

for x in range(0, 6):
	for y in range(0, 6):
		x - y

便可轉換為矩陣化的做法：

X-Y

3. 二元tuple元素前後的調換

我們再次回到zip本身，如何實現一個list（tuple）前後元素的調換再集合：

[1, 2, 3, 4, 5] -> [(2, 1), (3, 2), (4, 3), (5, 4)]

對一個list做兩次索引再zip：

>>> l = [1, 2, 3, 4, 5]
>>> [(j, i) for i, j in zip(l[:-1], l[1:])]
[(2, 1), (3, 2), (4, 3), (5, 4)]