下面是一个按行选择5个最高值,并输出这些值所在的列名的示例代码:
import numpy as np
# 创建一个5行10列的随机矩阵
matrix = np.random.randint(0, 100, size=(5, 10))
print("原始矩阵:")
print(matrix)
# 按行选择5个最高值
max_values = np.max(matrix, axis=1)
max_indices = np.argsort(max_values)[-5:] # 找到最大值的索引
# 输出最高值所在的列名
column_names = ['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J']
for index in max_indices:
max_column = np.argmax(matrix[index])
print("第", index+1, "行的最高值为", matrix[index][max_column], ",所在列名为", column_names[max_column])
这段代码首先创建了一个5行10列的随机矩阵,然后使用np.max
函数按行获取每行的最大值,使用np.argsort
函数找到最大值的索引,取最大的5个索引。最后,通过np.argmax
函数找到每行最大值所在的列索引,并根据列索引获取列名。
下一篇:按行旋转表格的百分比总和