假设矩阵
X = np.array([[ 1, 2, 3],
[ 4, 5, 6],
[ 7, 8, 9],
[10, 11, 12]])和向量divisor = [5, 10, 15]。有没有一种使用numpy将X的第一列除以divisor[0]、将X的第二列除以divisor[1]、将X的第二列除以divisor[3]的方法
发布于 2020-09-29 09:03:44
只需将它们分开,让numpy负责广播:
X/divisor输出:
[[0.2 0.2 0.2]
[0.8 0.5 0.4]
[1.4 0.8 0.6]
[2. 1.1 0.8]]如果您想划分行(而不是列),请使用:
X/divisor[:,None]这给除数增加了一个维度,这样numpy就可以正确地传播它。
https://stackoverflow.com/questions/64111004
复制相似问题