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提供了丰富的数组操作函数,使得处理多维数组变得更加高效和方便。
极客教程