假设矩阵
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]？
只需将它们分开，然后让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可以正确广播它。
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句