python - 保留已删除列的索引

Question

我想删除数据数组中方差较低的特征。通过使用 scikit-learn，代码将如下所示。

>>> from sklearn.feature_selection import VarianceThreshold

>>> X = [[0, 2, 0, 3], [0, 1, 4, 3], [0, 1, 1, 3]]
>>> selector = VarianceThreshold()
>>> selector.fit_transform(X)
array([[2, 0],
       [1, 4],
       [1, 1]])

我的问题是如何捕获已删除的列索引？假设我想使用它们删除同一列中的另一个数组(上例中的第 0 列和第 3 列)。有什么想法吗？

Answer 1

selector.get_support() 将返回一个数组，其中显示保留哪些列以及删除哪些列。在上述情况下:

selector.get_support()

将返回

array([False,  True,  True, False], dtype=bool)

这意味着原始输入 (X) 的第一个和最后一个索引被删除。