关注
找到一份通过的代码,你们看看 #!/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
相关推荐
牛客94579620...:同组诶
查看5道真题和解析 点赞 评论 收藏
分享
10-14 21:06
西北农林科技大学 数据分析师 点赞 评论 收藏
分享
牛客热帖
更多
正在热议
更多
# 你听到的“最没用”的秋招建议 #
6392次浏览 75人参与
# 第一次找实习,我建议__ #
1800次浏览 26人参与
# 京东开奖 #
539589次浏览 2870人参与
# 2025秋招体验点评 #
24293次浏览 254人参与
# 从mentor身上学到了__ #
3552次浏览 47人参与
# 秋招暂停,我将对以下公司做出处罚__ #
5457次浏览 19人参与
# 联影求职进展汇总 #
92128次浏览 428人参与
# 工作以后,你父母对你啥态度 #
3393次浏览 48人参与
# 薪资要看总包还是月薪? #
7805次浏览 88人参与
# 小红书取消大小周 #
76336次浏览 175人参与
# 什么样的公司千万别去 #
5870次浏览 48人参与
# 实习期间如何提升留用概率? #
192610次浏览 1612人参与
# 你在职场上见过哪些“水货”同事 #
9153次浏览 77人参与
# 军工所铁饭碗 vs 互联网高薪资,你会选谁 #
1133次浏览 16人参与
# 第一份工作应该选择高薪还是大平台 #
167812次浏览 927人参与
# 联想求职进展汇总 #
311301次浏览 2200人参与
# 双11,我的offer购物车开箱啦 #
4221次浏览 35人参与
# 实习心态崩了 #
89192次浏览 475人参与
# 你见过最离谱的招聘要求是什么? #
241186次浏览 1677人参与
# 快手工作体验 #
250062次浏览 2632人参与
# 实习生的蛐蛐区 #
830634次浏览 4002人参与

