我有一个
形式的结构数组output = np.zeros(names.size, dtype=[('name', 'U32'), ('r', float),('m',float)])
然后我尝试使用np.savetxt将其保存到csv文件中。我想知道是否有办法将每一列的标签另存为csv文件的标题?
谢谢。
答案 0 :(得分:0)
下面是示例代码:
output = np.zeros(names.size, dtype=[('name', 'U32'), ('r', float),('m',float)])
np.savetxt("foo.csv", output, delimiter=",", header="name,r,m", fmt="%s,%f,%f", comments='')
答案 1 :(得分:0)
您可以尝试类似于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]))
答案 2 :(得分:0)
我发现自己经常遇到这个问题,因此我编写了一个函数,该函数生成要与np.savetxt
一起使用的标头和格式字符串:
您可以在GitHub Gist上找到代码。
我尚未对其进行广泛的测试,但是它可以处理大多数数据类型并生成(可选)自动填充的输出。输出格式正确,易于阅读,支持拖放到excel中,并且在自动检测到字段名和dtypes的情况下(大多),都可以轻松读取。
样本输出:
# x y1 y2 bools verylongnamewithshortcontent bytes objects
0 -25 3.9e+03 1 a b'AvvOkBhFJZIUQsxdg' {'key1': 12423, 'key2': 'asdfjkl;'}
1 255 8e+03 1 a b'SxKvotv' {'key1': 12423, 'key2': 'asdfjkl;'}
2 -211 2.5e+03 0 a b'tvBQXgqqS' {'key1': 12423, 'key2': 'asdfjkl;'}
3 -830 5.7e+02 1 a b'NCFrZHfniaZjeUg' {'key1': 12423, 'key2': 'asdfjkl;'}
4 -3468 8.7e+03 0 a b'RxzuvyKCxKBsz' {'key1': 12423, 'key2': 'asdfjkl;'}
5 4644 2.2e+03 1 a b'yHijSVfCv' {'key1': 12423, 'key2': 'asdfjkl;'}
6 27946 4e+03 0 a b'ywyZeQICJrY' {'key1': 12423, 'key2': 'asdfjkl;'}
7 313770 3.2e+03 1 a b'HBEufqJuASVxHRIxpjd' {'key1': 12423, 'key2': 'asdfjkl;'}
8 -76304 7.7e+02 0 a b'UX' {'key1': 12423, 'key2': 'asdfjkl;'}
9 427810 8.4e+03 0 a b'jmnOEWCvTWg' {'key1': 12423, 'key2': 'asdfjkl;'}
输入/输出dtypes:
[('x', '<i4'), ('y1', '<i4'), ('y2', '<f8'), ('bools', '?'), ('verylongnamewithshortcontent', '<U7'), ('bytes', 'S20'), ('objects', 'O')]
[('x', '<i4'), ('y1', '<i4'), ('y2', '<f8'), ('bools', '<i4'), ('verylongnamewithshortcontent', '<U1'), ('bytes', '<U22'), ('objects', '<U35')]