python中的广播

如下图，为100g不同食物中卡路里来自碳水化合物、蛋白质和脂肪的情况。

不通过for循环，计算出不同食物中卡路里来自碳水化合物、蛋白质和脂肪的比例。如：Apples中的Carb = %（56.0）/ （56.0 + 1.2 + 1.8）= %94.9

代码：

import numpy as np

A = np.array([[56.0, 0.0, 4.4, 68.0], 
              [1.2, 104.0, 52.0, 8.0], 
              [1.8, 135.0, 99.0, 0.9]])
print(A)

cal = A.sum(axis=0)   # axis=0：让python在竖直方向求和，水平轴则为axis=1
print(cal)

percentage = 100 * A / cal.reshape(1, 4)    # 调用reshape指令确保它为正确的行向列或列向量
print(percentage)   # 百分比矩阵

 

执行效果：