极客笔记numpy append
在使用Python进行数据处理和数值计算时，经常会遇到需要在数组末尾添加元素的情况。numpy库提供了一个非常方便的函数np.append()来实现这个功能。本文将详细介绍np.append()的用法，并且讨论一些注意事项。
np.append()函数的基本用法
np.append()函数的基本语法如下：
numpy.append(arr, values, axis=None)
参数说明：
arr：要添加元素的数组values：要添加的元素，可以是一个值，也可以是一个数组axis：表示添加的方向。当axis=None时，np.append()会将values在arr的末尾添加成一维数组；当axis=0时，np.append()会将values添加成行；当axis=1时，np.append()会将values添加成列。
接下来，我们来看一些示例代码，演示np.append()函数的基本用法。
示例1：在一维数组末尾添加元素
import numpy as np
arr = np.array([1, 2, 3])
new_arr = np.append(arr, 4)
print(new_arr)
运行结果：
[1 2 3 4]
在这个示例中，我们定义了一个一维数组arr，然后使用np.append()函数在数组末尾添加了一个元素4，最终得到新的数组new_arr。
示例2：在二维数组末尾添加一行
import numpy as np
arr = np.array([[1, 2, 3],
[4, 5, 6]])
new_row = np.array([7, 8, 9])
new_arr = np.append(arr, [new_row], axis=0)
print(new_arr)
运行结果：
[[1 2 3]
[4 5 6]
[7 8 9]]
在这个示例中，我们定义了一个二维数组arr，然后使用np.append()函数在数组末尾添加了一行[7, 8, 9]，最终得到新的数组new_arr。
示例3：在二维数组末尾添加一列
import numpy as np
arr = np.array([[1, 2],
[3, 4]])
new_col = np.array([[5],
[6]])
new_arr = np.append(arr, new_col, axis=1)
print(new_arr)
运行结果：
[[1 2 5]
[3 4 6]]
在这个示例中，我们定义了一个二维数组arr，然后使用np.append()函数在数组末尾添加了一列[[5], [6]]，最终得到新的数组new_arr。
np.append()函数的注意事项
在使用np.append()函数时，有一些注意事项需要注意。
1. 原数组不会改变
值得注意的是，在调用np.append()函数后，原数组并不会被改变，函数会返回一个新的数组。如果想要改变原数组，需要将函数返回的新数组赋值给原数组。
import numpy as np
arr = np.array([1, 2, 3])
new_arr = np.append(arr, 4)
print(arr) # [1 2 3]
print(new_arr) # [1 2 3 4]
arr = new_arr
print(arr) # [1 2 3 4]
2. 添加多个元素
如果要一次添加多个元素，可以将这些元素组成一个数组传入np.append()函数的values参数。
import numpy as np
arr = np.array([1, 2, 3])
new_elements = np.array([4, 5, 6])
new_arr = np.append(arr, new_elements)
print(new_arr) # [1 2 3 4 5 6]
3. 添加多行或多列
要添加多行或多列时，需要将这些行或列组成一个二维数组传入np.append()函数的values参数。
import numpy as np
arr = np.array([[1, 2],
[3, 4]])
new_rows = np.array([[5, 6],
[7, 8]])
new_arr = np.append(arr, new_rows, axis=0)
print(new_arr)
4. 数据类型一致性
在使用np.append()函数时，要注意被添加的元素与原数组的数据类型应保持一致，否则可能会导致数据类型转换。
import numpy as np
arr = np.array([1, 2, 3])
new_element = 4.0
new_arr = np.append(arr, new_element)
print(new_arr) # [1. 2. 3. 4.]
总结
通过本文的介绍，我们了解了numpy库中np.append()函数的基本用法和一些注意事项。使用np.append()函数可以方便地在数组末尾添加元素，无论是添加单个元素、一行或一列，还是添加多个元素、多行或多列，都可以轻松实现。在实际应用中，根据需要选择合适的参数，灵活运用np.append()函数，可以提高数据处理和计算的效率。