python - 从数组中获取所需的列索引并将这些列添加到数据框中

Question

我有一个这样的数据框:

A B C D 
1 2 3 4
1 2 3 4
1 2 3 4
1 2 3 4

我正在编写一段代码来确定我需要的列索引是什么。

RequiredColumns = (skb.get_support(indices=True))
Out[64]: array([ 0,  1], dtype=int64)

我将只采用那些列并将其添加到新数据框中。

为此——我写了这段代码

FeatureSelected_DataFrame = pd.DataFrame()

for i in RequiredColumns:
    ColIndex = i
    ReqColumn = X[[i]]
    FeatureSelected_DataFrame = FeatureSelected_DataFrame.append(ReqColumn)

但这是我得到的输出:FeatureSelected_DataFrame

A   B
1   NaN
1   NaN
NaN 2
NaN 2

我的预期输出是 -

A B  
1 2 
1 2 

Answer 1

我认为您可以将 iloc 位置选择的列附加到列表中，然后 concat:

L = []
RequiredColumns = np.array([0,1])
for i in RequiredColumns:
    ReqColumn = X.iloc[:, i]
    L.append(ReqColumn)

FeatureSelected_DataFrame = pd.concat(L, axis=1)
print (FeatureSelected_DataFrame)
   A  B
0  1  2
1  1  2
2  1  2
3  1  2

更好的非循环解决方案是:

RequiredColumns = np.array([0,1])
df = X.iloc[:, RequiredColumns]
print (df)
   A  B
0  1  2
1  1  2
2  1  2
3  1  2