我该如何编写下面这个magic
函数?
>>> num = 123
>>> lst = magic(num)
>>>
>>> print lst, type(lst)
[1, 2, 3], <type 'list'>
num = 1234
lst = [int(i) for i in str(num)]
a = 123456
b = str(a)
c = []
for digit in b:
c.append (int(digit))
print c
你可以这样做:
>>> num = 123
>>> lst = map(int, str(num))
>>> lst, type(lst)
([1, 2, 3], <type 'list'>)
magic = lambda num: map(int, str(num))
那就这么做
magic(12345)
或者magic(someInt) #or whatever
>>> from collections import deque
>>> def magic(num):
digits = deque()
while True:
num,r = divmod(num,10)
digits.appendleft(r)
if num == 0:
break
return list(digits)
>>> magic(123)
[1, 2, 3]
根据我的时间记录,这个解决方案比字符串方法(magic2
)要快得多,即使是对于更小的示例也是如此。
>>> def magic2(num):
return [int(i) for i in str(num)]
时间:
魔术
>>> timeit.timeit(setup='from __main__ import magic', stmt='magic(123)')
1.3874572762508706
>>> timeit.timeit(setup='from __main__ import magic', stmt='magic(999999999)')
3.2624468999981673
magic2
>>> timeit.timeit(setup='from __main__ import magic2', stmt='magic2(123)')
3.693756106896217
>>> timeit.timeit(setup='from __main__ import magic2', stmt='magic2(999999999)')
10.485281719412114
map(int, str(n))
比两者都快(在Python3上测试过)。 - Felipe Buccionimap
只会创建一个生成器,而不是一个列表。你需要使用list(map(int, str(n))
。map(int, str(n))
本身几乎什么也不做。 - jamylak不要使用单词list作为变量名!它是Python内置数据类型的名称。
另外,请澄清您的问题。如果您正在寻找创建一个只有一个成员的列表的方法,请执行以下操作:
a = 123
my_list = [a]
并且“pythonizing” Cannonade的回答:
a = 123
my_list = [int(d) for d in str(a)]
num = map(int, list(str(num)))
def magic(x):
if x < 10:
return [x]
else:
return magic(x//10) + [x%10]
针对 Python 3.x:
num = 1234
lst = list(map(int, str(num)))
你可以尝试这个:
def convert_to_list(number):
return list(map(lambda x: int(x), str(number)))
convert_to_list(1245)