Question

我有以下列表清单：

list_of_lists=[[1,2,3], [4,5,2], [3,2,4]]

我想创建一个函数，为每个内部列表应用不同的权重。

所以当我写weighted_lists(list_of_lists,10,2,5.5)时：
- 第一个内部列表应乘以10
- 第二个内部列表应乘以2
- 第三个内部列表应乘以5.5

因此，结果我应该有以下内容：

weighted_lists=[[10,20,30], [8,10,4], [16.5,11,22]]

请注意，此函数应支持不同数量的内部列表（在某些情况下，我们可能有3个，在其他情况下，我们可能有400个）。

Answer 1

在这里，lol是列表清单。

def weighted_lists(lol, *weights):
    if len(lol) != len(weights):
        raise IndexError

    return [[weight*x for x in inner] for inner, weight in zip(lol, weights)]

演示：

list_of_lists=[[1,2,3], [4,5,2], [3,2,4]]
print(weighted_lists(list_of_lists, 10, 2, 5.5)) # [[10, 20, 30], [8, 10, 4], [16.5, 11.0, 22.0]]

Answer 2

您可能希望查看numpy这样的内容：

In [14]: import numpy as np

In [15]: list_of_lists=[[1,2,3],[4,5,2],[3,2,4]]

In [16]: weights = [10, 2, 5.5]

In [17]: (np.array(list_of_lists) * np.array(weights)[:, None]).tolist()
Out[17]: [[10.0, 20.0, 30.0], [8.0, 10.0, 4.0], [16.5, 11.0, 22.0]]

Answer 3

如果您更喜欢功能，可以使用itertools和operator.mul：

来完成

list_of_lists = [[1, 2, 3], [4, 5, 2], [3, 2, 4]]
from itertools import izip, starmap, repeat
from operator import mul

def weighted_lists(l, *args):
    return (list(starmap(mul, izip(*(s, repeat(i))))) for s, i in izip(l, args))


print(list(weighted_lists(list_of_lists, 10, 2, 5.5)))
[[10, 20, 30], [8, 10, 4], [16.5, 11.0, 22.0]]

将不同的权重应用于列表列表中的多个列表 - Python

3 个答案: