在示例中。手动给出数据集中列的变量。 但我的数据集已经将名称作为标题。我想用它们。如何使用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)'
答案 0 :(得分:-1)
我无法准确理解您尝试做什么 - 您需要提供更多详细信息。
如果您想使用csv文件,我认为您最好使用pandas库中的DataFrame数据结构。有一个方法'read_csv',您可以在其中选择您认为是标题的行。
试一试:
from pandas import read_csv
df = read_csv('YOUR_DATAFILE.csv',header=0)
print(df)
使用参数'header',您可以选择数据集中包含标题的行。在这种情况下,它是第一行。