根据列表从数据集中选择某些特征



我有一个数据集"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中的那些特征。我怎么做呢?

您可以将列表转换为元组数据结构,然后使用其操作符

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

假设h_trainpd.DataFrame,那么

h_train = h_train[H]

相关内容

  • 没有找到相关文章

最新更新