with open('tf_idf.txt', 'w', encoding='utf-8') as f: for word, tf in tf_values.items(): idf = idf_values.get(word, 0) tf_idf = tf * idf f.write('{}\t{}\n'.format(word, tf_idf))将上述代码的结果改为降序输出
时间: 2023-08-10 19:09:40 浏览: 84
TF-IDF.py.zip_TF-IDF WEIGHT_tf-idf_tf_idf_特征提取
可以将代码中的`tf_idf`值存入一个列表中,然后使用`sorted`函数对列表进行降序排序,最后再将排序后的结果写入文件中。修改后的代码如下:
```
with open('tf_idf.txt', 'w', encoding='utf-8') as f:
tf_idf_values = []
for word, tf in tf_values.items():
idf = idf_values.get(word, 0)
tf_idf = tf * idf
tf_idf_values.append((word, tf_idf))
tf_idf_values = sorted(tf_idf_values, key=lambda x: x[1], reverse=True)
for word, tf_idf in tf_idf_values:
f.write('{}\t{}\n'.format(word, tf_idf))
```
这样就可以将结果按照tf-idf值降序输出到文件中。
阅读全文