python:如何创建子矩阵?努比

python:如何创建子矩阵?努比,python,matrix,Python,Matrix,我有一个1500X2的矩阵,我必须创建10个150行的子矩阵。如果没有for循环,我如何执行此操作。我需要一个函数,因为[:]太慢和复杂您可以使用该函数选择矩阵的一系列行。您可以传递要选择的索引和要选择项目的轴 import numpy as np indices = list(range(0,3)) array = np.random.rand(5,2) print(array) res = np.take(array, indices, 0) print(res) 你会得到这样的结果

我有一个1500X2的矩阵,我必须创建10个150行的子矩阵。如果没有
for
循环,我如何执行此操作。我需要一个函数,因为[:]太慢和复杂

您可以使用该函数选择矩阵的一系列行。您可以传递要选择的索引和要选择项目的轴

import numpy as np

indices = list(range(0,3))

array = np.random.rand(5,2)
print(array)

res = np.take(array, indices, 0)
print(res)
你会得到这样的结果:

[[0.63680493 0.27066094]
 [0.71182288 0.48258969]
 [0.61321531 0.02215374]
 [0.98148503 0.5669895 ]
 [0.42720908 0.57326236]]

[[0.63680493 0.27066094]
 [0.71182288 0.48258969]
 [0.61321531 0.02215374]]
为什么不展示一个测试和一些基准来说明什么是慢的呢?