是否可以使用列表推导式模拟类似于sum()的函数?
例如 - 我需要计算列表中所有元素的乘积:
list = [1, 2, 3]
product = [magic_here for i in list]
#product is expected to be 6
做同样事情的代码:
def product_of(input):
result = 1
for i in input:
result *= i
return result
是否可以使用列表推导式模拟类似于sum()的函数?
例如 - 我需要计算列表中所有元素的乘积:
list = [1, 2, 3]
product = [magic_here for i in list]
#product is expected to be 6
做同样事情的代码:
def product_of(input):
result = 1
for i in input:
result *= i
return result
>>> from functools import reduce
>>> from operator import mul
>>> nums = [1, 2, 3]
>>> reduce(mul, nums)
6
Python 3 Hack
对于这种做法,例如[total := total + x for x in [1, 2, 3, 4, 5]]
,这是一个可怕的想法。使用列表推导式模拟 sum()
的一般思路违背了列表推导式的整个目的。在这种情况下不应该使用列表推导式。
在 Python 2.5
/ 2.6
中,您可以使用vars()['_[1]']
来引用当前正在构建的列表推导式。这是可怕的,绝不能使用,但它是最接近您在问题中提到的(使用列表推导式模拟乘积)。
>>> nums = [1, 2, 3]
>>> [n * (vars()['_[1]'] or [1])[-1] for n in nums][-1]
6
从 Python 3.8
开始,引入了 赋值表达式(PEP 572) (:=
运算符),我们可以在列表推导式中使用和递增变量,从而将列表减少为其元素的总和:
total = 0
[total := total + x for x in [1, 2, 3, 4, 5]]
# total = 15
这段代码:
total
为0
total
中(total := total + x
)list = [1,2,3,4,5]
product = [i for i in list if i % 2 ==0] # only sum even numbers in the list
print sum(product)
另一个例子:
# list of the cost of fruits in pence
list = [("apple", 55), ("orange", 60), ("pineapple", 140), ("lemon", 80)]
product = [price for fruit, price in list]
print sum(product)
dict = {"val":0}
list = [1, 2, 3]
product = [dict.update({"val" : dict["val"]*i}) for i in list]
print dict["val"] # it'll give you 6!
...但那太糟糕了 :)
类似这样的:
>>> a = [1,2,3]
>>> reduce(lambda x, y: x*y, a)
6
reduce
运算符的代码。list_of_numbers = [1, 5, 10, 100]
reduce(lambda x, y: x + y, list_of_numbers)
这也可以写成
list_of_numbers = [1, 5, 10, 100]
def sum(x, y):
return x + y
reduce(sum, list_of_numbers)
奖励:Python在内置的sum
函数中提供了此功能。在我看来,这是最易读的表达式。
list_of_numbers = [1, 5, 10, 100]
sum(list_of_numbers)
>>> reduce(int.__mul__,[1,2,3])
6
C:\Users\Henry>python -m timeit -s "" "reduce(int.__mul__,range(10000))"
1000 loops, best of 3: 910 usec per loop
C:\Users\Henry>python -m timeit -s "from operator import mul" "reduce(mul,range(10000))"
1000 loops, best of 3: 399 usec per loop
C:\Users\Henry>
可以使用带有列表推导式的lambda来实现。 由于我们无法在列表推导式中分配值,因此我们使用lambda。
解决方案:
>>> (lambda number_list, sum=0:[sum for number in number_list for sum in [sum + number]][-1])([1, 2, 3, 4, 5])
>>> 15
我可能有点晚参与这个讨论,但我想提一下列表推导式是图灵完备的,因此可以使用列表推导式来完成这个任务!
然而,这种方法比较混乱,所以我使用了以下技巧,它可以创建一个累积数组,并返回最后一个元素。
def sum(l):
return [c[-1] for c in [[0]] for e in l if c.append(c[-1] + e) is None][-1]
在http://code.activestate.com/recipes/436482/上找到了魔法。
>>> L=[2, 3, 4]
>>> [j for j in [1] for i in L for j in [j*i]][-1]
24
代码应该像以下的逻辑。
L=[2, 3, 4]
P=[]
for j in [1]:
for i in L:
for j in [j*i]:
P.append(j)
print(P[-1])
[j for j in [0] for i in L for j in [j+i]][-1]
但我绝不会推荐这种方法。要进行求和,您需要有一个起始列表,其中包含一个标识为[0]的元素。对于乘积,从[1]开始可以工作,因为您将第一个数字乘以1,因此需要更改为[0]。正如@ivan_pozdeev所指出的那样,这是使用操作应用于列表的每个元素来构建新列表,然后仅获取最终值。这与问题不同。 - Ryan Beesleyeval('*'.join(map(str,list)))
是我通常使用的方法。 - graviton