替换Numpy数组-Python

Question

a = np.array([[0, 0, 1], [1, 2, 3], [3, 4, 5]]) #a.shape = (3,3)

如何将a中的每个数组替换为其最大值？

例外输出-a = np.array([1, 2, 5]) #a.shape = (3,1)

感谢

Answer 1

您可以使用：

a = np.amax(a, axis=1)

请参阅此处的文档