温馨提示×

numpy怎么求每行最大值

小亿
229
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