数字各位求和
如果我想要计算一个数字的各位数字之和,比如:
- 输入:
932
- 输出:
14
,也就是(9 + 3 + 2)
那么,有什么最快的方法来做到这一点呢?
我本能地做了:
sum(int(digit) for digit in str(number))
然后我在网上找到了这个:
sum(map(int, str(number)))
哪种方法在速度上最好?还有没有其他更快的方法呢?
12 个回答
9
在一个解决问题的挑战网站上发现了这个。虽然不是我写的,但它确实有效。
num = 0 # replace 0 with whatever number you want to sum up
print(sum([int(k) for k in str(num)]))
17
如果你想一直把数字的各个数字相加,直到得到一个一位数的数字(这是我最喜欢的一个特点,特别是对于能被9整除的数字),你可以这样做:
def digital_root(n):
x = sum(int(digit) for digit in str(n))
if x < 10:
return x
else:
return digital_root(x)
其实这个方法本身也挺快的...
%timeit digital_root(12312658419614961365)
10000 loops, best of 3: 22.6 µs per loop
136
你发的这两行代码都没问题,但你可以只用整数来做,这样会更高效:
def sum_digits(n):
s = 0
while n:
s += n % 10
n //= 10
return s
或者可以用 divmod
函数:
def sum_digits2(n):
s = 0
while n:
n, remainder = divmod(n, 10)
s += remainder
return s
用一个赋值语句会稍微快一点:
def sum_digits3(n):
r = 0
while n:
r, n = r + n % 10, n // 10
return r
> %timeit sum_digits(n)
1000000 loops, best of 3: 574 ns per loop
> %timeit sum_digits2(n)
1000000 loops, best of 3: 716 ns per loop
> %timeit sum_digits3(n)
1000000 loops, best of 3: 479 ns per loop
> %timeit sum(map(int, str(n)))
1000000 loops, best of 3: 1.42 us per loop
> %timeit sum([int(digit) for digit in str(n)])
100000 loops, best of 3: 1.52 us per loop
> %timeit sum(int(digit) for digit in str(n))
100000 loops, best of 3: 2.04 us per loop