假设我有一串由可变长度的逗号分隔的整数。拆分字符串并使用值更新变量(如果存在)的最佳方法是什么?
目前,我有以下内容。
a, b, c = 10, 10, 1 #default values
mylist = [int(x) for x in input.split(',')]
if len(mylist) == 2: a, b = mylist
else: a, b, c = mylist
有没有更有效的方法来做到这一点?
假设我有一串由可变长度的逗号分隔的整数。拆分字符串并使用值更新变量(如果存在)的最佳方法是什么?
目前,我有以下内容。
a, b, c = 10, 10, 1 #default values
mylist = [int(x) for x in input.split(',')]
if len(mylist) == 2: a, b = mylist
else: a, b, c = mylist
有没有更有效的方法来做到这一点?
a, b, c = 10, 10, 1 #default values
mylist = [int(x) for x in input.split(',')]
a, b, c = mylist + [a, b, c][len(mylist):]
我认为这很难看的原因是聚合处理局部变量是非 Pythonic 的。实例成员会更合适。
您可以使用辅助函数:
def f(a=10, b=10, c=1):
return a, b, c
a, b, c = f(*map(int, input.split()))
这不会更快——这只是我想到的另一种方法。
使用切片将用户输入与默认参数列表结合起来:
>>> defaults = [10, 10, 1]
>>> user_input = '15 20'
>>> user_ints = map(int, user_input.split())
>>> combined = user_ints + defaults[len(user_ints):]
>>> a, b, c = combined
defaults=[10,10,1]
mylist=[int(x) for x in ipt.split(',')]
defaults[:len(mylist)]=mylist
a,b,c=defaults
不过,这会发生变化defaults
......为了避免这种情况,这样的事情会起作用:
defaults=[10,10,1]
mylist=[int(x) for x in ipt.split(',')]
temp_defaults=defaults[:]
temp_defaults[:len(mylist)]=mylist
a,b,c=temp_defaults
此外,请小心input
用作变量名。它是 python 内置的名称,因此您正在删除对该函数的轻松访问。
izip_longest
如果更长,则允许采用默认值的长度:
>>> from itertools import izip_longest
>>> inp = '3, 56'
>>> a, b, c = [i if i else j for i, j in izip_longest([int(x) for x in inp.split(',')], (10, 10, 1))]
>>> a, b, c
(3, 56, 1)
a, b, c = (map(int, user_input.split(',')) + [20,20,10])[:3]
或者
def parse(user_input, *args):
return (map(int, user_input.split(',')) + list(args))[:len(args)]
>>> a, b, c = parse('1,2', 20, 20, 10)
>>> a, b, c
(1, 2, 20)
>>> a, b, c, d = parse('1,2,3,4,5', 0, 0, 0, 0)
>>> a, b, c, d
(1, 2, 3, 4)