将多个数组保存到具有列名称的csv文件中

Question

看起来很简单,我在网上找不到任何解决方案.基本上,我有两个数组a和b我要保存到一个CSV文件.它将是两列.我也想添加列名.我在下面的代码中将数组转储到csv.

from np import array, savetxt

a = array([1,2,3,4])
b = array([5,6,7,8])
savetxt('submission2.csv', zip(a,b), delimiter=',', fmt='%f')

我该如何添加列名？我希望csv文件看起来像

Name1 Name2
 1     5
 2     6
 3     7
 4     8

很奇怪这个选项不在savetxt函数中.header选项确实这样做,因为它只是将注释粘贴到第一个单元格中.谢谢.

编辑:数组

Answer 1

您可以轻松地使用pandas包:

import pandas as pd
import numpy as np

a = np.array([1,2,3,4])
b = np.array([5,6,7,8])

df = pd.DataFrame({"name1" : a, "name2" : b})
df.to_csv("submission2.csv", index=False)

Answer 2

使用header选项,如下所示:

>>> import numpy
>>> a = numpy.array([[1,2],[3,4],[5,6]])
>>> numpy.savetxt("foo.csv", a, delimiter=',', header="A,B", comments="")

生成的文件如下所示:

A,B
1.000000000000000000e+00,2.000000000000000000e+00
3.000000000000000000e+00,4.000000000000000000e+00
5.000000000000000000e+00,6.000000000000000000e+00