巧妙使用`*`解包传递参数

有时候需要传递一个参数作为shape,格式是para1,para2,注意 不需要括号。

比如这个函数,np.random.randn()
我们希望生成一个和一直数组arr形状相同的随机数组。可以这样写:np.random.randn(arr.shape[0],arr.shape[1])

但是有点太low了,你还需要去看一下arr到底是几维的。这是用*来解包就非常方便:

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