python - 用tensorflow读取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 文件，我认为您最好使用 pandas 库中的 DataFrame 数据结构。有“read_csv”方法，您可以在其中选择您认为作为标题的行。

试试看：

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

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