numpy 廣播 帶有冒號

# encoding: utf-8

import numpy as np

a=np.array([[1,2,3],
           [2,3,4],
           [4,5,6],
           [7,8,9]])

b=np.array([1,2,3])


for i in range (4):
    
    a[i,:]+=b
print(a)

 

if __name__ == "__main__":

    print("hello https://tool.lu/")

則a=a=np.array([[2,4,6],
           [2,5,7],
           [5,7,9],
           [8,10,12]])

但是這種for方式比較低效,可以採用tile函數進行重複,在行上重複一次,列上重複4ci

a+np.tile(b,(4,1))

brodcast(廣播i)可將上述操作變得更爲簡單

直接採用a+b,numpy會在數組缺失的情況下自動補全

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章