Python - numpy 矩阵每列中前 n 个元素的索引

Question

从二维 numpy 矩阵的每一列中 select 前 n 个元素的最佳方法是什么？（除了使用 for 循环）

Answer 1

也许切片？

如果你有：

matr = np.random.randn(4, 5)

您可以简单地 select 例如前 2 个 cols 的前 2 个元素做：

matr[:2,0:2]

尝试这些代码片段并检查它是否是您要查找的内容。