如何在循环中逐行拼接numpy数组，无需额外条件判断即可实现？

你遇到的报错本质是初始的空Python列表没有匹配的维度信息，numpy.vstack无法完成对齐拼接。以下是几种更简洁的实现方案：

方案1：最高效的常规实现（无需循环内判断） #

numpy的拼接操作每次都会重新申请内存，循环内反复调用会严重拉低性能，更推荐先把所有行存在原生Python列表中，循环结束后一次性转为numpy数组：

import numpy as np

a = []
for i in range(3):
    line = np.array([i, i + 1, i + 2])
    a.append(line)
a = np.array(a)
print(a)

该写法无需任何条件判断，逻辑简单易维护，循环次数越多性能优势越明显。

方案2：循环内单行拼接写法（满足单行需求） #

如果必须在循环内逐行完成拼接，可以用三元表达式把条件判断浓缩为一行：

import numpy as np

a = []
for i in range(3):
    line = np.array([i, i + 1, i + 2])
    a = line[np.newaxis, :] if len(a) == 0 else np.vstack([a, line])
print(a)

方案3：初始化带维度的空numpy数组，无需判断直接拼接 #

如果提前已知每行的元素个数，可以直接初始化0行N列的空numpy数组，后续直接调用vstack不会报错：

import numpy as np

# 初始化0行3列的空数组，dtype和后续行数据类型保持一致即可
a = np.empty((0, 3), dtype=np.float64)
for i in range(3):
    line = np.array([i, i + 1, i + 2], dtype=np.float64)
    a = np.vstack([a, line])
print(a)

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