numpy 添加一行|极客教程

numpy 添加一行

在使用numpy进行数组操作时，经常会遇到需要在数组中添加一行的情况。本文将详细介绍如何使用numpy库添加一行到数组中。

numpy库简介

numpy是Python中用于科学计算的核心库之一。它提供了高效的多维数组对象（即numpy数组）以及对这些数组进行操作的各种函数。numpy的核心对象是numpy数组，它是一个固定大小的数组，所有元素都是相同类型的。

创建一个numpy数组

要在numpy数组中添加一行，首先我们需要创建一个numpy数组。可以通过传递Python列表或元组来创建一个numpy数组。

import numpy as np

# 创建一个3x3的numpy数组
arr = np.array([[1, 2, 3],
                [4, 5, 6],
                [7, 8, 9]])
print(arr)

运行以上代码片段，我们将得到如下输出：

[[1 2 3]
 [4 5 6]
 [7 8 9]]

现在我们得到了一个3×3的numpy数组arr。

添加一行到numpy数组

添加一行到numpy数组可以使用numpy.vstack()函数。numpy.vstack()函数用于垂直（按行）堆叠数组。

new_row = np.array([10, 11, 12])
arr = np.vstack((arr, new_row))
print(arr)

运行以上代码片段，我们将得到如下输出：

[[ 1  2  3]
 [ 4  5  6]
 [ 7  8  9]
 [10 11 12]]

可以看到，我们成功地将一行[10, 11, 12]添加到了原始数组arr的末尾。

添加多行到numpy数组

如果要添加多行到numpy数组，可以首先创建一个包含多个新行的numpy数组，然后使用numpy.vstack()函数将其添加到原始数组中。

new_rows = np.array([[13, 14, 15],
                     [16, 17, 18]])
arr = np.vstack((arr, new_rows))
print(arr)

运行以上代码片段，我们将得到如下输出：

[[ 1  2  3]
 [ 4  5  6]
 [ 7  8  9]
 [10 11 12]
 [13 14 15]
 [16 17 18]]

可以看到，我们成功地将新的两行[13, 14, 15]和[16, 17, 18]添加到了原始数组arr的末尾。

总结

本文介绍了如何使用numpy库添加一行或多行到numpy数组中。通过numpy.vstack()函数，我们可以轻松地在numpy数组中添加新的行。numpy提供了丰富的数组操作函数，使得处理多维数组变得更加高效和方便。

numpy 添加一行