深入理解NumPy数组的形状与维度：从一维到多维的创建与转换-Python教程-PHP中文网

深入理解NumPy数组的形状与维度：从一维到多维的创建与转换

花韻仙語

发布： 2025-11-23 13:40:33

原创

817人浏览过

深入理解numpy数组的形状与维度：从一维到多维的创建与转换

本文旨在深入探讨NumPy数组的形状（shape）和维度（ndim）概念，重点解析一维数组与二维数组在创建时的区别，以及为何 `np.array([x, y])` 默认生成一维数组 `(2,)` 而非二维 `(1, 2)`。文章将通过示例代码详细演示如何精确控制数组的维度，并介绍多种将一维数组转换为指定二维形状的方法，帮助读者避免常见误解，更高效地利用NumPy处理数据。

在NumPy中，数组的形状（shape）和维度（ndim）是理解和操作数据的核心概念。shape 描述了数组在每个维度上的大小，表示为一个元组；而 ndim 则指明了数组的维度数量。正确理解这两个属性对于高效地进行数值计算至关重要。

NumPy数组的形状与维度概述

考虑以下NumPy数组的创建及其形状输出：

import numpy as np

A = np.array([
        [-1, 3],
        [3, 2]
    ], dtype=np.dtype(float))

b = np.array([7, 1], dtype=np.dtype(float))

print(f"Shape of A: {A.shape}")
print(f"Shape of b: {b.shape}")

登录后复制

输出结果为：

Shape of A: (2, 2)
Shape of b: (2,)

登录后复制

对于数组 A，其形状 (2, 2) 明确表示它是一个2行2列的二维数组。然而，对于数组 b，其形状 (2,) 常常引起初学者的困惑，误以为它是一个1行2列的二维数组 (1, 2)。实际上，b 的形状 (2,) 表示它是一个包含2个元素的一维数组。

一维数组的本质：为何 (2,) 而非 (1, 2)

NumPy在创建数组时，会根据输入的数据结构自动推断其维度。当使用 np.array([7, 1]) 时，我们提供的是一个扁平的Python列表 [7, 1]。NumPy将其解释为一个包含两个元素的序列，因此创建了一个一维数组。

我们可以通过 ndim 属性来验证数组的维度：

import numpy as np

b = np.array([7, 1], dtype=np.dtype(float))
print(f"Dimension of b: {b.ndim}")

登录后复制

输出将是：

Dimension of b: 1

登录后复制

这表明 b 确实是一个一维数组。形状 (2,) 中的逗号表示这是一个元组，但只有一个元素，即该维度的大小为2。这与 (1, 2) 这种表示“1行，2列”的二维形状有着本质区别。

显式创建多维数组

要创建具有特定维度的数组，尤其是在需要二维或更高维度的场景下，需要通过嵌套列表来明确指定。嵌套的层数直接对应于数组的维度。

创建二维数组

如果目标是创建一个1行2列的二维数组，需要将内部元素 [7, 1] 再次封装在一个列表中，形成 [[7, 1]]：

import numpy as np

b_2d = np.array([[7, 1]], dtype=np.dtype(float))

print(f"Shape of b_2d: {b_2d.shape}")
print(f"Dimension of b_2d: {b_2d.ndim}")
print(f"b_2d content:\n{b_2d}")

登录后复制

输出结果：

Shape of b_2d: (1, 2)
Dimension of b_2d: 2
b_2d content:
[[7. 1.]]

登录后复制

此时，b_2d 才是我们期望的1行2列的二维数组。

What-the-Diff

检查请求差异，自动生成更改描述

103

查看详情

创建三维数组

类似地，如果需要一个三维数组，则需要三层嵌套：

import numpy as np

b_3d = np.array([[[7, 1]]], dtype=np.dtype(float))

print(f"Shape of b_3d: {b_3d.shape}")
print(f"Dimension of b_3d: {b_3d.ndim}")
print(f"b_3d content:\n{b_3d}")

登录后复制

