如何找到元組的維數？

Question

我有一個元組，看起來像這樣

array=(1,2,3,4) 
    lenM = numpy.shape(array) 
    print lenM 
    (4,) 

    if not lenM[1]: 
     "Code" 

現在我該如何自動化我的代碼，發現如果元組是一個維或二維？

Answer 1

您可以使用numpy.ndim此：

In [4]: np.ndim((1,2,3,4)) 
Out[4]: 1 
In [5]: np.ndim(((1,2),(3,4))) 
Out[5]: 2 

Answer 2

array=(1,2,3,4) 
lenM = numpy.shape(array) 
print lenM 
(4,) 

if len(lenM) == 1: 
    "1 dimensional code" 
elif len(lenM) == 2: 
    "2 dimensional code" 

len（lenM）會告訴你數組中是否有多個維度。如果len（lenM）是1，那麼只有一個維度。如果數組具有多個維度，則lenM將具有多個元素。