python - 按顺序堆叠数组

Question

我从文件中逐行读取data，我想将它们存储在array.

编辑：数据不能用loadtxt().

所以我这样做：

data = array([])
for frame in frames:
    # ....
    # get some lines and make some calculations e.g. final result is
    # line = array([1, 2, 3, 4])
    # ....
    if data.size == 0:
        data = line
    else:
        data = vstack( (data, line) )

这很好用，但是if-clausel 使解决方案看起来很难看。我想知道是否有可能搭上它。

有任何想法吗？

Answer 1

如果元素的数量line是固定的，并且您只想避免“丑陋”的解决方案，您可以这样做：

data = []
for f in frames:
    # do your calculation
    # line = [1, 2, 3, 4]
    data += line
data = np.array(data).reshape((-1,4))