输出结果：

Shape of b_3d: (1, 1, 2)
Dimension of b_3d: 3
b_3d content:
[[[7. 1.]]]

登录后复制

灵活转换数组维度

即使已经创建了一个一维数组，NumPy也提供了多种方法将其转换为所需的维度，而无需重新创建。

方法一：直接修改 shape 属性

可以直接为数组的 shape 属性赋值一个新的元组，但前提是新形状的元素总数必须与原数组的元素总数相同。

import numpy as np

b_original = np.array([7, 1], dtype=np.dtype(float))
print(f"Original b shape: {b_original.shape}") # Output: (2,)

b_original.shape = (1, 2) # 将一维数组转换为1行2列的二维数组

print(f"Modified b shape: {b_original.shape}") # Output: (1, 2)
print(f"Modified b content:\n{b_original}")

登录后复制

输出结果：

Original b shape: (2,)
Modified b shape: (1, 2)
Modified b content:
[[7. 1.]]

登录后复制

这种方法会直接修改原数组的形状。

方法二：使用 np.newaxis 或 None 扩展维度

NumPy的索引机制允许使用 np.newaxis（或其简写 None）在指定位置插入新维度。这是一种非常灵活且常用的方法。

import numpy as np

b_original = np.array([7, 1], dtype=np.dtype(float))
print(f"Original b shape: {b_original.shape}") # Output: (2,)

# 在第一个维度（行）之前插入一个新维度
b_reshaped_none = b_original[None, :]
print(f"Reshaped b (None, : ) shape: {b_reshaped_none.shape}") # Output: (1, 2)
print(f"Reshaped b (None, : ) content:\n{b_reshaped_none}")

# 也可以在第二个维度（列）之后插入一个新维度，如果需要 (2, 1) 的形状
b_reshaped_none_col = b_original[:, None]
print(f"Reshaped b (:, None) shape: {b_reshaped_none_col.shape}") # Output: (2, 1)
print(f"Reshaped b (:, None) content:\n{b_reshaped_none_col}")

登录后复制

输出结果：

Original b shape: (2,)
Reshaped b (None, : ) shape: (1, 2)
Reshaped b (None, : ) content:
[[7. 1.]]
Reshaped b (:, None) shape: (2, 1)
Reshaped b (:, None) content:
[[7.]
 [1.]]

登录后复制

None 在索引中作为一个占位符，表示在该位置添加一个大小为1的新维度。b[None, :] 等价于 b[np.newaxis, :]，它在数组的第一个维度前添加了一个新维度，将 (2,) 变成了 (1, 2)。同样，b[:, None] 在数组的第二个维度前（即原数组的末尾）添加了一个新维度，将 (2,) 变成了 (2, 1)。

总结与最佳实践

理解NumPy数组的 shape 和 ndim 是进行有效数据操作的基础。核心要点包括：

一维数组的形状表示：np.array([x, y]) 默认创建的是一维数组，其形状为 (N,)，其中 N 是元素数量。
多维数组的创建：通过嵌套列表来显式创建多维数组，嵌套层数决定了数组的维度。例如，[[...]] 用于创建二维数组。
维度转换的灵活性：
- 直接修改 array.shape 属性可以改变数组的形状，但需保证元素总数不变。
- 使用 np.newaxis 或 None 在索引中插入新维度是一种强大且常用的方法，可以灵活地将一维数组扩展为多维数组，例如将 (N,) 转换为 (1, N) 或 (N, 1)。

在实际应用中，根据数据的自然结构和后续计算的需求，选择最合适的数组创建和维度转换方法，可以显著提高代码的可读性和执行效率。例如，在机器学习中，单一样本通常需要被转换为 (1, N) 的形状以匹配模型输入的要求。掌握这些基本概念和技巧，将使您在NumPy的数据处理中游刃有余。

以上就是深入理解NumPy数组的形状与维度：从一维到多维的创建与转换的详细内容，更多请关注php中文网其它相关文章！