Lesen von CSV-Datei mit Tensorflow

Question

In den Beispielen. Die Variablen für die Spalten im Datensatz werden manuell angegeben. Aber mein Datensatz hat bereits die Namen als Header. Ich möchte sie benutzen. Wie erhalten Sie die Header-Namen einer CSV-Datei mit Tensor-Flow mit Python?

import tensorflow as tf 
filename_queue=tf.train.string_input_producer(
['final_data1.csv'],num_epochs=1) 
#to reada the csv file----- 
print(5) 
reader = tf.TextLineReader(skip_header_lines=1) 
print(4) 
_, csv_row = reader.read_up_to(filename_queue) 
print(type(csv_row)) 
print(3) 
with tf.Session() as sess: 
    print(reader.num_records_produced()) 
    tf.global_variables_initializer() 
    coord = tf.train.Coordinator() 
    threads = tf.train.start_queue_runners(coord = coord)' 

Answer 1

Ich kann nicht genau verstehen, was Sie versuchen zu tun - Sie müssen mehr Details geben.

Wenn Sie mit csv-Datei arbeiten möchten, sollten Sie besser DataFrame-Datenstruktur von Pandas-Bibliothek verwenden. Es gibt die Methode 'read_csv', in der Sie die Zeile auswählen können, die Sie als Überschrift betrachten.

Probieren Sie es aus:

from pandas import read_csv 
df = read_csv('YOUR_DATAFILE.csv',header=0) 
print(df) 

Mit dem Parameter 'Header' können Sie auswählen, welche Zeile im Dataset-Header enthält. In diesem Fall ist es die erste Reihe.