NumPy数组形状深度解析：理解维度与创建多维数组-Python教程-PHP中文网

NumPy数组形状深度解析：理解维度与创建多维数组

本文深入探讨numpy数组的维度与形状，重点解析一维数组 `(n,)` 与二维数组 `(1, n)` 之间的关键区别，并通过代码示例演示如何基于输入结构准确创建不同维度的数组。同时，文章还将介绍两种实用的数组形状转换技巧，帮助读者更灵活地处理数据维度。

在NumPy中，数组的形状（shape）和维度（ndim）是理解其数据结构的核心概念。shape 属性返回一个元组，表示数组在每个维度上的大小，而 ndim 属性则表示数组的维度数量。正确理解这两个概念对于高效地进行数据处理和科学计算至关重要。

一维数组的创建与形状：(N,) 的含义

当我们使用一个扁平的Python列表创建NumPy数组时，NumPy会将其解释为一维数组。例如，np.array([7, 1]) 创建的数组，其 shape 为 (2,)，表示它是一个包含两个元素的一维数组。这里的逗号表示这是一个包含一个元素的元组，即数组只有一个维度，其长度为2。

import numpy as np

# 创建一个一维数组
b_1d = np.array([7, 1], dtype=np.dtype(float))

print(f"数组 b_1d: {b_1d}")
print(f"b_1d 的形状 (shape): {b_1d.shape}") # 输出: (2,)
print(f"b_1d 的维度数量 (ndim): {b_1d.ndim}") # 输出: 1

从输出可以看出，b_1d 是一个真正的1D数组，其 ndim 为1。它不是一个一行两列的二维数组，而是一个简单的数据序列。

创建高维数组：嵌套列表的关键作用

要创建更高维度的NumPy数组，关键在于输入Python列表的嵌套层级。NumPy根据列表的嵌套深度来推断数组的维度。

创建二维数组：(1, N)

如果希望创建一个具有一行N列的二维数组，需要使用一个包含单个列表的嵌套列表。例如，np.array([[7, 1]]) 会被解释为一个二维数组。

# 创建一个二维数组（一行两列）
b_2d = np.array([[7, 1]], dtype=np.dtype(float))

print(f"数组 b_2d: {b_2d}")
print(f"b_2d 的形状 (shape): {b_2d.shape}") # 输出: (1, 2)
print(f"b_2d 的维度数量 (ndim): {b_2d.ndim}") # 输出: 2

这里，b_2d 的 shape 为 (1, 2)，明确表示它是一个2D数组，包含1行和2列。

创建三维数组：(1, 1, N)

依此类推，要创建三维数组，需要更深层次的列表嵌套。

# 创建一个三维数组
b_3d = np.array([[[7, 1]]], dtype=np.dtype(float))

print(f"数组 b_3d: {b_3d}")
print(f"b_3d 的形状 (shape): {b_3d.shape}") # 输出: (1, 1, 2)
print(f"b_3d 的维度数量 (ndim): {b_3d.ndim}") # 输出: 3

灵活转换数组形状的技巧

在实际应用中，我们可能需要将已创建的数组从一个维度转换为另一个维度，例如将一维数组转换为二维数组。NumPy提供了多种灵活的方法来实现这一点。

macOS UI 自动化技能包 —— 用 Python 控制你的 Mac

让 macOS 自动化像说话一样简单 —— 用代码控制你的 Mac，无需学习 AppleScript 系统要求 macOS 14+ (Sonoma) Python 3.9+ 需要开启：辅助功能 + 屏幕录制权限

下载

方法一：直接修改 shape 属性

可以直接修改数组的 shape 属性来改变其形状。需要注意的是，这种方法会原地修改数组，并且新形状的元素总数必须与原数组相同。

b_original = np.array([7, 1], dtype=np.dtype(float))
print(f"原始数组 b_original: {b_original}, 形状: {b_original.shape}")

# 将形状修改为 (1, 2)
b_original.shape = (1, 2)
print(f"修改形状后的 b_original: {b_original}, 形状: {b_original.shape}")
# 输出:
# 原始数组 b_original: [7. 1.], 形状: (2,)
# 修改形状后的 b_original: [[7. 1.]], 形状: (1, 2)

此方法简洁高效，但仅适用于元素总数不变的形状转换。

方法二：使用 None 或 np.newaxis 增加维度

通过在索引操作中使用 None 或 np.newaxis，可以在指定位置插入一个新的维度。这通常会返回一个数组的视图（view），而不是修改原数组。

b_array = np.array([7, 1], dtype=np.dtype(float))
print(f"原始数组 b_array: {b_array}, 形状: {b_array.shape}")

# 在第一个维度前添加一个新维度，使其成为行向量 (1, 2)
b_reshaped_none = b_array[None, :]
print(f"使用 [None, :] 转换后的 b_reshaped_none: {b_reshaped_none}, 形状: {b_reshaped_none.shape}")

# 或者使用 np.newaxis 达到同样效果
b_reshaped_newaxis = b_array[np.newaxis, :]
print(f"使用 [np.newaxis, :] 转换后的 b_reshaped_newaxis: {b_reshaped_newaxis}, 形状: {b_reshaped_newaxis.shape}")

# 如果想在最后一个维度后添加新维度，使其成为列向量 (2, 1)
b_reshaped_col = b_array[:, None]
print(f"使用 [:, None] 转换后的 b_reshaped_col (列向量): {b_reshaped_col}, 形状: {b_reshaped_col.shape}")
# 输出:
# 原始数组 b_array: [7. 1.], 形状: (2,)
# 使用 [None, :] 转换后的 b_reshaped_none: [[7. 1.]], 形状: (1, 2)
# 使用 [np.newaxis, :] 转换后的 b_reshaped_newaxis: [[7. 1.]], 形状: (1, 2)
# 使用 [:, None] 转换后的 b_reshaped_col (列向量): [[7.], [1.]], 形状: (2, 1)

这种方法非常灵活，可以精确控制新维度的插入位置，且通常不涉及数据复制，效率较高。