1. 程式人生 > >python中for迴圈的小技巧

python中for迴圈的小技巧

偶爾看到一個神奇的用法:

[[row[i] for row in matrix] for i in range(4)]

這是幹嘛的?就是把一個矩陣轉制。。轉制。。轉制。。我擦那麼吊?

然後就呼叫了一部分試試,發現樂是這麼個意思:

當for語句再外面有個[  ]的時候。可以在for前面新增一個簡短的語句,然後通過for形成一個list。好厲害的感覺!
如:
 [row[2]+1 for row in matrix]

這個結果就是再matrix中沒一行的第三個元素取出。這樣就能理解前面那個語句啦!