如何对数组中每个值的数字求和以创建新数组？

Question

• 起始数组： [1555, 1116, 221, 997]
• 结束数组： [16, 9, 5, 25]
• 其中结束数组中的每个数字是起始数组中相应数字的数字之和。
• 如何从起始数组到结束数组？

Answer 1

您可以使用列表理解，其中每个数字都转换为字符串->拆分为字符列表->转换为数字（数字）返回->求和：

start = [1555, 1116, 221, 997]
end = [sum(map(int, list(str(x)))) for x in start]
print(end) # output: [16, 9, 5, 25]

Answer 2

我相信您要问的是创建一个数组，其元素是原始数组中每个数字的数字之和。 数字总和被无耻地从Sum the numbers of a number - python 中窃取，并将它们放在一起你得到：

def sum_digits(n):
   r = 0
   while n:
       r, n = r + n % 10, n // 10
   return r

in_array = [1555, 1116, 221, 997]
out_array = [sum_digits(n) for n in in_array]
print(out_array)

希望这可以帮助！

Answer 3

• 使用sum和list-comprehension
• 不map & list

t = [1555, 1116, 221, 997]

result = [sum(int(d) for d in str(v)) for v in t]

print(result)
[16, 9, 5, 25]

• 对应for-loop

result = list()
for v in t:
    r = list()
    for d in str(v):
        r.append(int(d))
    result.append(sum(r))

%%timeit比较

• 给定以下测试列表

import numpy as np

np.random.seed(123)
t = [np.random.randint(100, 4000) for _ in range(4000000)]

• 这个答案

%%timeit
[sum(int(d) for d in str(v)) for v in t]

5.27 s ± 60.9 ms per loop (mean ± std. dev. of 7 runs, 1 loop each)

• 加迪普的回答

%%timeit
[sum(map(int, list(str(x)))) for x in t]

4.63 s ± 37 ms per loop (mean ± std. dev. of 7 runs, 1 loop each)

• 矿工哈里的回答

def sum_digits(n):
    r = 0
    while n:
        r, n = r + n % 10, n // 10
    return r

%%timeit
[sum_digits(n) for n in t]

1.72 s ± 10.7 ms per loop (mean ± std. dev. of 7 runs, 1 loop each)