如何用python制作term文档

2024-03-29 11:17:57 发布

您现在位置:Python中文网/ 问答频道 /正文

我有16000条来自imdb数据集的记录

Movie_Name         Synops 
Alien Predator     ['great','17th', 'abigail', 'by', 'century', 'is']
Shark Exorcist     ['demonic', 'devil', 'great', 'hell', 'holy', 'nun']
Jurassic Shark     ['abandoned', 'an', 'and', 'beautiful', 'abigail',]

我不知道如何像这样为Synops列中的每个单词制作术语文档

"great": Alien Predator,Shark Exorcist
"17th"  :Alien Predator
"abigail":Alien Predator,Jurassic Shark
.....

Tags: 数据nameby记录movieimdbgreatshark
2条回答
data = {
    "Alien Predator": ['great','17th', 'abigail', 'by', 'century', 'is'],
    "Shark Exorcist": ['demonic', 'devil', 'great', 'hell', 'holy', 'nun'],
    "Jurassic Shark": ['abandoned', 'an', 'and', 'beautiful', 'abigail',]
}

result = {}
for movie_name, keywords in data.items():
    for keyword in keywords:
        result.setdefault(keyword, []).append(movie_name)
print(result)

结果(为清晰起见添加了新行):

{
'great': ['Alien Predator', 'Shark Exorcist'], 
'17th': ['Alien Predator'], 
'abigail': ['Alien Predator', 'Jurassic Shark'], 
'by': ['Alien Predator'], 
'century': ['Alien Predator'], 
'is': ['Alien Predator'], 
'demonic': ['Shark Exorcist'], 
'devil': ['Shark Exorcist'], 
'hell': ['Shark Exorcist'], 
'holy': ['Shark Exorcist'], 
'nun': ['Shark Exorcist'], 
'abandoned': ['Jurassic Shark'], 
'an': ['Jurassic Shark'],
'and': ['Jurassic Shark'], 
'beautiful': ['Jurassic Shark']
}

先把它们放到字典或JSON中。一旦你有了它。你知道吗

dataset = {
"Alien Predator":['great','17th', 'abigail', 'by', 'century', 'is'],
"Shark Exorcist":['demonic', 'devil', 'great', 'hell', 'holy', 'nun'],
"Jurassic Shark":['abandoned', 'an', 'and', 'beautiful', 'abigail',],
}

您可以从此处轻松查询值。你知道吗

search_word = "great"
d = [movie for movie, synops in dataset.items() if search_word in synops]

回馈['Alien Predator', 'Shark Exorcist']

您可以将它们添加到字典中以生成完整的结果。你知道吗

final_dict = {}
final_dict[search] = d

那应该给你。你知道吗

>>> final_dict
{'great': ['Alien Predator', 'Shark Exorcist']}

现在,您可以使用一些for循环和所需关键字的列表来实现相同的功能,并自己完成任务。你知道吗

相关问题 更多 >