獲取用戶輸入並保存到最多三個變量

Question

我有一個問題，用下面的代碼：

cmd, arg, arg1 = input("> ").split(" ") 

我想要得到的投入這三個瓦爾。 但是如果我離開arg和arg1空，Python的抱怨：

沒有足夠的數值解壓縮（預計3，拿到1）

我怎樣才能避免這種情況？

我想使arg和arg1可選。

Answer 1

如果大小不同，則無法解壓變量。好吧，不是這樣的。

您可以使用extended iterable unpacking（又稱星拆包）（Python的3只）：

cmd, *args = input("> ").split(" ") 

現在如果你只輸入一個命令

，args是空的，否則它解壓您所輸入的參數列入清單。

if not args: 
    # there are no arguments 
    pass 
elif len(args)>2: 
    print("too many args") 
else: 
    print("args",args) 

請注意，你會發現split(" ")有限。我會做split()（沒有參數：組空白，以避免空參數），或處理與shlex.split(input("> "))

注意：與python 2，你必須拆分和測試長度。較不優雅，但會奏效。

Answer 2

您可以獲取輸入字符串之後單獨設置的變量：

inp=input("> ").split(" ") 
l = len(inp) 
if l >= 1: 
    cmd = inp[0] 
if l >= 2: 
    arg1 = inp[1] 
if l >= 3: 
    arg2 = inp[2]