What is the pythonic way to calculate dot product?

后端 未结 10 895
无人及你
无人及你 2020-12-08 00:39

I have two lists, one is named as A, another is named as B. Each element in A is a triple, and each element in B is just an number. I would like to calculate the result defi

10条回答
  •  温柔的废话
    2020-12-08 01:09

    import numpy
    result = numpy.dot( numpy.array(A)[:,0], B)
    

    http://docs.scipy.org/doc/numpy/reference/

    If you want to do it without numpy, try

    sum( [a[i][0]*b[i] for i in range(len(b))] )
    

提交回复
热议问题