将堆叠的 NumPy 数组保存到文本时出错

Question

我将来自三个不同输入 arrays 的列数据组合成一个新的 csv。 为此，我使用了 NumPy 堆栈 function。 现在我有一个 [12,3] 堆叠的 NumPy 数组，我正在尝试将其导出到 csv。

VI_Samples_v4 = numpy.stack((samplename,sample_start_date,sample_type_code), axis =1)

当我使用 numpy 时保存文本

numpy.savetxt('array.csv', [VI_Samples_v4], fmt='%s', delimiter=',')

我收到此错误：“ValueError：预期的 1D 或 2D 数组，得到了 3D 数组。” 我相当肯定它是一个二维数组。 当我向 Numpy 询问阵列形状时，我得到 (12,3)。

VI_Samples_v4.shape

那我的错误是什么？

Answer 1

问题是您在调用numpy.savetxt时放置在数组周围的方括号。 通过传递包含二维数组的列表，您将导致numpy.savetxt将其读取为 3D 数组。 只需传递不带方括号的数组，如下所示：

numpy.savetxt('array.csv', VI_Samples_v4, fmt='%s', delimiter=',')