将NumPy数组中的行乘以基于另一行的特定值

Question

我有以下清单：

ls = [[1,2,3], [3,4] , [5] , [7,8], [23], [90, 81]]

这是我的numpy数组：

array([[    1,     0,     4,     3],
       [   10,   100,  1000, 10000]])

我需要将数组第二行中的值乘以 ls中列表的长度，该长度在第一行中对应数字的索引处 ：

10 * len(ls[1])和100 * len(ls[0])等。

目标输出将是以下数组：

array([[    1,     0,     4,     3],
       [   20,   300,  1000, 20000]])

有任何有效的方法吗？

Answer 1

使用list comprehesion找到长度并将其乘以数组的第二行，如下所示：

ls = [[1,2,3], [3,4] , [5] , [7,8]]
arr = np.array([[    1,     0,     2,     3],
                [   10,   100,  1000, 10000]])

arr[1,:] = arr[1,:]*([len(l) for l in ls])

arr
array([[    1,     0,     2,     3],
       [   30,   200,  1000, 20000]])

编辑：

arr[1,:] = arr[1,:]*([len(ls[l]) for l in arr[0,:]])

arr
array([[    1,     0,     2,     3],
       [   20,   300,  1000, 20000]])