numpy 添加一行

2024年06月08日 numpy 添加一行 极客笔记

numpy 添加一行

在使用 numpy 进行数组操作时,有时候需要在已有数组的末尾添加一行数据。本文将详细介绍如何使用 numpy 来添加一行数据到现有的数组中。

numpy 简介

NumPy 是 Python 中用于科学计算的一个重要库,提供了高效的多维数组对象和各种用于数组操作的工具。在 NumPy 中,最基本的对象是 ndarray,即 n-dimensional array,它是一个由相同类型的元素组成的多维数组。

创建一个示例数组

首先,让我们创建一个示例数组,假设我们有一个二维数组 arr

import numpy as np

arr = np.array([[1, 2, 3],
                [4, 5, 6]])
print("原始数组:")
print(arr)

输出为:

原始数组:
[[1 2 3]
 [4 5 6]]

现在我们有一个包含两行三列的二维数组 arr

添加一行数据

要向数组中添加一行数据,可以使用 numpy.vstack() 函数,该函数用于在垂直方向堆叠数组。我们可以先创建一个新的一维数组,然后使用 vstack() 函数将其添加到原始数组中。

new_row = np.array([7, 8, 9])
new_arr = np.vstack((arr, new_row))

print("添加新行后的数组:")
print(new_arr)

运行结果为:

添加新行后的数组:
[[1 2 3]
 [4 5 6]
 [7 8 9]]

可以看到,新的一行 [7, 8, 9] 被成功添加到了原始数组 arr 的末尾。

修改原始数组

上述的方式是将新的一行添加到原始数组的末尾得到一个新的数组,如果需要直接在原始数组上进行修改的话,可以使用 numpy.append() 函数。append() 函数可以沿着指定的轴(默认是最后一个轴)在数组的末尾添加值。

arr = np.append(arr, [new_row], axis=0)

print("修改后的原始数组:")
print(arr)

输出为:

修改后的原始数组:
[[1 2 3]
 [4 5 6]
 [7 8 9]]

同样,新的一行 [7, 8, 9] 被成功添加到了原始数组 arr 的末尾。但这种方式直接修改了原始数组。

总结

本文介绍了如何使用 numpy 来向已有数组中添加一行数据。通过 numpy.vstack() 函数可以创建一个新的数组,将新的一行添加到原始数组的末尾;通过 numpy.append() 函数可以直接在原始数组上进行修改。这些方法在处理数组数据时非常有用,尤其是在数据处理和科学计算领域。

本文链接:http://so.lmcjl.com/news/6176/

展开阅读全文