Python/Numpy - Salva array con colonne e titoli di riga

Question

Desidero salvare un array 2D in un file CSV con informazioni di "intestazione" di riga e colonna (come una tabella). So che potrei usare l'argomento header su numpy.savetxt per salvare i nomi delle colonne, ma c'è un modo semplice per includere anche qualche altro array (o lista) come prima colonna di dati (come i titoli delle righe)?

Di seguito è riportato un esempio di come lo faccio attualmente. C'è un modo migliore per includere quei titoli di riga, forse qualche trucco con salvataggio di cui non sono a conoscenza?

import csv 
import numpy as np 

data = np.arange(12).reshape(3,4) 
# Add a '' for the first column because the row titles go there... 
cols = ['', 'col1', 'col2', 'col3', 'col4'] 
rows = ['row1', 'row2', 'row3'] 

with open('test.csv', 'wb') as f: 
    writer = csv.writer(f) 
    writer.writerow(cols) 
    for row_title, data_row in zip(rows, data): 
     writer.writerow([row_title] + data_row.tolist()) 

Answer 1

Forse si preferisce fare qualcosa di simile:

# Column of row titles 
rows = np.array(['row1', 'row2', 'row3'], dtype='|S20')[:, np.newaxis] 
with open('test.csv', 'w') as f: 
    np.savetxt(f, np.hstack((rows, data)), delimiter=', ', fmt='%s') 

Questo è implicitamente conversione data ad un array di stringhe, e dura circa 200 ms per ogni milione di articoli nel mio computer.

Il dtype '|S20' significa stringhe di venti caratteri. Se è troppo basso, i numeri avranno tagliato:

>>> np.asarray([123], dtype='|S2') 
array(['12'], 
    dtype='|S2') 

Un'altra opzione, che dal mio test limitato è più lento, ma ti dà molto più controllo e non ha il problema di taglio sarebbe utilizzando np.char.mod, come

# Column of row titles 
rows = np.array(['row1', 'row2', 'row3'])[:, np.newaxis] 
str_data = np.char.mod("%10.6f", data) 
with open('test.csv', 'w') as f: 
    np.savetxt(f, np.hstack((rows, str_data)), delimiter=', ', fmt='%s')