深入理解 NumPy 中 axis 参数的含义与多维数组操作

山间有风

发布时间：2026-10-15| 2人浏览过

原创

本文详细解析 NumPy 中 axis 参数的核心机制，解释其如何在多维数组中指定运算方向。通过 sum、mean、min、max、sort 及 prod 等函数的具体示例，阐述 axis=0 与 axis=1 在二维及高维数组中的行为差异，帮助开发者准确掌握数据处理的维度控制。

首先探讨 axis 这一概念存在的必要性。

在 NumPy 模块中，核心处理对象通常是矩阵或多维数组。由于对这类数据结构的操作具有多种可能性，为了支持对数组或矩阵执行各种复杂功能，系统引入了 axis 参数来指定操作的方向。

接下来通过具体案例，展示选取不同 axis 值时，对二维数组执行 sum、mean、min 和 max 等操作的结果差异。

>>> import numpy as np
>>> arr=np.arange(16).reshape(2,4,2)
>>> arr
array([[[ 0,  1],
        [ 2,  3],
        [ 4,  5],
        [ 6,  7]],

       [[ 8,  9],
        [10, 11],
        [12, 13],
        [14, 15]]])

深入解析 axis 参数的实际作用与理解方式。

NumPy 是 Python 进行科学计算不可或缺的基础模块。随着深度学习的兴起，NumPy 的应用日益广泛。熟悉 NumPy 的开发者都知道，众多函数都依赖 axis 参数，且 axis 取值不同会导致结果截然不同。可以说，axis 让 NumPy 的多维数组操作更加灵活，但也增加了理解的难度。本文将通过详细示例，厘清 axis 的本质及其在 NumPy 中的作用。

引入 axis 的原因十分直观：NumPy 旨在处理矩阵或多维数组的运算，而在多维空间中，数据操作的可能性繁多。我们先通过一个简单示例来观察。假设存在一个二维数组：

>>> import numpy as np
>>> data = np.array([
... [1,2,1],
... [0,3,1],
... [2,1,4],
... [1,3,1]])

该数组用于表示样本数据的特征，其中每一行代表一个样本的三个特征值，每一列则对应不同样本的同一特征。若在分析过程中需要对每个样本的三个特征求和，应如何处理？方法如下：

>>> np.sum(data, axis=1)
array([4, 4, 7, 5])

若需计算每种特征的最小值，处理方式同样直接：

>>> np.min(data, axis=0)
array([0, 1, 1])

若希望获取所有样本所有特征的平均值，操作依然简便：

>>> np.average(data)
1.6666666666666667

由此可见，通过指定不同的 axis，NumPy 会沿不同方向执行操作：若不设置 axis，则对所有元素进行全局操作；若设置 axis=0，则沿纵轴（列方向）操作；若设置 axis=1，则沿横轴（行方向）操作。这仅适用于简单的二维数组。对于更高维度的数组，可总结为一条规则：设 axis=i，则 NumPy 沿第 i 个下标变化的方向进行操作。例如前例可表示为 data = [[a00, a01], [a10, a11]]，当 axis=0 时，沿第 0 个下标变化的方向操作，即 a00->a10, a01->a11，对应纵坐标方向；axis=1 时同理。下面通过一个四维数组的 sum 示例进行验证：

>>> data = np.random.randint(0, 5, [4,3,2,3])
>>> data
array([[[[4, 1, 0],
         [4, 3, 0]],
        [[1, 2, 4],
         [2, 2, 3]],
        [[4, 3, 3],
         [4, 2, 3]]],

       [[[4, 0, 1],
         [1, 1, 1]],
        [[0, 1, 0],
         [0, 4, 1]],
        [[1, 3, 0],
         [0, 3, 0]]],

       [[[3, 3, 4],
         [0, 1, 0]],
        [[1, 2, 3],
         [4, 0, 4]],
        [[1, 4, 1],
         [1, 3, 2]]],

       [[[0, 1, 1],
         [2, 4, 3]],
        [[4, 1, 4],
         [1, 4, 1]],
        [[0, 1, 0],
         [2, 4, 3]]]])

当 axis=0 时，NumPy 沿第 0 维方向求和，即第一个元素值为 a0000+a1000+a2000+a3000=11，第二个元素值为 a0001+a1001+a2001+a3001=5，依此类推，最终结果如下：

>>> data.sum(axis=0)
array([[[11,  5,  6],
        [ 7,  9,  4]],

       [[ 6,  6, 11],
        [ 7, 10,  9]],

       [[ 6, 11,  4],
        [ 7, 12,  8]]])

当 axis=3 时，NumPy 沿第 3 维方向求和，即第一个元素值为 a0000+a0001+a0002=5，第二个元素值为 a0010+a0011+a0012=7，依此类推，最终结果如下：

>>> data.sum(axis=3)
array([[[ 5,  7],
        [ 7,  7],
        [10,  9]],

       [[ 5,  3],
        [ 1,  5],
        [ 4,  3]],

       [[10,  1],
        [ 6,  8],
        [ 6,  6]],

       [[ 2,  9],
        [ 9,  6],
        [ 1,  9]]])

使用 axis 相关函数的具体场景。

在 NumPy 中，使用 axis 参数的场景非常广泛。除了前文提到的 average、max、min 和 sum，常见的还有 sort 和 prod（乘积）。下面分别举例说明：

关于 sort 函数的应用示例。

>>> data = np.random.randint(0, 5, [3,2,3])
>>> data
array([[[4, 2, 0],
        [0, 0, 4]],

       [[2, 1, 1],
        [1, 0, 2]],

       [[3, 0, 4],
        [0, 1, 3]]])
>>> np.sort(data)  ## 默认对最大的axis进行排序，这里即是axis=2
array([[[0, 2, 4],
        [0, 0, 4]],

       [[1, 1, 2],
        [0, 1, 2]],

       [[0, 3, 4],
        [0, 1, 3]]])
>>> np.sort(data, axis=0)  # 沿着第0维进行排序，原先的a000->a100->a200转变为a100->a200->a000
array([[[2, 0, 0],
        [0, 0, 2]],

       [[3, 1, 1],
        [0, 0, 3]],

       [[4, 2, 4],
        [1, 1, 4]]])
>>> np.sort(data, axis=1)  # 沿着第1维进行排序
array([[[0, 0, 0],
        [4, 2, 4]],

       [[1, 0, 1],
        [2, 1, 2]],

       [[0, 0, 3],
        [3, 1, 4]]])
>>> np.sort(data, axis=2)  # 沿着第2维进行排序
array([[[0, 2, 4],
        [0, 0, 4]],

       [[1, 1, 2],
        [0, 1, 2]],

       [[0, 3, 4],
        [0, 1, 3]]])
>>> np.sort(data, axis=None)  # 对全部数据进行排序
array([0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 2, 2, 2, 3, 3, 4, 4, 4])

关于 prod（即 product，乘积）函数的应用示例。

 >>> np.prod([[1.,2.],[3.,4.]])
 24.0

 >>> np.prod([[1.,2.],[3.,4.]], axis=1)
 array([  2.,  12.])

 >>> np.prod([[1.,2.],[3.,4.]], axis=0)
 array([ 3.,  8.])

相信通过上述讲解与实例，您已对 axis 有了较为清晰的认识。个人认为，若未能理解 axis 的真正含义，很难熟练运用 NumPy 进行高效的数据处理。