Python 输入验证

Question

我有一个菜单，除非输入是 d 或 D，否则它将返回“e”。我想在不创建另一个变量并在一行上执行它的情况下执行此操作

encrypt = 'd' if (raw_input("Encrypt or Decrypt a file(E/d):") == ('d' or 'D')) else 'e'

[编辑] 好的，这是一个更难的

我该怎么做

file_text = 'a.txt' if (raw_input("File name(a.txt):")=='a.txt' else [What I typed in]

Answer 1

使用in运算符：

encrypt = 'd' if raw_input("Encrypt or decrypt a file (E/d):") in ('d', 'D') else 'e'

或者，您可以将输入转换为小写并将其与“d”进行比较：

encrypt = 'd' if raw_input("Encrypt or decrypt a file (E/d):").lower() == 'd' else 'e'

最后，如果你想确保他们输入的是 e 或 d，你可以将其包装在一个 while 循环中：

while True:
    encrypt = raw_input("Encrypt or decrypt a file (E/d):")

    # Convert to lowercase
    encrypt = encrypt.lower()

    # If it's e or d then break out of the loop
    if encrypt in ('e', 'd'):
        break

    # Otherwise, it'll loop back and ask them to input again

编辑：要回答你的第二个问题，我猜你可以使用 lambda 吗？

file_text = (lambda default, inp: default if inp.lower() == default else inp)("a.txt", raw_input("File name(a.txt):"))

虽然，这显然有些迟钝，也太“聪明”了一半。

Answer 2

不是真的认真，而是另一个单行解决方案（我认为它不可读）：

encrypt = {'d':'d','D':'d'}.get(raw_input("Encrypt or decrypt a file (E/d):"), 'e')

至少它很短。 有时字典实际上对类似的情况很有用（如果有更多选择的话）。