numpy数组如何添加新元素

要向 NumPy 数组中添加新元素，可以使用 NumPy 提供的 `append()` 函数。该函数可以将新元素添加到数组的末尾，返回一个新的数组。下面是一个例子：

import numpy as np

# 创建一个 1 维数组
arr = np.array([1, 2, 3])

# 添加一个新元素
new_arr = np.append(arr, 4)

print(new_arr)  # 输出 [1 2 3 4]

如果要添加多个元素，可以将它们作为一个列表传递给 `append()` 函数：

import numpy as np

# 创建一个 1 维数组
arr = np.array([1, 2, 3])

# 添加多个新元素
new_arr = np.append(arr, [4, 5, 6])

print(new_arr)  # 输出 [1 2 3 4 5 6]

需要注意的是，每次调用 `append()` 函数都会创建一个新的数组，因此频繁添加元素会导致性能下降。如果需要高效地添加元素，建议在创建数组时就指定好其大小，然后使用索引直接赋值。例如：

import numpy as np

# 创建一个 1 维数组，初始大小为 3
arr = np.zeros(3)

# 直接赋值添加元素
arr[0] = 1
arr[1] = 2
arr[2] = 3

# 添加一个新元素
arr = np.append(arr, 4)

print(arr)  # 输出 [1. 2. 3. 4.]