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

虽然看起来很简单，但我在网上找不到任何解决问题的方法。基本上，我有两个数组要保存到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

奇怪的是，此选项不在a函数中。b选项会这样做，因为它只是将注释粘贴到第一个单元格中。谢谢。
编辑：阵列

最佳答案

使用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