2012-05-30 14 views
8

W jaki sposób mogę mieć genfromtxt, aby zwrócić mi listę nazw kolumn, które były automatycznie pobierane przez names = True? Gdy wykonujęNumpy genfromtxt Nazwy kolumn

data = np.genfromtxt("test.csv",names=True,delimiter=",",dtype=None) 
print data['col1'] 

To drukuje całą kolumnę wartości dla col1. Ale muszę przechodzić przez wszystkie nazwy kolumn; jak mogę to zrobić? Próbowałem data.keys() i różnych innych metod, ale to, co jest zwracane przez genfromtxt, nie wydaje się być obiektem kompatybilnym ze słownikiem. Sądzę, że sam mógłbym podać listę nazw kolumn, ale na dłuższą metę nie da się tego utrzymać.

Wszelkie pomysły?

Odpowiedz

13

genfromtxt zwraca numpy.ndarray.

można uzyskać typ danych z

data.dtype 

lub tylko nazwy z

data.dtype.names 

która jest krotką można iteracyjne nad i dostęp do kolumn, jak chcesz.