Python:选择矩阵的第N行

时间:2018-09-01 15:18:27

标签: python matrix

是否有人知道如何选择矩阵的多行以形成一个新的行-例如我想选择矩阵的第三行,并使用这些行建立一个新矩阵。

非常感谢您的帮助,

尼古拉斯

1 个答案:

答案 0 :(得分:0)

使用numpys ndarray创建一个以10行3列为例的矩阵的示例

import numpy as np

matrix = np.ndarray(shape=(10,3))

rows = np.shape(matrix)[0] #number of rows
columns = np.shape(matrix)[1] #number of columns
l = range(rows)[0::3] #indexes of each third element including the first element

new_matrix = np.ndarray(shape=(len(l),columns)) #Your new matrix

for i in range(len(l)):
    new_matrix[i] = matrix[l[i]] #adding each third row from matrix to new_matrix