Python怎么获取NumPy数组的形状（shape）_NumPy数组维度与形状查询-Python教程-PHP中文网

Python怎么获取NumPy数组的形状（shape）_NumPy数组维度与形状查询

尼克

发布： 2025-09-12 18:29:01

原创

623人浏览过

答案是访问数组的.shape属性。该属性返回一个元组，表示数组在各维度上的大小，例如一维数组为(元素个数,)，二维数组为(行数, 列数)，三维数组为(深度, 行数, 列数)，且其长度等于数组维度ndim，是数据处理中进行校验、优化和操作的核心依据。

python怎么获取numpy数组的形状（shape）_numpy数组维度与形状查询

在Python中，要获取NumPy数组的形状（shape），最直接且常用的方式就是访问数组对象的

.shape

登录后复制

属性。它会返回一个元组（tuple），其中包含了数组在每个维度上的大小。这个元组的长度就是数组的维度数量。

解决方案

获取NumPy数组形状的核心方法是使用其内置的

.shape

登录后复制

属性。当你在一个NumPy数组对象上调用这个属性时，它会返回一个元组，这个元组的每个元素对应数组在相应维度上的大小。

例如，一个二维数组的

shape

登录后复制

可能是

(行数, 列数)

登录后复制

，而一个三维数组则可能是

(深度, 行数, 列数)

登录后复制

。

import numpy as np

# 1D 数组
arr_1d = np.array([1, 2, 3, 4, 5])
print(f"arr_1d 的形状: {arr_1d.shape}") # 输出: (5,)

# 2D 数组
arr_2d = np.array([[1, 2, 3], [4, 5, 6]])
print(f"arr_2d 的形状: {arr_2d.shape}") # 输出: (2, 3)

# 3D 数组
arr_3d = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])
print(f"arr_3d 的形状: {arr_3d.shape}") # 输出: (2, 2, 2)

# 标量（0维数组）
scalar_arr = np.array(10)
print(f"scalar_arr 的形状: {scalar_arr.shape}") # 输出: ()

登录后复制

除了直接访问

.shape

登录后复制

属性，你也可以使用NumPy的全局函数

np.shape()

登录后复制

。这个函数在处理非NumPy数组但结构类似列表或元组的数据时特别有用，它会尝试将其转换为NumPy数组并返回其形状。不过，对于已经存在的NumPy数组，

.shape

登录后复制

属性通常是更直接、更推荐的做法，因为它避免了潜在的函数调用开销。

立即学习“Python免费学习笔记（深入）”；

# 使用 np.shape()
list_data = [[10, 20], [30, 40]]
print(f"list_data 的形状 (通过 np.shape()): {np.shape(list_data)}") # 输出: (2, 2)

# 与 arr_2d.shape 结果一致
print(f"arr_2d 的形状 (通过 np.shape()): {np.shape(arr_2d)}") # 输出: (2, 3)

登录后复制

NumPy数组的维度（ndim）与形状（shape）究竟有何不同？

这确实是初学者，甚至有时是我自己，在处理NumPy数组时容易混淆的一个点。简单来说，

ndim

登录后复制

（number of dimensions）告诉你数组有多少个“轴”或者说“层级”，而

shape

登录后复制

则更具体，它是一个元组，告诉你每个轴上有多少个元素。

举个例子，想象一个表格，它有行和列。这个表格就是二维的，所以它的

ndim

登录后复制

是2。而

shape

登录后复制

可能会是

(5, 3)

登录后复制

，这意味着它有5行和3列。如果我有一个简单的列表

[1, 2, 3]

登录后复制

，它的

ndim

登录后复制

是1，因为只有一个轴（序列），而

shape

登录后复制

是

(3,)

登录后复制

，表示这个轴上有3个元素。

从技术角度讲，

ndim

登录后复制

是

len(arr.shape)

登录后复制

的缩写。它只是一个整数，告诉你元组里有多少个数字。而

shape

登录后复制

元组里的每个数字，才是真正定义了数组在那个特定方向上能容纳多少数据。理解这个区别非常关键，因为

ndim

登录后复制

通常用于泛型操作，比如判断一个数组是不是多维的，而

shape

登录后复制

则直接影响到索引、切片和广播等更精细的操作。我个人在写一些通用函数时，会先检查

ndim

登录后复制

来确定处理逻辑，然后深入到

shape

登录后复制

的具体值来执行精细的数组操作。

即构数智人

即构数智人是由即构科技推出的AI虚拟数字人视频创作平台，支持数字人形象定制、短视频创作、数字人直播等。

查看详情

如何解读NumPy数组shape输出中的各个数值？

解读

shape

登录后复制

元组中的每个数值，其实就是理解数组的“布局”。每个数值都代表了对应维度上的大小或长度。这个顺序是固定的，通常是从最外层（最“粗”的维度）到最内层（最“细”的维度）。

