以下是一个按行找到最大值,并显示其对应的列名的示例代码:
import numpy as np
# 创建一个2维数组
arr = np.array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])
# 获取数组的行数和列数
rows, cols = arr.shape
# 遍历每一行,找到最大值及其对应的列名
for i in range(rows):
max_value = arr[i][0] # 假设第一列为最大值
max_col = 0 # 记录最大值对应的列名
# 遍历当前行的每一列
for j in range(1, cols):
if arr[i][j] > max_value:
max_value = arr[i][j]
max_col = j
print("第", i+1, "行的最大值为:", max_value, ",对应的列名为:", max_col+1)
运行以上代码,输出结果为:
第 1 行的最大值为: 3 ,对应的列名为: 3
第 2 行的最大值为: 6 ,对应的列名为: 3
第 3 行的最大值为: 9 ,对应的列名为: 3
这个示例使用了NumPy库来创建和操作数组。首先,我们定义了一个2维数组arr。然后,使用arr.shape获取数组的行数和列数。接下来,使用嵌套的for循环遍历每一行的每一列,找到每行的最大值及其对应的列名。最后,打印每行的最大值和对应的列名。