利用列表推导式处理矩阵

根据列表推导式表达矩阵中的元素的时候，我开始以为会得到[4, 5, 6]，结果出乎意料。

>>> matrix = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
>>> row = [row[1] for row in matrix]
>>> print(row)
[2, 5, 8]

 然后我就使用for循环去逐步分解：

matrix = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
# 遍历列表中每个元素
for i in matrix:
...     print(i)
...
[1, 2, 3]
[4, 5, 6]
[7, 8, 9]

# 查找元素中的元素
for i in matrix:
...     print(i[1])
...
2
5
8

扩展：如果matrix中是三个中括号，根据切片查找矩阵中的元素。

>>> matrix = [[[1, 2, 3], [4, 5, 6], [7, 8, 9]]]
# 遍历matrix中每个元素
>>> for i in matrix:
...     print(i)
...
[[1, 2, 3], [4, 5, 6], [7, 8, 9]]
# 看一下i的长度，此时i类似与[a,b,c]
>>> print(len(i))
3
# 查找i中的元素
>>> for i in matrix:
...     print(i[1])
...
[4, 5, 6]