用Python實現棧

這是Python核心編程裡的一個例子:

#!/usr/bin/env python

#定義一個列表來模擬棧
stack = []

#進棧,調用列表的append()函數加到列表的末尾,strip()沒有參數是去掉首尾的空格
def pushit():
stack.append(raw_input('Enter new string: ').strip())

#出棧,用到了pop()函數
def popit():
if len(stack) == 0:
print 'Cannot pop from an empty stack!'
else:
print 'Removed [', stack.pop(), ']'

#編歷棧
def viewstack():
print stack

#CMDs是字典的使用
CMDs = {'u': pushit, 'o': popit, 'v': viewstack}

#pr為提示字符
def showmenu():
pr = """
p(U)sh
p(O)p
(V)iew
(Q)uit
Enter choice: """

while True:
while True:
try:
#先用strip()去掉空格,再把第一個字符轉換成小寫的
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

#CMDs[]根據輸入的choice從字典中對應相應的value,比如說輸入u,從字典中得到value為pushit,執行pushit()進棧操作
if choice == 'q':
break
CMDs[choice]()

#判斷是否是從本文件進入,而不是被調用
if __name__ == '__main__':
showmenu()

需要注意的就是對齊問題.