Python 数组每行最大值计算 - 使用 NumPy amax 函数

可以使用 NumPy 库中的 'amax' 函数来计算数组 'a' 中每行的最大值。

import numpy as np

a = np.array([[9, 9, 4], [8, 8, 1], [5, 3, 6], [3, 3, 3], [2, 1, 9]])

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

输出结果为：

[9 8 6 3 9]