Python數據結構之堆棧實例代碼

stack = [] # 初始化一個列表數據類型對象, 作為一個棧
 
def pushit(): # 定義一個入棧方法
 stack.append(raw_input('Enter New String: ').strip()) 
 # 提示輸入一個入棧的 String 對象, 調用 Str.strip() 保證輸入的 String 值不包含多余的空格
 
def popit(): # 定義一個出棧方法
 if len(stack) == 0:
 print "Cannot pop from an empty stack!"
 else:
 print 'Remove [', `stack.pop()`, ']'
 # 使用反單引號(` `)來代替 repr(), 把 String 的值用引號擴起來, 而不僅顯示 String 的值
 
def viewstack(): # 定義一個顯示堆棧中的內容的方法
 print stack
 
CMDs = {'u':pushit, 'o':popit, 'v':viewstack}
# 定義一個 Dict 類型對象, 將字符映射到相應的 function .可以通過輸入字符來執行相應的操作
 
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()
 # Str.strip() 去除 String 對象前后的多余空格
 # Str.lower() 將多有輸入轉化為小寫, 便于后期的統一判斷
 # 輸入 ^D(EOF, 產生一個 EOFError 異常)
 # 輸入 ^C(中斷退出, 產生一個 keyboardInterrupt 異常)
 
 except (EOFError, KeyboardInterrupt, IndexError):
 choice = 'q'
 
 print '
You picked: [%s]' % choice
 
 if choice not in 'uovq':
 print 'Invalid option, try again'
 else:
 break
 
 
 if choice == 'q':
 break
 CMDs[choice]()
 # 獲取 Dict 中字符對應的 functionName, 實現函數調用
 
if __name__ == '__main__':
 showmenu()