在没有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是一个c实现的版本,elementwiseoperation具有标准名称的优势,使用任何可迭代类型并且速度更快(在某些版本上;请参阅@ 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)]