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

Question

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

Answer 1

也许是切片？

如果你有：

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

您可以简单地选择例如前 2 个列的前 2 个元素：

matr[:2,0:2]

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