数字列表上的数学运算

Question

我想知道是否有可能在数值变量列表之间进行数学运算？例如,我有......

pointA = [ 22, 44, 83 ]
pointB = [ -17, 11, -25 ]

pointC = pointA - pointB
#result: [ 5, 55, 61 ]

或者我应该创建自己的功能？谢谢!

Answer 1

安装numpy.

>>> import numpy
>>> numpy.add([ 22, 44, 83 ], [ -17, 11, -25 ])
array([ 5, 55, 58])

数组对象大多与列表兼容,但功能更强大.

>>> pointA = numpy.array([ 22, 44, 83 ])
>>> pointB = numpy.array([ -17, 11, -25 ])
>>> pointA + pointB
array([ 5, 55, 58])
>>> pointA * pointB
array([ -374,   484, -2075])
>>> pointA.dot(pointB)
-1965

支持大量其他操作,矩阵和多维数组......