Python如何查看数组纬度|极客笔记

Python如何查看数组纬度

在Python中，可以使用NumPy库来处理数组数据。在NumPy中，数组又被称为ndarray（N-dimensional array），即多维数组。在处理数据时，经常需要查看数组的维度信息，以便更好地理解和操作数据。

本文将详细介绍在Python中如何查看数组的维度信息，以及如何利用NumPy库中的相关函数来实现这一功能。

在Python中，可以通过以下几种方法来查看数组的维度信息：

在NumPy中，ndarray具有一个shape属性，用来表示数组的维度信息。shape是一个元组，其中的每个元素表示对应维度的大小。通过访问数组的shape属性，可以轻松地查看数组的维度信息。

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
print(arr.shape)

运行以上代码，输出为：

(2, 3)

上面的代码创建了一个二维数组arr，并通过打印其shape属性来查看数组的维度信息。可以看到，数组arr的shape为(2, 3)，表示该数组有两行三列。

除了shape属性外，ndarray还具有一个ndim属性，用来表示数组的维度数量。通过访问数组的ndim属性，可以直接查看数组的维度数量。

import numpy as np

arr = np.array([1, 2, 3, 4])
print(arr.ndim)

运行以上代码，输出为：

上面的代码创建了一个一维数组arr，并通过打印其ndim属性来查看数组的维度数量。可以看到，数组arr的ndim为1，表示该数组是一个一维数组。

另外，ndarray还具有一个dtype属性，用来表示数组中元素的数据类型。尽管dtype属性并不直接提供数组的维度信息，但通过查看数组中元素的数据类型，也可以对数组的维度信息进行一定的推断。

import numpy as np

arr = np.array([1, 2, 3])
print(arr.dtype)

运行以上代码，输出为：

int64

上面的代码创建了一个一维数组arr，并通过打印其dtype属性来查看数组中元素的数据类型。可以看到，数组arr中的元素是整数类型（int64）。

除了查看一维数组的维度信息外，我们还可以查看多维数组的维度信息。下面通过一个示例来演示如何查看多维数组的维度信息：

import numpy as np

arr = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])
print(arr.shape)
print(arr.ndim)

运行以上代码，输出为：

(2, 2, 2)
3

上面的代码创建了一个三维数组arr，并通过打印其shape属性和ndim属性来查看数组的维度信息。可以看到，数组arr的shape为(2, 2, 2)，表示该数组有两个二维数组，每个二维数组中有两行两列；ndim为3，表示该数组是一个三维数组。

本文介绍了在Python中如何查看数组的维度信息，并通过示例代码演示了如何利用NumPy库中的相关属性来实现这一功能。通过查看数组的维度信息，可以更好地理解和操作数组数据，进而提高数据处理的效率和准确性。