Python – 每列的最大值

在处理数据的时候，有时需要获取每列的最大值。针对这种需求，Python 提供了多种实现方式。下面将介绍其中两种常用的方法。

方法一：使用 numpy 库

numpy 是 Python 中常用的科学计算库，它提供了强大的数组操作功能。使用 numpy 库可以很方便地获取每列的最大值。

import numpy as np

# 生成示例数据
data = np.array([
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
])

# 获取每列的最大值
max_vals = np.max(data, axis=0)

print(max_vals)  # 输出 [7 8 9]

在上述示例中，我们首先使用 numpy 库生成了一个 3 行 3 列的矩阵，然后使用 np.max 函数获取每列的最大值，并指定 axis=0 表示按列进行计算。最后，将返回的最大值数组打印出来。

方法二：使用 Python 自带的 max 函数

Python 中自带的 max 函数可以用于获取一个列表或元组中的最大值，因此我们也可以通过将矩阵的每列取出来转换为列表，再使用 max 函数求出最大值。

# 生成示例数据
data = [
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
]

# 获取每列的最大值
max_vals = [max(col) for col in zip(*data)]

print(max_vals)  # 输出 [7, 8, 9]

在上述示例中，我们使用了 zip(*data) 将矩阵的列解压出来，再对每列使用 max 函数获取最大值，并将结果保存在列表中。最后再将列表打印出来即可。

无论是使用 numpy 库还是 Python 自带的 max 函数，我们都可以轻松地获取矩阵每列的最大值，方便我们进行后续的数据处理。