一维数组（
ndim=1
登录后复制
）:
```
shape
```
登录后复制
会是一个包含单个整数的元组，例如
```
(5,)
```
登录后复制
。这个数字直接表示数组中元素的总数量。就像一个长条，它有多长。
```
arr = np.array([10, 20, 30])
print(arr.shape) # 输出: (3,)
```
登录后复制
二维数组（
ndim=2
登录后复制
）:
```
shape
```
登录后复制
会是一个包含两个整数的元组，例如
```
(行数, 列数)
```
登录后复制
。第一个数字代表行数，第二个数字代表列数。这就像我们最常接触的表格数据。
```
arr = np.array([[1, 2, 3], [4, 5, 6]])
print(arr.shape) # 输出: (2, 3) - 2行3列
```
登录后复制
三维数组（
ndim=3
登录后复制
）:
```
shape
```
登录后复制
会是一个包含三个整数的元组，例如
```
(深度, 行数, 列数)
```
登录后复制
。第一个数字通常可以理解为“层”或“块”的数量，第二个是每层的行数，第三个是每层的列数。想象一下堆叠起来的表格，或者一张彩色图像（高度、宽度、通道数）。
```
arr = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])
print(arr.shape) # 输出: (2, 2, 2) - 2个2x2的矩阵
```
登录后复制
更高维数组: 依此类推，
```
shape
```
登录后复制
元组的长度会增加，每个数字都对应一个维度的大小。理解这些数字的含义，对于进行正确的数组切片、索引以及广播操作至关重要。我发现很多时候，调试NumPy代码就是从正确理解
```
shape
```
登录后复制
开始的。当你看到一个
```
ValueError: operands could not be broadcast together with shapes (X, Y) (A, B)
```
登录后复制
时，第一反应就应该是检查两个数组的
```
shape
```
登录后复制
，看看它们是否兼容。

掌握NumPy数组shape对数据处理和性能优化的实际意义是什么？

理解并能够正确获取和解读NumPy数组的

shape

登录后复制

，绝不仅仅是语法层面的知识，它对数据处理的效率、正确性以及代码的可维护性有着深远的实际意义。

数据校验与错误预防： 在处理外部数据或进行多步数据转换时，我经常会检查数组的
```
shape
```
登录后复制
。比如，一个模型可能要求输入是
```
(batch_size, features)
```
登录后复制
的形状，如果我的数据是
```
(features, batch_size)
```
登录后复制
，那么在传入模型前就必须通过
```
transpose
```
登录后复制
或其他方法调整。如果忘记检查
```
shape
```
登录后复制
，运行时很可能就会遇到维度不匹配的错误，这在深度学习中尤为常见。通过在关键步骤打印或断言
```
shape
```
登录后复制
，可以提前发现并解决问题。
内存管理与性能优化： NumPy数组是内存连续存储的，其
```
shape
```
登录后复制
直接决定了它在内存中占用的空间。理解
```
shape
```
登录后复制
有助于我们预估内存消耗，尤其是在处理大型数据集时。例如，一个
```
(10000, 10000)
```
登录后复制
的
```
float64
```
登录后复制
数组会占用大约800MB内存。如果我发现一个临时数组的
```
shape
```
登录后复制
远大于预期，这可能意味着我的算法效率低下，或者存在内存泄漏的风险。合理地规划数组
```
shape
```
登录后复制
，比如避免不必要的维度扩展，或者在不需要时及时释放大数组，都是优化性能的关键。
高效的数组操作： NumPy的强大之处在于其向量化操作。很多操作都是基于数组的
```
shape
```
登录后复制
来执行的。例如，广播机制（broadcasting）就是NumPy根据数组的
```
shape
```
登录后复制
规则，自动扩展维度较小的数组，使其能够与维度较大的数组进行元素级运算。如果你不理解
```
shape
```
登录后复制
，广播行为可能会让你感到困惑甚至出错。而一旦你掌握了
```
shape
```
登录后复制
，就能巧妙地利用广播来编写简洁高效的代码，避免显式的循环，从而大幅提升计算速度。
算法设计与实现： 在设计图像处理、信号处理或机器学习算法时，数据通常以多维数组的形式存在。算法的每一步操作，比如卷积、池化、矩阵乘法，都对输入数组的
```
shape
```
登录后复制
有严格要求。清楚地知道数据的
```
shape
```
登录后复制
，能帮助我更清晰地构思算法逻辑，例如如何进行切片以提取特定区域，如何重塑（
```
reshape
```
登录后复制
）数组以适应不同的函数接口，或者如何堆叠（
```
stack
```
登录后复制
）多个数组以形成新的批次。在我看来，
```
shape
```
登录后复制
就是NumPy世界的“蓝图”，它定义了数据的结构，也指导着我们如何去构建和操作数据。