2

假设我有一串由可变长度的逗号分隔的整数。拆分字符串并使用值更新变量(如果存在)的最佳方法是什么?

目前,我有以下内容。

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

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

4

6 回答 6

5
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 的。实例成员会更合适。

于 2012-07-03T15:49:10.933 回答
3

您可以使用辅助函数:

def f(a=10, b=10, c=1):
    return a, b, c

a, b, c = f(*map(int, input.split()))

这不会更快——这只是我想到的另一种方法。

于 2012-07-03T15:54:39.493 回答
1

使用切片将用户输入与默认参数列表结合起来:

>>> 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
于 2012-07-03T15:47:55.257 回答
1
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 内置的名称,因此您正在删除对该函数的轻松访问。

于 2012-07-03T15:40:03.463 回答
0

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)
于 2012-07-03T15:58:47.720 回答
0
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)
于 2012-07-03T16:21:35.140 回答