按行查找最大值并相应显示其列名

import numpy as np

# 定义一个示例的矩阵
matrix = np.array([[1, 2, 3],
                   [4, 5, 6],
                   [7, 8, 9]])

# 遍历每一行
for row in matrix:
    # 找到当前行的最大值和对应的列索引
    max_value = np.max(row)
    max_index = np.argmax(row)
    
    # 根据列索引获取列名
    column_names = ['A', 'B', 'C']  # 假设列名为A、B、C
    max_column_name = column_names[max_index]
    
    # 输出结果
    print(f"最大值为：{max_value}，对应的列名为：{max_column_name}")