如何获取NumPy数组的长度|极客教程

如何获取NumPy数组的长度

在数据科学和机器学习领域，NumPy是Python编程语言中最基础也是最重要的库之一。NumPy提供了一个强大的数组对象，即ndarray，它是进行高效数值计算的基石。在处理NumPy数组时，经常需要获取数组的长度或大小，这是基本操作之一。本文将详细介绍如何在NumPy中获取数组的长度，并提供多个示例代码来演示不同情况下的操作。

1. 一维数组的长度

获取一维数组的长度是最直接的，可以使用size属性或者len()函数。这里的长度指的是数组中元素的数量。

示例代码 1：使用`len()`函数获取一维数组长度

import numpy as np

# 创建一个一维数组
array_1d = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10])
# 获取数组长度
length = len(array_1d)
print("Length of the array is:", length)

Output:

如何获取NumPy数组的长度

示例代码 2：使用`size`属性获取一维数组长度

import numpy as np

# 创建一个一维数组
array_1d = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10])
# 获取数组长度
length = array_1d.size
print("Length of the array is:", length)

Output:

如何获取NumPy数组的长度

2. 多维数组的长度

对于多维数组，len()函数返回的是最外层维度的大小，而size属性返回的是数组中所有元素的总数。

示例代码 3：获取多维数组的长度

import numpy as np

# 创建一个二维数组
array_2d = np.array([[1, 2, 3], [4, 5, 6]])
# 使用len()函数获取数组的长度
length = len(array_2d)
print("Length of the array is:", length)

Output:

如何获取NumPy数组的长度

示例代码 4：使用`size`属性获取多维数组的总元素数量

import numpy as np

# 创建一个二维数组
array_2d = np.array([[1, 2, 3], [4, 5, 6]])
# 使用size属性获取数组的总元素数量
total_elements = array_2d.size
print("Total number of elements in the array is:", total_elements)

Output:

如何获取NumPy数组的长度

3. 使用`shape`属性获取数组的维度信息

shape属性返回一个元组，表示数组的每个维度的大小。通过这个属性，我们可以更详细地了解数组的结构。

示例代码 5：使用`shape`属性获取数组的维度信息

import numpy as np

# 创建一个三维数组
array_3d = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])
# 使用shape属性获取数组的维度信息
dimensions = array_3d.shape
print("Dimensions of the array are:", dimensions)

Output:

如何获取NumPy数组的长度

4. 特殊情况：空数组的长度

空数组是没有任何元素的数组。对于这种数组，len()函数和size属性都将返回0。

示例代码 6：获取空数组的长度

import numpy as np

# 创建一个空数组
empty_array = np.array([])
# 获取空数组的长度
length = len(empty_array)
print("Length of the empty array is:", length)

Output:

如何获取NumPy数组的长度

示例代码 7：使用`size`属性获取空数组的元素数量

import numpy as np

# 创建一个空数组
empty_array = np.array([])
# 使用size属性获取数组的元素数量
total_elements = empty_array.size
print("Total number of elements in the empty array is:", total_elements)

Output:

如何获取NumPy数组的长度