如何在制表符分隔的数据框中获取列名?

时间:2017-04-01 18:32:15

标签: python csv pandas

我有一个制表符分隔文件,我在pandas dataframe中提取,如下所示:

import pandas as pd
data1 = pd.DataFrame.from_csv(r"C:\Users\Ashish\Documents\indeed_ml_dataset\train.tsv", sep="\t")

DATA1

以下是data1的外观:

data1 dataframe pandas

现在,我想查看列名标签。我不知道是否应该将其称为专栏,但我尝试使用标准访问它:

data2=data1[['tags']]

但错误了。我使用indexloc尝试了其他几项,但所有这些都失败了。有什么建议吗?

2 个答案:

答案 0 :(得分:1)

要解决此问题,您需要通过重置从索引中删除描述。试试以下内容:

data2 = data1.reset_index()
data2['tags']

然后,您就可以通过“标签”进行选择。

答案 1 :(得分:1)

尝试使用 pd.read_csv 而不是 pd.DataFrame.from_csv 来阅读您的数据,因为它默认将第一列作为索引。 有关详细信息,请参阅pandas网站上的此文档:http://pandas.pydata.org/pandas-docs/stable/generated/pandas.DataFrame.from_csv.html