Save a numpy array to csv with a string header

Question

I have a large-ish numpy array containing floats, which I save as a csv file with np.savetxt("myFile.csv", myArray, delimiter=",")

Now, as the array has many columns and it can become hard to remember what is what, I want to add a string header to the array before exporting it. Since numpy doesn't accept strings in float arrays, is there a trick to accomplish this?

[Solution] Thanks to Cyborg's advice, I managed to make this work installing Pandas.

import pandas as pd
df = pd.DataFrame(A) # A is a numpy 2d array
df.to_excel("A.xls", header=C,index=False) # C is a list of string corresponding to the title of each column of A

cyborg · Accepted Answer · 2013-11-30 23:55:54Z

18

The header argument (the docs):

numpy.savetxt(fname, X, delimiter=' ', header='')

But you may prefer Pandas if you are actually dealing with a table.

answered Nov 30, 2013 at 23:55

cyborg

10.2k4 gold badges40 silver badges56 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

Lucien S. Over a year ago

I am indeed dealing with a 2d array, this is why I believed the 'header' attribute wouldn't work. I'll try panda! Cheers!

mins Over a year ago

To save current fields as headers: np.savetxt(path, X, delimiter=',', header= ','.join(X.dtype.names))

Collectives™ on Stack Overflow

Save a numpy array to csv with a string header

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related