通过另一个数组的元素规范化矩阵

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.]])

Answer 2

您可以使用简单的嵌套列表理解：

>>> A = [[10, 20, 30], [15, 30, 25], [12, 30, 6]]
>>> B = [2, 5, 6]
>>> [[e / b for e in a] for a,b in zip(A,B)]
[[5.0, 10.0, 15.0], [3.0, 6.0, 5.0], [2.0, 5.0, 1.0]]
>>>