歌詞歌詞詞頻統計

main.py

lyrics = []
while True:
    i = input()
    if i == '===':
        break
    i = i.split()
    for j in i:
        j = j.lower()
        if j == "we're":
            lyrics.append('we')
            lyrics.append('are')
            continue
        if j == "i'd":
            lyrics.append('i')
            lyrics.append('would')
            continue
        j = j.replace('.', '').replace(':', '').replace('?', '').replace(',', '').replace('\n', '')
        lyrics.append(j)
d = {}
for i in sorted(lyrics):
    if i in d:
        d[i] += 1
        continue
    d[i] = 1
for key, value in d.items():
    print('{}:{}'.format(key,value))