2009-10-09 22 views
5

Czy ktoś może wskazać, co robię źle tutaj?python numpy savetxt

import numpy as np 

a = np.array([1,2,3,4,5],dtype=int) 
b = np.array(['a','b','c','d','e'],dtype='|S1') 

np.savetxt('test.txt',zip(a,b),fmt="%i %s") 

Wyjście jest:

Traceback (most recent call last): 
    File "loadtxt.py", line 6, in <module> 
    np.savetxt('test.txt',zip(a,b),fmt="%i %s") 
    File "/Users/tom/Library/Python/2.6/site-packages/numpy/lib/io.py", line 785, in savetxt 
    fh.write(format % tuple(row) + '\n') 
TypeError: %d format: a number is required, not numpy.string_ 

Odpowiedz

12

musisz skonstruować Ci tablicę inaczej:

z = np.array(zip([1,2,3,4,5], ['a','b','c','d','e']), dtype=[('int', int), ('str', '|S1')]) 
np.savetxt('test.txt', z, fmt='%i %s') 

gdy jesteś przejazdem sekwencję , savetext wykonuje asarray(sequence) call, a wynikowa tablica jest typu |S4, czyli wszystkie elementy są ciągami! właśnie dlatego widzisz ten błąd.

1

Myślę, że problem masz jest, że przechodzą przez krotki ciąg formatowania i nie można interpretować krotki z% I. Spróbuj użyć FMT = „% s”, zakładając, że to jest to, czego szukasz na wyjściu:

1 a 
2 b 
3 c 
4 d 
5 e 
+0

To po prostu niepoprawne. 'fmt ="% s "' działa z zupełnie innych powodów, 'fmt ="% s% s "' działa też, btw. – SilentGhost

+0

Masz rację, jak tylko napisałem, zdałem sobie sprawę, że zadziałało, ale nie z powodu, o którym myślałem. Mój błąd. Post przez SilentGhost jest znacznie lepszy. Dzięki. – dwelch

4

Jeśli chcesz zapisać plik CSV można również użyć rec2csv function (wliczone w matplotlib.mlab)

>>> from matplotlib.mlab import rec2csv 
>>> rec = array([(1.0, 2), (3.0, 4)], dtype=[('x', float), ('y', int)]) 
>>> rec = array(zip([1,2,3,4,5], ['a','b','c','d','e']), dtype=[('x', int), ('y', str)]) 
>>> rec2csv(rec, 'recordfile.txt', delimiter=' ') 

nadzieją, programiści jeden dzień pylab będzie realizować godnej poparcia zapisu plików csv.