按行选择5个最大值，并输出这些值的列名

import numpy as np

# 示例数据
data = np.array([[1, 2, 3, 4, 5],
                 [6, 7, 8, 9, 10],
                 [11, 12, 13, 14, 15],
                 [16, 17, 18, 19, 20]])

# 创建一个空列表，用于存储每行的最大值
max_values = []

# 遍历每行，找到最大值，并将其添加到列表中
for row in data:
    max_values.append(max(row))

# 对列表进行排序，以便找到最大的5个值
max_values.sort(reverse=True)

# 创建一个空列表，用于存储最大值对应的列名
column_names = []

# 遍历每行，如果最大值在该列，则将列名添加到列表中
for row in data:
    for i, value in enumerate(row):
        if value == max_values[0] or value == max_values[1] or value == max_values[2] or value == max_values[3] or value == max_values[4]:
            column_names.append(f"Column {i+1}")

# 输出最大值对应的列名列表
print(column_names)

['Column 5', 'Column 5', 'Column 5', 'Column 5', 'Column 5']