TheAlgorithms-Python

1
def get_word_pattern(word: str) -> str:
2
    """
3
    Returns numerical pattern of character appearances in given word
4
    >>> get_word_pattern("")
5
    ''
6
    >>> get_word_pattern(" ")
7
    '0'
8
    >>> get_word_pattern("pattern")
9
    '0.1.2.2.3.4.5'
10
    >>> get_word_pattern("word pattern")
11
    '0.1.2.3.4.5.6.7.7.8.2.9'
12
    >>> get_word_pattern("get word pattern")
13
    '0.1.2.3.4.5.6.7.3.8.9.2.2.1.6.10'
14
    >>> get_word_pattern()
15
    Traceback (most recent call last):
16
    ...
17
    TypeError: get_word_pattern() missing 1 required positional argument: 'word'
18
    >>> get_word_pattern(1)
19
    Traceback (most recent call last):
20
    ...
21
    AttributeError: 'int' object has no attribute 'upper'
22
    >>> get_word_pattern(1.1)
23
    Traceback (most recent call last):
24
    ...
25
    AttributeError: 'float' object has no attribute 'upper'
26
    >>> get_word_pattern([])
27
    Traceback (most recent call last):
28
    ...
29
    AttributeError: 'list' object has no attribute 'upper'
30
    """
31
    word = word.upper()
32
    next_num = 0
33
    letter_nums = {}
34
    word_pattern = []
35

36
    for letter in word:
37
        if letter not in letter_nums:
38
            letter_nums[letter] = str(next_num)
39
            next_num += 1
40
        word_pattern.append(letter_nums[letter])
41
    return ".".join(word_pattern)
42

43

44
if __name__ == "__main__":
45
    import pprint
46
    import time
47

48
    start_time = time.time()
49
    with open("dictionary.txt") as in_file:
50
        word_list = in_file.read().splitlines()
51

52
    all_patterns: dict = {}
53
    for word in word_list:
54
        pattern = get_word_pattern(word)
55
        if pattern in all_patterns:
56
            all_patterns[pattern].append(word)
57
        else:
58
            all_patterns[pattern] = [word]
59

60
    with open("word_patterns.txt", "w") as out_file:
61
        out_file.write(pprint.pformat(all_patterns))
62

63
    total_time = round(time.time() - start_time, 2)
64
    print(f"Done!  {len(all_patterns):,} word patterns found in {total_time} seconds.")
65
    # Done!  9,581 word patterns found in 0.58 seconds.
66