从索引中检索列名（Python）

Question

我循环到一个包含多个列的大型数据框中，以查找一些元素，同时用计数器跟踪行和列的位置。

当我找到元素时，我想找到与基于计数器的列对应的列标题。有没有一种简单的方法可以做到这一点？

例如，给定：

data = [{'a'：200，'b'：150，'c'：200，'d'：140，'e'：100}]
df6 = pd.DataFrame（数据）

和counter = 1，如何获得相应的列标题“b”？

谢谢！

Answer 1

您可以使用数据框的列索引中的索引选择。

data = [{'a': 200, 'b': 150, 'c': 200, 'd': 140, 'e':100}]
df6 = pd.DataFrame(data)
counter = 1
df6.columns[counter]

输出：

'b'