如何从python中的索引取回列名？

Question

我有数据框，

从我转换为ndarray的

X = df.iloc[:].values

然后我进行了特征选择，这给了我一些列[3,45,67,89,101,145，...]

因为有很多列，我可能知道如何获取数据框中的相应列名称

我可以使用下面的代码做相反的事情

cols = ['Amount','Savings_Amount','Loan_Term']
[df.columns.get_loc(c) for c in cols if c in df]

但是我需要从索引中查找名称。

Answer 1

cols = ['Amount','Savings_Amount','Loan_Term']
[df.columns.get_loc(c) for c in cols if c in df]

获得列的所有索引后，您只需使用以下代码即可获取所有列名称的列表

ColumnNames = list(my_dataframe.columns.values)

[CName for indx,Cname in columnNames if indx in AllFilteredIndexes return Cname]

您只需返回与已过滤功能选定索引的列名index中匹配的所有索引。

Answer 2

不是，您想要什么：

[df.columns[i] for i in indexes]

Answer 3

您可以将索引传递给列索引

names = df.columns[ix].values

其中ix是您的向量索引