根据列表从数据集中选择某些要素。

时间:2016-09-16 11:40:11

标签: python pandas scikit-learn

我有一个数据集“h_train”,它有大约26个功能,我有一个列表 H ,它有一些来自数据集的选定功能“h_train”。我只想将这些特征保留在列表H中的“h_train”中。

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

所以我希望将h_train简化为H中的那些功能。我该怎么做?

2 个答案:

答案 0 :(得分:0)

您可以从列表转换为元组数据结构,而不是起诉其运算符

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

答案 1 :(得分:0)

假设h_trainpd.DataFrame你可以做

h_train = h_train[H]