博客

按需求取矩阵指定元素

生成一个由0，1组成的4×4矩阵

import numpy
matrix=numpy.random.randint(0,2,size=(4,4))
#matrix=numpy.random.randint(0,high=2,size=(4,4))
print(matrix)

输出结果

[[0 1 0 1]
 [0 0 0 1]
 [0 1 0 0]
 [0 0 0 1]]

显示矩阵的形状

print(matrix.shape)

输出结果

(4, 4)

选取矩阵特定行

mat_row1=matrix[0]
mat_row13=matrix[0:3]
print(mat_row1)
print(mat_row13)

输出结果

[0 1 0 1]
[[0 1 0 1]
 [0 0 0 1]
 [0 1 0 0]]

选取矩阵特定列

mat_col1=matrix[:,0]
mat_col13=matrix[:,0:3]
print(mat_col1)
print(mat_col13)

输出结果

[0 0 0 0]
[[0 1 0]
 [0 0 0]
 [0 1 0]
 [0 0 0]]

选取矩阵指定行列元素

mat_r1c1=matrix[0,0]
print(mat_r1c1)

输出结果

0

选取矩阵的单数行

mat_col_singular=matrix[range(0,matrix.shape[0],2)]
print(mat_col_singular)

输出结果

[[0 1 0 1]
 [0 1 0 0]]

选取矩阵的偶数行

mat_col_evennumber=matrix[range(1,matrix.shape[0],2)]
print(mat_col_evennumber)

输出结果

[[0 0 0 1]
 [0 0 0 1]]

选取矩阵的单数列

mat_col_singular=matrix[:,range(0,matrix.shape[0],2)]
print(mat_col_singular)

输出结果

[[0 0]
 [0 0]
 [0 0]
 [0 0]]

选取矩阵的偶数列

mat_col_evennumber=matrix[:,range(1,matrix.shape[0],2)]
print(mat_col_evennumber)

输出结果

[[1 1]
 [0 1]
 [1 0]
 [0 1]]