在python中查找三维矩阵中每行的最大值和索引

pip install numpy

import numpy as np

# 定义一个三维矩阵
matrix = np.array([[[1, 2, 3], [4, 5, 6], [7, 8, 9]],
                   [[10, 11, 12], [13, 14, 15], [16, 17, 18]],
                   [[19, 20, 21], [22, 23, 24], [25, 26, 27]]])

# 查找每行的最大值
max_values = np.max(matrix, axis=2)

# 查找每行的最大值的索引
max_indices = np.argmax(matrix, axis=2)

# 打印结果
for i in range(matrix.shape[0]):
    print("第", i+1, "行的最大值为：", max_values[i])
    print("第", i+1, "行的最大值的索引为：", max_indices[i])