如何获得所选特征及其分数？

Question

参考Map object is not subscriptable error 我使用 Moses Xu 的答案来获取特征名称。它会产生错误“地图对象不可下标”。代码如下。我正在使用 python 3.x

top_ranked_features = sorted(enumerate(ch2.scores_),key=lambda x:x[1], reverse=True)[:1000]
top_ranked_features_indices = map(list,zip(*top_ranked_features))[0]
for feature_pvalue in zip(np.asarray(train_vectorizer.get_feature_names())[top_ranked_features_indices],ch2.pvalues_[top_ranked_features_indices]):
    print( feature_pvalue).

错误在第二行代码。

输出

('00 8b 4d fc', 3.4028916591534005e-61)
('51 00 22 05', 3.4028916591534005e-61)
('00 74 00 61', 8.3973527363656966e-61)

Answer 1

在 python 3.x 中，map 返回一个迭代器，而不是一个列表。因此，返回的对象是不可下标的。 您可以将其更改为list(map(list,zip(*a)))[0] 以解决问题。

已编辑：

您也可以使用next(map(list,zip(*a))) 不将分数加载到内存中，并让代码更加pythonic。