python - 在没有 numpy (python) 的情况下对列表进行元素操作的正确样式

Question

我想在不使用 numpy 的情况下逐个元素地对列表进行操作，例如，我想要 add([1,2,3], [2,3,4]) = [3,5,7 ] 和 mult([1,1,1],[9,9,9]) = [9,9,9]，但我不确定该怎么做它是否被认为是“正确”的风格。

我想到的两个解决方案是

def add(list1,list2):
    list3 = []
    for x in xrange(0,len(list1)):
        list3.append(list1[x]+list2[x])
    return list3

def mult(list1, list2):
    list3 = []
    for x in xrange(0,len(list1)):
        list3.append(list1[x]*list2[x])
    return list3

def div(list1, list2):
    list3 = []
    for x in xrange(0,len(list1)):
        list3.append(list1[x]/list2[x])
    return list3

def sub(list1, list2):
    list3 = []
    for x in xrange(0,len(list1)):
        list3.append(list1[x]-list2[x])
    return list3

每个运算符都有一个单独的函数

和

def add(a,b)
    return a+b
def mult(a,b)
    return a*b
def div(a,b)
    return a/b
def sub(a,b)
    return a-b
def elementwiseoperation(list1, list2, function):
    list3 = []
    for x in xrange(0,len(list1)):
        list3.append(function(list1[x],list2[x]))
    return list3

其中定义了所有基本函数，我有一个单独的函数可以在每个元素上使用它们。我浏览了 PEP8，但没有发现任何直接相关的内容。哪种方式更好？

Answer 1

通常的做法是使用 map 或 itertools.imap:

import operator
multiadd = lambda a,b: map(operator.add, a,b)
print multiadd([1,2,3], [2,3,4]) #=> [3, 5, 7]

Ideone: http://ideone.com/yRLHxW

map 是您的 elementwiseoperation 的 c 实现版本，具有标准名称的优点，可与任何可迭代类型一起使用并且速度更快(在某些版本上；有关一些分析，请参阅@nathan 的回答)。

或者，您可以使用 partial 和 map 来获得令人愉悦的无点样式:

import operator
import functools

multiadd = functools.partial(map, operator.add)
print multiadd([1,2,3], [2,3,4]) #=> [3, 5, 7]

Ideone: http://ideone.com/BUhRCW

无论如何，您已经自己迈出了函数式编程的第一步。我建议您围绕该主题进行阅读。

作为一般风格问题，如果您想访问每个项目，使用 range 按索引迭代通常被认为是错误的事情。通常的做法是直接迭代结构。使用 zip 或 itertools.izip 并行迭代:

for x in l:
    print l

for a,b in zip(l,k):
    print a+b

通常迭代创建列表的方法不是使用append，而是列表理解:

[a+b for a,b in itertools.izip(l,k)]