Python获取用户输入并将输入与字符串进行比较

Question

我正在编写一个程序，用户可以选择一种颜色，并使用树莓派点亮所述颜色的LED。在接受用户输入并尝试将其与颜色进行比较后，我得到"NameError:name‘red’未定义"。我该怎么解决这个问题？

这是我的代码：

import RPi.GPIO as GPIO
import time
GPIO.setwarnings(False)
#use raspberry pi board numbers
GPIO.setmode(GPIO.BOARD)  
#GPIO output channel
GPIO.setup(7, GPIO.OUT)
GPIO.setup(16, GPIO.OUT)
GPIO.setup(32, GPIO.OUT)
GPIO.setup(37, GPIO.OUT)
GPIO.setup(40, GPIO.OUT)
GPIO.setup(33, GPIO.OUT)
GPIO.setup(13, GPIO.OUT)
GPIO.setup(22, GPIO.OUT)
#get led color
ledColorList = input("What color of light do you want to turn on? Red, green, blue, yellow, or all").split(' ')
#blink function
def blink(pin):
    GPIO.output(pin,1)
    time.sleep(.25)
    GPIO.output(pin,0)
    time.sleep(.25)
    return
if ledColorList[0] == RED:
    blink(7)  
#turn off all pins
GPIO.cleanup()

Answer 1

我认为您最好将输入强制为小写，并通过raw_input:进行输入

ledColorList = raw_input("What color of light do you want to turn on? Red, green, blue, yellow, or all").lower().split(' ')

然后对照进行检查

if ledColorList[0] == "red":
    blink(7)  

Answer 2

RED是一个字符串litral：它应该放在引号之间。

Answer 3

对于python 2.x，您可以使用raw_input()而不是input()，因为

在Python2.x中，input（）需要一个Python表达式，这意味着如果你键入red，它会将其解释为一个名为red的变量。如果你键入"red"，那么它就可以了。