python的进栈出栈遍历

小程序：扫一扫查出行
【扫一扫了解最新限行尾号】
复制小程序

python实现出栈进栈

要求：

进栈
出栈
遍历所有
退出

stack = [] #创建列表

#进栈

def pushstack():

    stack.append(input('Enter a nubmber: '))

#出栈

def popstack():

    if len(stack) == 0:

        print('Connnot pop none stack')

    else:

        stack.pop()

#遍历/查询

def viewstack():

    print(stack)

#字典对应功能

CMD = {'v':viewstack, 'u':pushstack, 'o':popstack}

#主函数

def main():

    lable = '''

    'q':quit, 'v':view, 'u':push, 'o'pop

    Enter a CMD --> '''

    while True:

        try:

            a = input(lable)

            if a not in 'qvuo':

                print('Wrong cmd :', a)

            elif a == 'q':#退出程序

                break

            CMD[a]()

        except Exception as e:

            print('try it again!')

if __name__ == '__main__':

    main()