可以使用numpy的amax函數(shù)來求每行的最大值。amax函數(shù)可以接受一個數(shù)組作為參數(shù)，并返回數(shù)組中的最大值。
以下是一個示例代碼：
import numpy as np
# 創(chuàng)建一個二維數(shù)組
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 求每行的最大值
max_values = np.amax(arr, axis=1)
print(max_values)
運(yùn)行結(jié)果為：
[3 6 9]
在上述代碼中，我們首先創(chuàng)建了一個2維數(shù)組arr。然后使用amax函數(shù)，指定axis=1參數(shù)來表示按行求最大值。最后將得到的每行最大值打印出來。
億速云公眾號
手機(jī)網(wǎng)站二維碼