如何有效地计算字符串别名？

Question

我正在从事一个个人项目，该项目计算文本中提到的名称实例。我知道我可以用collections.Counter() 做到这一点，但我不确定如何有效地考虑别名。

例如，假设我要计算的名字之一是"Tim"，但我也想计算他的任何昵称，例如"Timmy" 和"Timster"。

我有一些字符串说，"Oh Tim is going to the party?"、"Yeah, my boy Timmy, wouldn't miss it, he loves to party!"、"Whoa, the Timster himself is going? Count me in!"

我想把它们都算作"Tim" 这样的变量。我知道我可以简单地单独计算它们，然后将计数加在一起。但我觉得有更好的方法可以做到这一点。

I.E.我希望我的代码看起来更像。

names = {
    'Tim':{'Tim', 'Timmy', 'Timster'},
    ... other names here.}
# add any occurrence of Tim names to Tim and other occurrences of other names to their main name.

相对于类似的东西

total_tim = Counter(tim) + Counter(timmy) + Counter(timster), etc..

每个名字。有谁知道我会怎么做？

Answer 1

这是一个使用正则表达式的非常简单的解决方案。

此解决方案的优点是您不必明确命名变体。如果你知道那个人名字的开头变体，你应该没问题。

from collections import Counter
import re

TEXT = '''
    Blah Tim blah blah Timmy blah Timster blah Tim
    Blah Bill blah blah William blah Billy blah Bill Bill
'''

tim_search = '(Tim([a-z]*)?(?=\ ?))'
bill_search = '((B|W)ill([a-z]*)?(?=\ ?))'
def name_counter(regex_string): 
   return Counter([i for i, *j in re.findall(regex_string, TEXT)])

name_counter(tim_search)
Counter({'Tim': 2, 'Timmy': 1, 'Timster': 1})

name_counter(bill search)
Counter({'Bill': 3, 'Billy': 1, 'William': 1})

Answer 2

使用正则表达式将有助于解决这个问题。

import re
your_dict = {"Tim":["Tim","Timmy","Timster"]}
s = "Oh Tim is going to the party? Yeah, my boy Timmy, wouldn't miss it, he loves to party! Whoa, the Timster himself is going? Count me in!"
for each in your_dict:
    print(each,"count = ", len(re.findall("|".join(sorted(your_dict[each],reverse=True)),s)))

如果你想忽略大小写，那么只需在re.findall 中使用re.IGNORECASE 参数

Answer 3

from collections import Counter

TEXT = '''
    Blah Tim blah blah Timmy blah Timster blah Tim
    Blah Bill blah blah William blah Billy blah Bill Bill
'''
words = TEXT.split()

# Base names a their aliases.
ALIASES = dict(
    Tim = {'Tim', 'Timmy', 'Timster'},
    Bill = {'Bill', 'William', 'Billy'},
)

# Given any name, find its base name.
BASE_NAMES = {a : nm for nm, aliases in ALIASES.items() for a in aliases}

# All names.
ALL_NAMES = set(nm for aliases in ALIASES.values() for nm in aliases)

# Count up all names.
detailed_tallies = Counter(w for w in words if w in ALL_NAMES)

# Then build the summary counts from those details.
summary_tallies = Counter()
for nm, n in detailed_tallies.items():
    summary_tallies[BASE_NAMES[nm]] += n

print(detailed_tallies)
print(summary_tallies)

# Counter({'Bill': 3, 'Tim': 2, 'Timmy': 1, 'Timster': 1, 'William': 1, 'Billy': 1})
# Counter({'Bill': 5, 'Tim': 4})