用最大值替换numpy数组

Question

我有一个数组，

a = np.array([[0,9,8],[5,6,4]])

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

输出除外 - a = np.array([9,6]) 其中 9 是 [0,9,8] 中的最大值，6 是 [5,6,4] 中的最大值

谢谢

Answer 1

你应该使用

np.max(a, axis=1)

Link to documentation

Answer 2

您可以执行此操作的另一种实现

np.array([max(i) for i in a])