将多个数组保存到列名为csv的文件中

看起来很简单，我在网上找不到任何解决方案。基本上，我有两个数组a和b，我想保存到一个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

它就是这么奇怪，这个选项是不是在savetxt功能。 header选项会这样做，因为它只是将注释粘贴到第一个单元格中。谢谢。

编辑：数组

来源

2015-11-03 Koba

第一行是在那里你会放列名;你想要排名吗？你能举一个你想要文件看起来像样的样本吗？ –

@BurhanKhalid添加详细信息 – Koba

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

来源

2015-11-03 06:15:45

是否有将列添加到现有csv文件的方法？另外，我怎样才能摆脱电子？ – duldi

，你可以这样做：

import np import array, savetxt 

a = array([1,2,3,4]) 
b = array([5,6,7,8]) 
f = open("submission2.csv", "w") 
f.write("{},{}\n".format("Name1", "Name2")) 
for x in zip(a, b): 
    f.write("{},{}\n".format(x[0], x[1])) 
f.close()

来源

2015-11-03 06:05:45 Hackaholic

您可以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)

来源

2015-11-03 06:13:17

注意savetxt（和loadtxt）也需要文件句柄。

因此，如果你想要一个更先进的头，你可以这样做：

a = array([1,2,3,4]) 
b = array([5,6,7,8]) 
with open('submission2.csv','w') as f: 
    f.write('# This is a very complex header\n') 
    f.write('A,B\n') 
    savetxt(f, zip(a,b), delimiter=',', fmt='%f')

或者，正如已经指出的那样，使用header=str(...)说法。

来源

2015-11-03 06:26:21 zeroth

我已经找到了保存多个numpy的一维数组作为列的解决方案：

import numpy as np 
data = [] 
for i in single_np_arrays: 
    data.append(i) 
data = np.array(data).T #transpose the array to have proper columns 
np.savetxt('columns_from_np_arrays.csv',data,delimiter=',')

来源

2018-01-12 10:38:19 Dawid

将多个数组保存到列名为csv的文件中

回答

相关问题