numpy不同维度的数组进行运算

    技术2024-01-28  94

    numpy中不同维度的数组进行相加的时候,只要其中一个数组的维度是另一个数组的子维度,那么他们就可以进行运算,相当于把低维数组通过复制,升级到高维,然后再进行运算 比如:

    a = np.arange(24).reshape(4,6) a array([[ 0, 1, 2, 3, 4, 5], [ 6, 7, 8, 9, 10, 11], [12, 13, 14, 15, 16, 17], [18, 19, 20, 21, 22, 23]]) b = np.arange(6) b array([0, 1, 2, 3, 4, 5]) a+b array([[ 0, 2, 4, 6, 8, 10], [ 6, 8, 10, 12, 14, 16], [12, 14, 16, 18, 20, 22], [18, 20, 22, 24, 26, 28]])

    相当于a的每一行都加上了b

    Processed: 0.010, SQL: 9