讀取張量流的csv文件

Question

在示例中。手動給出數據集中列的變量。 但我的數據集已經有名稱作爲標題。我想用它們。如何使用python使用張量流獲取.csv文件的標題名稱？

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

我無法確切地理解你想要做什麼 - 你需要提供更多細節。

如果你想使用csv-file，我想你最好使用熊貓庫中的DataFrame數據結構。有一種方法'read_csv'，你可以選擇你認爲是行的頭部。

試試：

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

隨着參數「頭」，你可以選擇哪一行數據集中的包含頭文件。在這種情況下，它是第一行。