python获取矩阵某⼀列元素今天写线性回归问题的时候遇到了⼀个问题:
对于⼀个⼆维矩阵, python如何遍历其某⼀列元素.
遍历⼀⾏是很简单的, 直接使⽤索引即可, 但是遍历⼀列呢?
python怎么读取excel某一列⽅法⼀: 直接遍历法
使⽤⼀个循环, 对每⼀⾏单独出这列上的对应元素.
⽅法⼆: 列表解析法
这个⽅法其实和上⾯的那个⽅法是⼀样的, 不过列表解析的技术更酷⼀些.
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])