添加可变长度的数组

Question

我的一个函数生成一个可变长度列表的结果，例如 [1, 2, 3, ...] 等。随着这些列表的出现，我想将它们全部添加到此列表中的某个主列表中时尚：

out: [1, 2]
out: [3] 
  sum:[4,2]
out: [0, 0, 0, 2] 
  sum: [4, 2, 0, 2]
out: [3, 1] 
  sum: [7, 3, 0, 2] 

等等。

我还需要能够对这些进行操作 - 例如：

[1, 2, 3, 4]/N = [1/N, 2/N, ...] 

numpy 数组是我想要的格式吗？我似乎无法让它工作。

Answer 1

您可以使用 python 2.x 中的 map/itertools.izip_longest 或 python 3.x 中的 itertools.zip_longest 来完成功能

话虽如此，

• 遍历map函数结果中的所有元素
• 对于each子列表，可以使用python的sum()内置函数。
• 要处理无值，请检查None 是否存在于each 变量中。如果存在，则遍历该子列表，检查每个元素 i 是否为 None，如果是，则将其替换为 0。

也就是说，

>>> addFn = lambda a,b: [sum(each) if None not in each else sum(i if i else 0 for i in each )for each in map(None,a,b)]
>>> addFn([1,2,3],[4])
[5, 2, 3]
>>> addFn(addFn([1,2,3],[4]),[7,3,0,2])
[12, 5, 3, 2]

对于第二个功能，如果l 如果您的列表，那么

对于l中的每个元素i，转换分子或分母float并执行操作！

>>> divByN = lambda l,N:[i/float(N) for i in l]
>>> divByN([5,6,7,8],2)
[2.5, 3.0, 3.5, 4.0]

正如 Baldrickk 所说，这里有一个使用 itertools 的示例，

from itertools import izip_longest as izip_l 
sum = [a+b for a,b in izip_l(arr1,arr2, fillvalue=0)]

Answer 2

import numpy as np

def Add(a, b):
    a, b = map(np.array, (a, b))
    if a.size > b.size: 
        a[:b.size] += b
        return a
    else:
        b[:a.size] += a
        return b

示例用法：

>>> Add([1,2], [3])
array([ 4.,  2.])
>>> Add([0,0,0,2], _)
array([ 4.,  2.,  0.,  2.])

该函数的第一行用于将这些数组的副本转换为浮点数，因此您可以稍后进行除法而无需四舍五入。

Answer 3

另一种解决方案：

def add_shortest(a, b):
    a = numpy.array(a)
    b = numpy.array(b)

    common_length = min(a.size, b.size)
    a[:common_length] += b[:common_length]
    return a

add_shortest(numpy.arange(2), numpy.arange(3))  # array([0, 2])
add_shortest(numpy.arange(3), numpy.arange(3))  # array([0, 2, 4])
add_shortest(numpy.arange(3), numpy.arange(2))  # array([0, 2, 2])
add_shortest(add_shortest([0, 0, 0, 2], add_shortest([1, 2], [3])), [3, 1])  # array([7, 3, 0, 2])