Python 询问使用者 yes / no

Sam Tang

8 years ago

编写在 CLI 执行的程式, 很多时都会用到询问使用者 yes / no 是否执行, 以下记录一下在 Python 的写法:

例如:

#!/usr/bin/python

action = raw_input('continue?')

if action == 'yes':
    print 'Hello World!'

#!/usr/bin/python

action = raw_input('continue?')

if action == 'yes':

print 'Hello World!'

上面程式很简单询问是否继续, 如果输入 “yes” 则会输出 “Hello World!”.

上面程式必须输入 “yes” 才会输出 “Hello World!”, 如果输入 “Yes” 则不会继续, 其中一个方法是将 if 判断式加入多一个选项, 例如改成这样:

#!/usr/bin/python

action = raw_input('continue?')

if action == 'yes' or action == 'Yes':
    print 'Hello World!'

#!/usr/bin/python

action = raw_input('continue?')

if action == 'yes' or action == 'Yes':

print 'Hello World!'

上面程式码对输入 “yes” 及 “Yes” 均有效, 但如果使用者输入 “YES”, “YEs”, “yEs”…… 则不会继续, 所以较好的写法是用 lower() 将输入转换为英文小写, 再进行判断, 例如:

#!/usr/bin/python

action = raw_input('continue?')

if action.lower() == 'yes':
    print 'Hello World!'

#!/usr/bin/python

action = raw_input('continue?')

if action.lower() == 'yes':

print 'Hello World!'

以上程式码只要是 “yes” 的输入均有效, 那一个是大小写英文字母也可以。