np.append當值中的一個可能是空的或無

Question

我有這樣的僞代碼：

a = np.zeros([3,3]) 
b = np.zeros([3,3]) 
c = np.append(a,b,axis=0) 

但問題是，a或b或者兩者都是無而非陣列，他們是。所以我這樣做：

def emptyappend(a,b): 
    if type(a)==np.ndarray and b is None: 
     return a 
    elif type(b)==np.ndarray and a is None: 
     return b 
    elif type(a)==np.ndarray and type(b)==np.ndarray: 
     return np.append(a,b,axis=0) 
    else: 
     return None 

這看起來真醜，不是很「Python化」，但我不能想出一個簡單的方法來做到這一點...任何想法？

Answer 1

可以追加數組只有在條件匹配，否則返回一個兩個短路與or：

def emptyappend(a, b): 
    if type(a) is np.ndarray and type(b) is np.ndarray: 
     return np.append(a, b, axis=0) 
    return a or b