python - 如何将图像的每一帧的二维数组写入csv？

Question

我有一个 getX() ，它给出一个 2D array(shape=(18,2)) 作为图像每一帧的输出。我想将这个不断更新的数组添加到 CSV 中。

我使用了以下代码。但它只给了我一行(可能只有一帧)。

with open("data.csv",mode ='w') as csvfile:
            wr = csv.writer(csvfile, quoting = csv.QUOTE_ALL,dialect="excel")
            get_val = oneObject.getX(currentFrameIndex)
            for  humanPos in get_val:
                wr.writerow(humanPos)

Answer 1

为什么不累积子数组，然后使用 np.savetxt 写入 CSV 一次？

arrs = []
for currentFrameIndex in listOfFrameIndices:  # Demonstrative only.
    arrs.append(oneObject.getX(currentFrameIndex))
# Or,
# arrs = np.array([oneObject.getX(c) for c in listOfFrameIndices])

np.savetxt('file.csv', np.array(arrs), delimiter=',')