python - 在 Python 中将可变长度字符串拆分为变量的最佳方法是什么？

Question

假设我有一串由可变长度的逗号分隔的整数。拆分字符串并将整数存储到变量中的最佳方法是什么？

目前，我有以下内容。

input = sys.argv[1]
mylist = [int(x) for x in input.split(',')]
if len(mylist) == 2: a, b = mylist
else: a, b, c = mylist

有没有更有效的方法来做到这一点？

Answer 1

添加哨兵，然后将列表限制为 3 个元素：

a, b, c = (mylist + [None] * 3)[:3]

现在 a、b 和 c 至少设置为 None，如果项目数超过三个，则仅使用前三个值。

演示：

>>> mylist = [1, 2]
>>> a, b, c = (mylist + [None] * 3)[:3]
>>> print a, b, c
1 2 None
>>> mylist = [1, 2, 3, 4]
>>> a, b, c = (mylist + [None] * 3)[:3]
>>> print a, b, c
1 2 3

如果您需要至少 2 个元素，请使用更少的 None 值并捕获 ValueError：

try:
    a, b, c = (mylist + [None])[:3]
except ValueError:
    print "You mast specify at least 2 values"
    sys.exit(1)

Answer 2

只是Martjin的一个附录。把它变成一个函数来展示你为什么会使用它。您可以使用动态哨兵

def store(mylist,expsiz = 10, dflt = None):
    return mylist + [dflt]*(expsiz-len(mylist))

>>> mylist = [1,2,5]
>>> fixedlen = store(mylist)
>>> print fixedlen
[1,2,5,None,None,None,None,None,None,None]