如何在python中制作术语文档

Question

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

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

.....

Answer 1

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']}

Answer 2

首先将它们放入字典或 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 循环和所需关键字列表来实现相同的功能，并自己完成任务。