我想在不使用 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,但没有发现任何直接相关的内容。哪种方式更好?
最佳答案
通常的做法是使用 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)]
关于python - 在没有 numpy (python) 的情况下对列表进行元素操作的正确样式,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/17595590/