获取给定元素python的标头列表

Question

我有一个CSV文件，如下所示。

当ID（来自column1的元素）作为输入给出时，我试图获取列表（不一定是Python列表结构，但基本上是所有格式的所有实例），例如，如果我给A1输入，它应该返回Head1,Head2,Head4。同样地，对于C1，它将是Head1,Head3，并且对于ID的其余部分也是如此。我可以使用list(data.columns.values)获取标题列表，而不管ID是什么，但我现在需要对ID进行处理。我非常感谢任何帮助。

编辑 - 我需要标题而不是行值。通过标题我的意思是，如果我将A1作为输入，它应该返回Head1,Head2,Head4而不是2,.,NA。

Answer 1

如果Code是您的索引，那么以下内容将执行您想要的操作：

In [43]:

df[df.index == 'SFDT-09-04-0001'].dropna(axis=1).columns
Out[43]:
Index(['dID', 'cID', 'sID', 'Other_Data'], dtype='object')