代码之家  ›  专栏  ›  技术社区  ›  minovsky

美国运输部两个二维数组

  •  1
  • minovsky  · 技术社区  · 4 年前

    a = np.array([[1, 2], [3, 4], [5, 6]])
    b = np.array([[2, 2], [3, 3], [4, 4]])
    

    我想得到一个1d数组,在每个对应的索引处都有列表的点积,即。

    [(1*2 + 2*2), (3*3 + 4*3), (5*4 + 6*4)]
    [6, 21, 44]
    

    我该怎么做呢?提前谢谢!

    1 回复  |  直到 4 年前
        1
  •  3
  •   meTchaikovsky    4 年前

    你能做到的

    np.sum(a*b,axis=1)