Python：将 3D 数组中的值存储到 csv

我有以下问题。我有一个 3D 数组，例如matrix = np.zeros((30,30,100))其中每个条目都是一个坐标并获取一个值。所以matrix [0][0][0]是坐标 x=0,y0,z=0 且值为 0。现在我想将所有值存储在 csv 中，如下所示，其中前 3 行是坐标，第 4 行是相应的值：

numpy 有没有快速的方法来做到这一点？

您可以使用 pandas，它既可以重塑数组并将其另存为 csv。

import numpy as np
import pandas as pd
# create an example array
a = np.arange(24).reshape([2,3,4])
# convert it to stacked format using Pandas
stacked = pd.Panel(a.swapaxes(1,2)).to_frame().stack().reset_index()
stacked.columns = ['x', 'y', 'z', 'value']
# save to disk
stacked.to_csv('stacked.csv', index=False)

否则，您可以申请

np.ravel()

到您的数组，然后使用其中一种方法恢复索引在这个问题中.