关注
找到一份通过的代码,你们看看 #!/usr/bin/env python
import math
import random
import sys
class Solution2(object):
def tfidf(self, line, idf):
tf = {}
tfidf = []
data = []
words = line.split(",")
length = len(words)
for word in words:
tf[word] = tf.get(word, 0) + 1 / length
for word in words:
tfidf.append(str(round(tf.get(word) * idf.get(word), 2)))
data.append("tf:%s,idf:%s" % (tf.get(word), idf.get(word)))
return (tfidf)
def getWordDocs(self, docs):
N = len(docs)
idf = {}
for doc in docs:
words = set(doc.split(","))
for word in words:
idf[word] = idf.get(word, 0) + 1
for (k, v) in idf.items():
idf[k] = round(math.log(N / v, 2), 2)
return idf
if __name__ == "__main__":
solution2 = Solution2()
n = int(sys.stdin.readline().strip())
dd = []
for i in range(0, n):
dd.append(sys.stdin.readline().strip())
idf = solution2.getWordDocs(dd)
for item in dd:
tfidf = solution2.tfidf(item, idf)
print(",".join(tfidf))
查看原帖
点赞 3
相关推荐
点赞 评论 收藏
分享
牛客热帖
更多
正在热议
更多
# 实习在多还是在精 #
33323次浏览 233人参与
# 我的求职进度条 #
80142次浏览 1093人参与
# 智慧芽求职进展汇总 #
1513次浏览 5人参与
# 秋招踩过的“雷”,希望你别再踩 #
79632次浏览 1049人参与
# 实习下班不想学习,正常吗? #
18896次浏览 170人参与
# 你现在会用到哪些AI技能? #
4447次浏览 72人参与
# 未岚大陆求职进展汇总 #
6423次浏览 82人参与
# 你的房租占工资的比例是多少? #
64311次浏览 797人参与
# 你见过哪些工贼行为 #
15812次浏览 87人参与
# 你还有多少年退休? #
26475次浏览 192人参与
# 校招谈薪一定要知道的事 #
12517次浏览 111人参与
# 反问环节如何提问 #
114927次浏览 2452人参与
# 找工作中的小确幸 #
25190次浏览 258人参与
# 小马智行求职进展汇总 #
13328次浏览 49人参与
# 顺丰求职进展汇总 #
63083次浏览 313人参与
# 你觉得什么岗位会被AI替代 #
15686次浏览 174人参与
# 大厂VS公务员你怎么选 #
23880次浏览 349人参与
# 如果不考虑收入,你最想做什么工作? #
32459次浏览 185人参与
# 非技术岗投递进展 #
157699次浏览 1314人参与
# 总结:哪家公司面试体验感最好 #
69886次浏览 416人参与
# 金蝶求职进展汇总 #
53619次浏览 262人参与
# 在国企工作的人,躺平了吗? #
367834次浏览 3915人参与