根據列表從數據集中選擇某些特徵。

我有一個數據集「h_train」其具有約26的功能和我有一個列表ħ其具有從數據集「h_train」一些選定的特徵。我只想讓這些功能中的「h_train」這是目前在列表H.根據列表從數據集中選擇某些特徵。

h_train #Dataset with 26 features [A - Z] 
H = ["A", "B", "C", "D"]

所以我想h_train減少到只有H.這些功能我該怎麼辦呢？

2016-09-16 Sahil

'h_train = h_train [H]'假設' h_train'是一個'DataFrame'。 – Sevanteri

@Sevanteri非常感謝！它確實有效。 :) – Sahil

dataframes通過傳遞列標籤列表來支持子選擇，因此Sevanteri的答案是正確和最簡潔的答案 – EdChum

可以transfrom列表到元組數據結構，比起訴其運營

a = tuple(h_train) 
b = tuple(H) 
c = a & b 
only_H_features_left = list(c)

2016-09-16 11:51:12 dima

假設h_train是pd.DataFrame你可以做

h_train = h_train[H]

2016-09-17 09:37:14 Sevanteri

回答