python - np.argmax()中axis参数的作用是什么？

Question

我们知道，在轴参数中，0,1 表示按列和行的最大元素索引，但对于 2,3 等等，它表示什么？这里给出了一个示例代码。这段代码的输出意义是什么？

标签： pythonpython-3.xnumpy

Answer 1

当你有一个更高维度的数组时，你也会有新的轴。例如，在一个 3 维数组（例如一个立方体）中，您将有 3 个轴（行、列、深度）。

当您在中传递轴时，np.argmax您正在告诉 numpy 您想要最大参数的轴。3 将引发错误，因为您的数组只有 3 个轴（0、1、2）。

这是一篇关于 numpy 数组轴的文章。