在NumPy中,可以使用numpy.append
函数在数组末尾添加数值。下面是使用numpy.append
函数添加数值的完整攻略。
函数介绍
numpy.append(arr, values, axis=None)
函数将“值”添加到“数组”的末尾,返回一个新的数组。其中,参数arr
指定原始数组,values
指定要添加到数组末尾的新值或数组。
-
如果
axis
无指定或为None
,则将输入的数组视为一维数组,返回的数组也为一维数组。 -
如果
axis
有指定值,则输入的数组视为多维数组,返回的数组也将是多维数组。
使用示例
示例1:在一维数组末尾添加一个值
import numpy as np
# 创建一个一维数组
a = np.array([1, 2, 3, 4])
print("原始数组:", a)
# 添加一个值到数组末尾
b = np.append(a, 5)
print("添加值后的新数组:", b)
输出结果如下:
原始数组: [1 2 3 4]
添加值后的新数组: [1 2 3 4 5]
示例2:在二维数组末尾添加一行数据
import numpy as np
# 创建一个二维数组
c = np.array([[1, 2], [3, 4]])
print("原始数组:\n", c)
# 添加一行数据到数组末尾
d = np.append(c, [[5, 6]], axis=0)
print("添加值后的新数组:\n", d)
输出结果如下:
原始数组:
[[1 2]
[3 4]]
添加值后的新数组:
[[1 2]
[3 4]
[5 6]]
在本示例中,axis=0
表示在第一个轴(即行)上添加新值,因此添加的新值需要使用二维数组 [5,6]
包装。