python - 用另一个数组的元素标准化一个矩阵？

Question

我想编写Python代码来解决下面提到的问题。

我有一个列表列表:

A = [[10, 20, 30], [15, 30, 25], [12, 30, 6]]

我有一个数字列表:

B = [2, 5, 6] 

我想迭代列表 B 并将列表 A 的每一行除以 B 中的元素。例如:

[10, 20, 30] / B[1]
[15, 30, 25] / B[2]
[12, 30, 6] / B[3]

并获得如下输出:

output = [[5, 10, 15], [3, 6, 5], [2, 5, 1]]  

我不知道使用 for 循环是否是一个好主意？因为我无法得到正确的结果。

有什么想法可以找出解决方案吗？

Answer 1

在 Numpy 中，您可以像下面这样简单地完成:

A/B[:, None]

演示:

In [60]: A
Out[60]: 
array([[10, 20, 30],
       [15, 30, 25],
       [12, 30,  6]])

In [61]: B
Out[61]: array([2, 5, 6])

In [63]: A/B[:, None]
Out[63]: 
array([[  5.,  10.,  15.],
       [  3.,   6.,   5.],
       [  2.,   5.,   1.]])