python - 使用带有标题的 np.savetxt 保存结构化的 numpy 数组

Question

我有一个结构数组的形式

output = np.zeros(names.size, dtype=[('name', 'U32'), ('r', float),('m',float)])

然后我尝试使用 np.savetxt 将其保存到 csv 文件中。我想知道是否有办法将每列的标签保存为 csv 文件的标题？

先感谢您。

Answer 1

您可以尝试类似于 this SO answer 的解决方案旋转数据

dtypes = [('name', 'U32'), ('r', float),('m',float)]
a = np.zeros(5, dtype=dtypes)
b = numpy.vstack(map(list, a))

在 recarray 元组上映射列表，然后垂直堆叠它们。

然后你可以做以下...

names = [n for n, t in dtypes]
numpy.savetxt('test.csv', b, header=','.join(names), fmt=','.join(['%s']*b.shape[1]))