python 实现单词本的任务

import sys
data_set = set()
print('=' * 20)
print('欢迎使用单词记录本')
print('1.查看单词本')
print('2.背单词')
print('3.添加新单词')
print('4.删除单词')
print('5.清空单词')
print('6.退出')
print('=' * 20)
while True:
    fun_num = input('请输入功能编号：')
    if fun_num == '1':
        if len(data_set) == 0:
             print('单词本内容为空')
        else:
            print(data_set)
    elif fun_num == '2':
        if len(data_set) == 0:
            print('单词本内容为空')
        else:
            for random_words in data_set:
                word = random_words.split(':')
                in_words = input("请输入"+word[0]+'翻译'+':\n')
                if in_words == word[1].split():
                    print('太棒了')
                else:
                    print('再想想')
    elif fun_num == '3':
        new_words =input('请输入新单词：')
        for i in data_set:
            if new_words in i.split(':')[0]:
                print('添加单词重复')
                break
            else:
                new_chinese = input('请输入单词翻译：')
                words = new_words+':'+new_chinese
                data_set.add(words)
                print('单词添加成功')
    elif fun_num == '4':
        if len(data_set) == 0:
            print('单词本为空')
        else:
            print(data_set)
            del_wd = input("请输入要删除的单词：")
            word_li =[]
            for i in data_set:
                word_li.append(i.split(':')[0])
            if del_wd not in word_li:
                print('删除的单词不存在')
            else:
                for i in list(data_set):
                    if del_wd in i:
                        print('删除成功!')
    elif fun_num == '5':
        if len(data_set) == 0:
            print('单词本为空')
        else:
            data_set.clear()
            print('单词本清空成功')
    elif fun_num == '6':
        print('退出成功')
        break

 

运行结果：