今天寫線性迴歸問題的時候遇到了一個問題:

對於一個二維矩陣, python如何遍歷其某一列元素.

遍歷一行是很簡單的, 直接使用索引即可, 但是遍歷一列呢?

方法一: 直接遍歷法

使用一個迴圈, 對每一行單獨找出這列上的對應元素.

方法二: 列表解析法

這個方法其實和上面的那個方法是一樣的, 不過列表解析的技術更酷一些.

a = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
print([i[2] for i in a])

方法三: numpy矩陣法

numpy庫中自帶了遍歷矩陣列元素的方法

import numpy as np
c = np.arange(9).
 
reshape(3, 3)
print(c[:, 2])

如果不是np.ndarray型別, 可以利用numpy進行型別轉換, 然後再利用上面這種方法.

b = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
b = np.array(b)
print(b[:, 1])