如何使用Numpy来添加行到一个现有的数组中|极客笔记

如何使用Numpy来添加行到一个现有的数组中

在数据处理和科学计算中，经常需要对数组进行操作，比如添加行或列。Numpy是Python中一个非常强大的库，它提供了一系列用于处理数组的函数。本文将详细介绍如何使用Numpy来添加行到一个现有的数组中。

1. 创建初始数组

在我们开始添加行之前，首先需要有一个初始的数组。这里我们将使用numpy.array来创建一个二维数组。

import numpy as np

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

Output:

如何使用Numpy来添加行到一个现有的数组中

2. 使用`numpy.append`添加行

Numpy提供了一个非常方便的函数numpy.append来添加行或列。这个函数的第一个参数是原始数组，第二个参数是要添加的内容，而axis参数用来指定是添加行还是列。

示例1: 添加一行到二维数组

import numpy as np

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

# 创建要添加的行
new_row = np.array([7, 8, 9])

# 添加行
result_array = np.append(initial_array, [new_row], axis=0)
print(result_array)

Output:

如何使用Numpy来添加行到一个现有的数组中

示例2: 添加多行到二维数组

import numpy as np

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

# 创建要添加的多行
new_rows = np.array([[7, 8, 9], [10, 11, 12]])

# 添加多行
result_array = np.append(initial_array, new_rows, axis=0)
print(result_array)

Output:

如何使用Numpy来添加行到一个现有的数组中

3. 使用`numpy.vstack`添加行

另一个添加行的方法是使用numpy.vstack（垂直堆叠）。这个方法将数组沿着垂直方向堆叠。

示例3: 使用`vstack`添加一行

import numpy as np

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

# 创建要添加的行
new_row = np.array([7, 8, 9])

# 使用vstack添加行
result_array = np.vstack((initial_array, new_row))
print(result_array)

Output:

如何使用Numpy来添加行到一个现有的数组中

示例4: 使用`vstack`添加多行

import numpy as np

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

# 创建要添加的多行
new_rows = np.array([[7, 8, 9], [10, 11, 12]])

# 使用vstack添加多行
result_array = np.vstack((initial_array, new_rows))
print(result_array)

Output:

如何使用Numpy来添加行到一个现有的数组中

4. 使用`numpy.concatenate`添加行

numpy.concatenate也是一个用于合并数组的函数，它可以在指定的轴上连接多个数组。

示例5: 使用`concatenate`添加一行

import numpy as np

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

# 创建要添加的行
new_row = np.array([[7, 8, 9]])

# 使用concatenate添加行
result_array = np.concatenate((initial_array, new_row), axis=0)
print(result_array)

Output:

如何使用Numpy来添加行到一个现有的数组中

示例6: 使用`concatenate`添加多行

import numpy as np

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

# 创建要添加的多行
new_rows = np.array([[7, 8, 9], [10, 11, 12]])

# 使用concatenate添加多行
result_array = np.concatenate((initial_array, new_rows), axis=0)
print(result_array)

Output:

如何使用Numpy来添加行到一个现有的数组中

5. 使用切片和赋值添加行

虽然不是最常用的方法，但我们也可以通过切片和赋值来添加行到numpy数组中。

示例7: 使用切片添加一行

import numpy as np

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

# 创建要添加的行
new_row = np.array([7, 8, 9])

# 扩展数组以添加新行
result_array = np.zeros((3, 3), dtype=int)
result_array[:2, :] = initial_array
result_array[2, :] = new_row
print(result_array)

Output:

如何使用Numpy来添加行到一个现有的数组中

示例8: 使用切片添加多行

import numpy as np

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

# 创建要添加的多行
new_rows = np.array([[7, 8, 9], [10, 11, 12]])

# 扩展数组以添加新行
result_array = np.zeros((4, 3), dtype=int)
result_array[:2, :] = initial_array
result_array[2:, :] = new_rows
print(result_array)

Output:

如何使用Numpy来添加行到一个现有的数组中