python - 带有是/否输入的 While 循环 (Python)

Question

我正在构建一个脚本来绘制平滑值和绘图值。但是，我无法获得控制 while 循环的是/否函数。

我将 while 条件设置为“N”并等待用户在退出前说他们喜欢绘图(输入 Y)。

我收到“NameError: name 'reply' is not defined.”

import matplotlib.pyplot as plt
import numpy

while True:
    reply[0] = 'n'
    def yes_or_no(question):
        reply = str(input(question+' (y/n): ')).lower().strip()
        if reply[0] == 'y':
            return 1
        if reply[0] == 'n':
            return 0
        else:
            return yes_or_no("Please Enter (y/n) ")

# a bunch of math
# plots for the user

yes_or_no('Do you like the plot')
break
print("done")

当我调整 reply[0] 回复程序挂起(见下文)

print("started")
while True:
    reply = 'n'
    def yes_or_no(question):
        reply = str(input(question+' (y/n): ')).lower().strip()
        if reply[0] == 'y':
            return 1
        if reply[0] == 'n':
            return 0
        else:
            return yes_or_no("Please Enter (y/n) ")

yes_or_no('Do you like the plot')
print("done")

Answer 1

尝试:

def yes_or_no(question):
    reply = str(input(question+' (y/n): ')).lower().strip()
    if reply[0] == 'y':
        return 1
    elif reply[0] == 'n':
        return 0
    else:
        return yes_or_no("Please Enter (y/n) ")

print("started")
while True:
    # DRAW PLOT HERE;
    print("See plot....")
    if(yes_or_no('Do you like the plot')):
        break
print("done")

为了清楚起见，最好将函数定义与循环分开。另外，否则会在每个循环中读取，浪费资源。

输出:

$ python ynquestion.py 
started
See plot....
Do you like the plot (y/n): n
See plot....
Do you like the plot (y/n): N 
See plot....
Do you like the plot (y/n): NO
See plot....
Do you like the plot (y/n): No
See plot....
Do you like the plot (y/n): no
See plot....
Do you like the plot (y/n): yes
done
$