如何返回至少4D的數組：有效的方法來模擬numpy.atleast_4d

Question

numpy提供了三個方便的例程來將數組變成至少一維，二維或三維數組，例如，通過numpy.atleast_3d

我需要一個維度的等價物：atleast_4d。我可以想到使用嵌套if語句的各種方法，但我想知道是否有更有效和快速的方法返回有問題的數組。在你的回答中，如果可以的話，我會有興趣看到執行速度的估計值（O（n））。

Answer 1

的np.array方法有一個可選關鍵字ndmin論點，即：

指定結果數組 應有尺寸的最小數目。將根據需要預先設置形狀以符合 這一要求。

如果你也設置copy=False你應該接近你以後的東西。

---

這樣做它自己的選擇，如果你想額外維度尾隨而不是領導：

arr.shape += (1,) * (4 - arr.ndim) 

Answer 2

爲什麼不可以僅僅是這麼簡單的東西：

def atleast_4d(x): 
    if x.ndim < 4: 
     y = expand_dims(np.atleast_3d(x), axis=3) 
    else 
     y = x 

    return y 

即。如果維數小於4，則調用atleast_3d並在末尾附加一個額外的維度，否則返回數組不變。