列表-Ⅵ

BrightXiong

>>> oho = [1, 2, 3, 4,5]
>>> for i in range(len(oho)):
...         oho[i] = oho[i] * 2
...
>>> oho
[2, 4, 6, 8, 10]

>>> aa = [1, 2, 3, 4, 5]
>>> aa = [i * 2 for i in aa]
>>> aa
[2, 4, 6, 8, 10]

>>> matrix = [[1, 2, 3],
...                       [4, 5, 6],
...                       [7, 8, 9]]

>>> col2 = [row[1] for row in matrix]
>>> col2
[2, 5, 8]

>>> diag = [matrix[i][i] for i in range(len(matrix))]
>>> diag
[1, 5, 9]