Python如何计算numpy数组每列最大值并以对象形式打印输出

调用np.max时指定axis=0参数即可，该参数会让计算沿着数组第0轴（行方向）执行，最终返回每一列的最大值，和你需要的输出匹配。

完整可运行代码 #

import numpy as np

# 构造目标数组
arr = np.array([
    [7, 0, 0, 6],
    [5, 6, 6, 1],
    [4, 1, 6, 7],
    [5, 3, 4, 7]
])

# 计算每列最大值并存入对象
col_max = np.max(arr, axis=0)

# 打印结果
print(col_max)

运行结果 #

执行代码后输出如下：

[7 6 6 7]

和你期望的[7, 6, 6, 7]数值完全一致，numpy数组打印时默认用空格分隔元素，不存在结果差异。

补充：如果需要计算每行最大值，把参数改为axis=1即可；如果不指定axis参数，np.max会返回整个数组的全局最大值。

内容的提问来源于stack exchange，提问作者yomi