如何在循环中向空numpy数组以元素形式追加数组？

在NumPy中，动态向数组中追加元素并保持二维结构的最优思路是先利用Python列表收集所有子数组，最后一次性转换为NumPy数组——这比直接在NumPy数组上反复追加效率高得多，因为NumPy数组是固定内存结构，每次扩容都会触发内存重新分配和数据复制，而Python列表的append操作是均摊O(1)的高效操作。

方法一：Python列表收集 + 转NumPy数组（推荐） #

这是最简单且高效的实现方式，完全不需要处理首次追加的特殊情况：

import numpy as np

temp_list = []
for i in range(3):
    temp_list.append(np.array(['x', 'y', 'z']))

# 一次性转换为二维NumPy数组
a = np.array(temp_list)
print(a)

输出结果：

[['x' 'y' 'z']
 ['x' 'y' 'z']
 ['x' 'y' 'z']]

方法二：预分配数组（已知循环次数时最优） #

如果提前知道要追加的子数组数量，直接预分配对应形状的NumPy数组，然后逐个赋值，这是效率最高的方案：

import numpy as np

# 预分配3行3列的字符串数组
a = np.empty((3, 3), dtype=str)
for i in range(3):
    a[i] = np.array(['x', 'y', 'z'])

print(a)

方法三：动态NumPy数组拼接（不推荐，仅作参考） #

如果必须在循环中直接操作NumPy数组，可以通过初始化None来避免冗余判断，每次将子数组转为二维行向量后拼接：

import numpy as np

a = None
for i in range(3):
    current = np.array(['x', 'y', 'z'])
    # 将一维数组转为二维行向量
    current_2d = current[np.newaxis, :]
    if a is None:
        a = current_2d
    else:
        a = np.concatenate([a, current_2d], axis=0)

print(a)

为什么你原来的代码不行？ #

你使用np.append(a, np.array(['x','y','z']))时，默认会将输入数组扁平化后拼接，所以最终得到的是一维数组。如果要指定axis=0实现二维拼接，首次的空数组np.array([])形状为(0,)，和子数组(3,)的形状不匹配，会触发维度错误——这就是需要特殊处理首次追加的原因。

内容的提问来源于stack exchange，提问作者yashaswini srirangarajan