numpyで各行の最大値を求めるには

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]

まず、多次元配列arrを作成します。その後にamax関数を使ってaxis=1で各行の最大値を求めます。そして、その各行の最大値を出力します。