温馨提示×

numpy怎么求每行最大值

小亿
204
2023-11-24 05:02:24
栏目: 编程语言

可以使用numpy的amax函数来求每行的最大值。amax函数可以接受一个数组作为参数,并返回数组中的最大值。

以下是一个示例代码:

import numpy as np

# 创建一个二维数组
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 求每行的最大值
max_values = np.amax(arr, axis=1)

print(max_values)

运行结果为:

[3 6 9]

在上述代码中,我们首先创建了一个2维数组arr。然后使用amax函数,指定axis=1参数来表示按行求最大值。最后将得到的每行最大值打印出来。

0