Python输出数组形状

在Python中，数组是一种常见的数据结构，用于存储一系列相同类型的数据。数组的形状是指数组的维度和每个维度的大小。在本文中，我们将介绍如何使用Python输出数组的形状，并提供相关的代码示例。

数组的形状

数组的形状由其维度和每个维度的大小组成。维度是指数组的秩或轴的数量，而每个维度的大小是指该维度上的元素数量。例如，一个二维数组的形状可以表示为(m, n)，其中m表示行数，n表示列数。

在Python中，可以使用shape属性来获取数组的形状。对于一维数组，形状只包含一个元素，即数组的长度。对于二维或多维数组，形状将包含每个维度的大小。

使用shape属性输出数组形状

下面是一个使用shape属性输出数组形状的示例代码：

import numpy as np

# 创建一个二维数组
arr = np.array([[1, 2, 3], [4, 5, 6]])

# 输出数组的形状
print(arr.shape)

输出结果为(2, 3)，表示该二维数组有2行和3列。

使用ndim属性获取数组的维度

除了使用shape属性获取数组形状外，还可以使用ndim属性获取数组的维度。维度是指数组的轴的数量。

下面是一个使用ndim属性获取数组维度的示例代码：

import numpy as np

# 创建一个三维数组
arr = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])

# 输出数组的维度
print(arr.ndim)

输出结果为3，表示该三维数组有3个维度。

使用len()函数获取数组长度

对于一维数组，可以使用len()函数获取数组的长度，即数组中元素的个数。

下面是一个使用len()函数获取一维数组长度的示例代码：

import numpy as np

# 创建一个一维数组
arr = np.array([1, 2, 3, 4, 5])

# 输出数组的长度
print(len(arr))

输出结果为5，表示该一维数组有5个元素。

使用mermaid语法绘制状态图

在本文中，我们使用mermaid语法绘制一个简单的状态图，以说明数组形状的概念。状态图可以通过stateDiagram标识符来创建。

下面是一个使用mermaid语法绘制状态图的示例代码：

stateDiagram
    [*] --> Array
    Array --> Shape
    Shape --> Dimension
    Shape --> Length

上述代码将绘制一个状态图，其中包含了数组、形状、维度和长度之间的关系。

总结

在本文中，我们介绍了如何使用Python输出数组的形状。通过使用shape属性、ndim属性和len()函数，我们可以轻松地获取数组的形状、维度和长度。此外，我们还使用mermaid语法绘制了一个状态图，以帮助理解数组形状的概念。希望本文能够帮助您更好地理解和应用数组的形状。

代码示例：

import numpy as np

# 创建一个二维数组
arr = np.array([[1, 2, 3], [4, 5, 6]])

# 输出数组的形状
print(arr.shape)

# 创建一个三维数组
arr = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])

# 输出数组的维度
print(arr.ndim)

# 创建一个一维数组
arr = np.array([1, 2, 3, 4, 5])

# 输出数组的长度
print(len(arr))

状态图：

stateDiagram
    [*] --> Array
    Array --> Shape
    Shape --> Dimension
    Shape --> Length

参考资料：

  • [NumPy User Guide](https