获取给定元素的标题列表

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')