巧妙使用`*`解包傳遞參數

有時候需要傳遞一個參數作爲shape,格式是para1,para2,注意 不需要括號。

比如這個函數,np.random.randn()
我們希望生成一個和一直數組arr形狀相同的隨機數組。可以這樣寫:np.random.randn(arr.shape[0],arr.shape[1])

但是有點太low了,你還需要去看一下arr到底是幾維的。這是用*來解包就非常方便:

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