NumPy数组与权重求和

发布于 2021-01-29 16:53:41

我有一个二维的numpy数组。

每行的长度为三个元素,并且是整数0-3。这代表一个6位整数,每个单元依次代表两位。

我正在尝试将它们转换为完整的整数。

例如

for i in range(len(myarray)):
  myarray[i] = myarray[i][0] * 16 + myarray[i][1] * 4 + myarray[i][2]

例如,我试图对每行求和,但要根据[16,4,1]的某个权重向量。

最优雅的方法是什么?我想我必须先进行某种点乘积运算,然后求和,但我不确定100%决定在何处进行点运算。

关注者
0
被浏览
140
1 个回答
  • 面试哥
    面试哥 2021-01-29
    为面试而生,有面试问题,就找面试哥。

    点积的倾斜度是正确的,并且包括所需的总和。因此,要获得目标数组元素和一组权重的乘积之和:

    >>> a = np.array([[0,1,2],[2,2,3]])
    >>> a
    array([[0, 1, 2],
           [2, 2, 3]])
    >>> weights = np.array([16,4,2])
    >>> np.dot(a,weights)
    array([ 8, 46])
    


知识点
面圈网VIP题库

面圈网VIP题库全新上线,海量真题题库资源。 90大类考试,超10万份考试真题开放下载啦

去下载看看