Numpy 数组的大小|极客教程

Numpy 数组的大小

Numpy 是一个强大的 Python 库，主要用于进行大规模的数值计算。本文将详细介绍如何使用 Numpy 来处理和操作数组的大小（size），包括数组的创建、修改大小、以及多维数组的处理等方面。我们将通过一系列的示例代码来深入理解 Numpy 数组的大小相关操作。

1. 创建 Numpy 数组

在 Numpy 中，数组可以通过多种方式创建，每种方式都可以指定数组的大小。以下是一些创建数组的基本方法。

示例代码 1：使用 `np.array` 创建数组

import numpy as np

# 创建一个一维数组
array1 = np.array([1, 2, 3, 4, 5], dtype='int32')
print("数组1：", array1)

Output:

Numpy 数组的大小

示例代码 2：使用 `np.zeros` 创建指定大小的数组

import numpy as np

# 创建一个形状为 (3, 4) 的二维零数组
array2 = np.zeros((3, 4), dtype='float32')
print("数组2：", array2)

Output:

Numpy 数组的大小

示例代码 3：使用 `np.ones` 创建数组

import numpy as np

# 创建一个形状为 (2, 3) 的二维全一数组
array3 = np.ones((2, 3), dtype='int16')
print("数组3：", array3)

Output:

Numpy 数组的大小

示例代码 4：使用 `np.arange` 创建数组

import numpy as np

# 创建一个从 0 到 9 的一维数组
array4 = np.arange(10)
print("数组4：", array4)

Output:

Numpy 数组的大小

示例代码 5：使用 `np.linspace` 创建数组

import numpy as np

# 创建一个从 0 到 1，包含 5 个元素的一维数组
array5 = np.linspace(0, 1, 5)
print("数组5：", array5)

Output:

Numpy 数组的大小

2. 查询 Numpy 数组的大小

了解如何查询 Numpy 数组的大小是进行数组操作的基础。

示例代码 6：查询数组的元素总数

import numpy as np

array6 = np.array([[1, 2, 3], [4, 5, 6]])
print("数组6的元素总数：", array6.size)

Output:

Numpy 数组的大小

示例代码 7：查询数组的形状

import numpy as np

array7 = np.array([[1, 2], [3, 4], [5, 6]])
print("数组7的形状：", array7.shape)

Output:

Numpy 数组的大小

3. 修改 Numpy 数组的大小

修改数组的大小是数据处理中常见的需求，Numpy 提供了多种方式来调整数组的大小。

示例代码 8：使用 `np.reshape` 修改数组形状

import numpy as np

array8 = np.arange(6)
array8_reshaped = array8.reshape((2, 3))
print("重塑后的数组8：", array8_reshaped)

Output:

Numpy 数组的大小

示例代码 9：使用 `np.resize` 修改数组大小

import numpy as np

array9 = np.array([1, 2, 3, 4])
array9_resized = np.resize(array9, (2, 3))
print("调整大小后的数组9：", array9_resized)

Output:

Numpy 数组的大小

示例代码 10：使用 `np.flatten` 将数组转换为一维数组

import numpy as np

array10 = np.array([[1, 2], [3, 4]])
array10_flattened = array10.flatten()
print("展平后的数组10：", array10_flattened)

Output:

Numpy 数组的大小

4. 多维数组的操作

处理多维数组是 Numpy 的一个强大功能，了解如何操作多维数组对于数据分析和科学计算非常重要。

示例代码 11：多维数组的切片

import numpy as np

array11 = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
slice_of_array11 = array11[:, 1:3]
print("数组11的切片：", slice_of_array11)

Output:

Numpy 数组的大小

示例代码 12：多维数组的维度交换

import numpy as np

array12 = np.array([[1, 2, 3], [4, 5, 6]])
array12_transposed = array12.T
print("转置后的数组12：", array12_transposed)

Output:

Numpy 数组的大小

5. 性能优化

在处理大型数组时，性能成为一个关键问题。Numpy 提供了多种方法来优化数组操作的性能。

示例代码 13：使用 `np.copy` 避免在原数组上修改

import numpy as np

array13 = np.array([1, 2, 3, 4, 5])
array13_copy = np.copy(array13)
array13_copy[0] = 99
print("原始数组13：", array13)
print("复制并修改后的数组13：", array13_copy)

Output:

Numpy 数组的大小

示例代码 14：使用布尔索引

import numpy as np

array14 = np.array([1, -1, -2, 3, -4])
positive_elements = array14[array14 > 0]
print("数组14中的正数：", positive_elements)

Output:

Numpy 数组的大小

结论

本文详细介绍了如何使用 Numpy 来处理和操作数组的大小，包括数组的创建、修改大小、多维数组的处理等方面。通过提供的示例代码，我们可以看到 Numpy 在数值计算中的强大功能和灵活性。掌握这些基础知诀将有助于在数据科学和机器学习领域进行更高效的数据处理和分析。

Numpy 数组的大小