python - 使用 python 的文本文件标题中的条目列表

Question

我有一个 .txt 文件，其中包含文本标题和数字数据。我正在使用 python 2.7，并在工作中使用 pandas 和 numpy。文件结构如下图所示:

该文件的数据可以从here获取。在这个文件中，我想获取所有标签的列表。例如，在上面显示的图片中，我希望列表如下所示:

[Tag1, Tag1, Tag1, Tag5, Tag5, Tag6, Tag6]

目前，我正在使用以下方式读取文件:

df = pd.read_csv('dum.txt',sep='\t', header=[0,1], index_col=0)

当我尝试 lst = df.columns.levels[1] 时，我得到 Index([u'Tag1', u'Tag5', u'Tag6'], dtype= 'object', name=u'Tag') 作为我的输出而不是我想要的列表。

如何获取问题中的标签列表，即 [Tag1, Tag1, Tag1, Tag5, Tag5, Tag6, Tag6] ？提前致谢。

Answer 1

您可以使用get_level_values(1)而不是 levels[1]，然后使用 tolist() 转换为列表:

>>> df.columns.get_level_values(1).tolist()
['Tag1', 'Tag1', 'Tag1', 'Tag5', 'Tag5', 'Tag6', 'Tag6']

原因是，如您所见，levels[1] 将为您提供所有唯一级别的列表，而 get_level_values 实际上会返回:

Return vector of label values for requested level, equal to the length of the index