numpy索引(取出矩阵中的元素)

import numpy as np
a = np.arange(3, 13).reshape((2, 5))
print(a)
print(a[1])  # 取出矩阵中的第一行
print(a[0][2])  # 取出矩阵中第0行第二列的数
print(a[0, 2])  # 取出矩阵中第0行第二列的数
print(a[1, 1:3])  # 取出矩阵中第一行 第1列到第3列(前闭后开)的数据
print(a[:, 2])  # :表示所有
# 按行遍历矩阵
for row in a:
    print(row)
#按列遍历矩阵
![for column in a.T:
    print(column)
#生成一个迭代其器,把矩阵变为一个向量
print(a.flatten())

numpy索引(取出矩阵中的元素)_第1张图片

你可能感兴趣的:(Python)