用tensorflow读取csv文件

时间:2017-07-03 09:38:23

标签: python csv tensorflow tensor

在示例中。手动给出数据集中列的变量。 但我的数据集已经将名称作为标题。我想用它们。如何使用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)'

1 个答案:

答案 0 :(得分:-1)

我无法准确理解您尝试做什么 - 您需要提供更多详细信息。

如果您想使用csv文件,我认为您最好使用pandas库中的DataFrame数据结构。有一个方法'read_csv',您可以在其中选择您认为是标题的行。

试一试:

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

使用参数'header',您可以选择数据集中包含标题的行。在这种情况下,它是第一行。