使用 Python 检查文本中是否存在不同的单词组合

Question

I want to write a function that finds certain word combinations in text and tells it belongs to which list. Example:

my_list1 = ["Peter Parker", "Eddie Brock"]
my_list2 = ["Harry Potter", "Severus Snape", "Dumbledore"]

Example input: "Harry Potter was very sad"
Example output: my_list1

Answer 1

You could iterate over the string and then append the occuring words into a list and then check for which words occur most to determine which list the whole string belongs to:

my_list1 = ["Peter Parker", "Eddie Brock"]
my_list2 = ["Harry Potter", "Severus Snape", "Dumbledore"]


to_check = "Harry Potter was very sad"

def which_list(to_check):
    belong_l1 = 0
    belong_l2 = 0    
    for i in to_check:
        if i in my_list1:
            belong_l1 += 1
        elif i in my_list2:
            belong_l2 += 1
    if belong_l1 > belong_l2:
        print("string belongs to list 1")
    elif belong_l1 < belong_l2:
        print("string belongs to list 2")
    else:
        print("belonging couldn't be determined")
        

Answer 2

First, I would include the name in the list.

lst = (
          ("list1", ("Peter Parker", "Eddie Brock")),
          ("list2", ("Harry Potter", "Severus Snape", "Dumbledore")),
          ("list3", ("Harry Potter",)),
        )
while True:
  txt = input( "Enter some text: ")
  if len(txt) == 0: break
  for names in lst:
    for name in names[1]:
      if name in txt:
        print( f"'{name}' found in {names[0]}.")

And the result:

Enter some text: Harry Potter was here
'Harry Potter' found in list2.
'Harry Potter' found in list3.
Enter some text: Harry Potter and Dumbledore were here
'Harry Potter' found in list2.
'Dumbledore' found in list2.
'Harry Potter' found in list3.

Answer 3

Assuming the following data structure:

lists = {
    'my_list1': ["Peter Parker", "Eddie Brock"],
    'my_list2': ["Harry Potter", "Severus Snape", "Dumbledore"]
}

and this argument:

arg = "Harry Potter was very sad"

this comprehension will return names of all lists that contain any keyword from the argument:

list_names = [
    list_name
    for list_name, keywords in lists.items()
    if any(kw in arg for kw in keywords)
]