python实现增删查问题如何优化?

用python实现用户增删查问题:
背景:用户列表和密码列表如下所示
补充程序,实现对用户和密码列表进行增加、删除、查询。
程序显示菜单如下
请选择
1 用户增加
2 用户删除
3 用户查询
4 退出系统

当用户输入1
输入用户姓名和密码后,若用户在列表中,显示用户已存在 无法增加。
若不存在,则将用户名和密码追加到相应的列表中。并显示用户已增加。

当用户输入2
输入用户姓名后,
若用户不在列表中,显示用户不存在 不能删除。若在列表中,则将用户名和密码删除,并显示用户已删除。

当用户输入3
输入用户姓名后,
若用户不在列表中,显示用户不存在 不能查询。若在列表中,则显示密码。

当用户输入4,则退出循环

yhlb=["mary","harry","mike","tom","rose"]
mmlb=["123456","888888","999999","765432","678901"]
print("请选择
1 用户增加
2 用户删除
3 用户查询
4 查看数据,退出系统")
#代码开始
while True:
    cmd=input("请输入")
    if cmd=="4":
        break
    elif cmd=="1":
        name=input("请输入用户名")
        passwd=input("请输入密码")
        if name in yhlb:
            print("用户已经存在 无法增加")
        else:
            yhlb.append(name)
            mmlb.append(passwd)
            print("用户已增加")
    elif cmd=="2":
        name=input("请输入用户名")
        if name in yhlb:
            index=-1
            for i in range(len(yhlb)):
                if yhlb[i]==name:
                    index=i
                    break
            yhlb.pop(index)
            mmlb.pop(index)
            print("用户已删除")
        else:
            print("用户不存在 无法删除")
    elif cmd=="3":
        name=input("请输入用户名")
        if name in yhlb:
            index=-1
            for i in range(len(yhlb)):
                if yhlb[i]==name:
                    index=i
                    break
            print(mmlb[index])
        else:
            print("用户不存在")
        
    else:
        
        print("输入有误,请输入1~4")
#代码结束
for i in range(0,len(yhlb)):
    print("用户名{:6}密码{:6}".format(yhlb[i],mmlb[i]))

如何改进才能实现以下功能:

请选择
1 用户增加
2 用户删除
3 用户查询
4 查看数据,退出系统
请输入请输入用户名用户不存在
请输入请输入用户名密码123456
请输入用户名mary  密码123456
用户名harry 密码888888
用户名mike  密码999999
用户名tom   密码765432
用户名rose  密码678901

原来的代码不行吗?看要求没什么特殊的,除了菜单第四项少了查看数据。不存在时是要加上输入的用户名?

yhlb=["mary","harry","mike","tom","rose"]
mmlb=["123456","888888","999999","765432","678901"]
print("""请选择
1 用户增加
2 用户删除
3 用户查询
4 查看数据,退出系统""")
#代码开始
while True:
    cmd=input()
    if cmd=="4":
        for i in range(0,len(yhlb)):
            print("用户名{:6}密码{:6}".format(yhlb[i],mmlb[i]))
        break
    elif cmd=="1":
        name=input('请输入用户名密码:')
        passwd=input()
        if name  in yhlb:
            print('%s用户已存在 无法增加'%name)
        else:
            yhlb.append(name)
            mmlb.append(passwd)
            print('用户已增加')
    elif cmd=="2":
        name=input('请输入用户名:')
        if name in yhlb:
            index=-1
            for i in range(len(yhlb)):
                if yhlb[i]==name:
                    index=i
                    break
            yhlb.pop(index)
            mmlb.pop(index)
            print('用户已删除')
        else:
            print('%s用户不存在'%name)
    elif cmd=="3":
        name=input('请输入用户名:')
        if name in yhlb:
            index=-1
            for i in range(len(yhlb)):
                if yhlb[i]==name:
                    index=i
                    break
            print('密码%s'%mmlb[index])
        else:
            print('%s用户不存在'%name)
    else:
        print('输入值有误,请输入1~4')
#代码结束
您好,我是有问必答小助手,您的问题已经有小伙伴帮您解答,感谢您对有问必答的支持与关注!
PS:问答VIP年卡 【限时加赠:IT技术图书免费领】,了解详情>>> https://vip.csdn.net/askvip?utm_source=1146287632