在 numpy 数组中查找包含最大值的行或列

Question

如何在 2d numpy 数组中找到包含数组范围最大值的行或列？

原文由 Ferguzz 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

如果您只需要一个或另一个：

 np.argmax(np.max(x, axis=1))

对于专栏，和

np.argmax(np.max(x, axis=0))

对于行。

原文由 ecatmur 发布，翻译遵循 CC BY-SA 3.0 许可协议

Answer 2

您可以使用 np.argmax 以及 np.unravel_index 如

x = np.random.random((5,5))
print np.unravel_index(np.argmax(x), x.shape)

原文由 Geoff Reedy 发布，翻译遵循 CC BY-SA 3.0 许可协议