Python核心编程：用列表模拟堆栈

最新推荐文章于 2024-06-10 22:12:15 发布

原创最新推荐文章于 2024-06-10 22:12:15 发布 · 558 阅读

1 ·

CC 4.0 BY-SA版权

本文介绍了一个使用Python实现的简易命令行堆栈应用。该应用支持三种基本操作：压栈（输入新字符串）、弹栈（移除顶部元素）及查看当前堆栈状态。通过简单的用户界面，用户可以选择进行不同的堆栈操作。

#!/usr/bin/env python

stack = []

def pushit():
stack.append(raw_input('Enter New string: ').strip())

def popit():
if len(stack)==0:
print 'cannot pop from an empty stack!'
else:
print 'remove[','stack.pop',']'

def viewstack():
print stack #calls str() internally

CMDs = {'u': pushit,'o': popit,'v': viewstack}

def showmenu():
pr="""
p(U)sh
p(O)p
(V)iew
(Q)uit

ENter choice:"""

while True:
   while True:
   try:
   choice = raw_input(pr).strip()[0].lower()
   except (EOFError,KeyboardInterrupt,IndexError):
   choice = 'q'

   print'\nYou picked: [%s]' % choice
   if choice not in 'uovq':
   print 'Invalid option, try again'
   else:
      break

   if choice == 'q':
   break
   CMDs[choice]()

if __name__ == '__main__':
showmenu()