什么错这个python脚本

Question

我有一个简单的脚本返回不少误区：

import numpy as np 
def test(array): 
    ncol=np.shape(array)[1] 
    return ncol 

这应该返回数组的列数。它有什么问题？数组是numpy数组。下面是输出：

ncol=np.shape(array)[1] Display all 195 possibilities? (y or n) ArithmeticError( continue AssertionError( copyright( AttributeError( credits( BaseException( def BlockingIOError( del

Answer 1

您需要添加一个try.. catch围绕ncol=np.shape(array)[1]，因为它没有当数组是一维尺寸：

import numpy as np 

arr = np.random.normal(size=10) 
arr1 = np.random.normal(size=(10,5)) 

def test(array): 
    try: 
     ncol=np.shape(array)[1] 
     return ncol 
    except Exception as e: 
     print("no columns in array") 
     return None 

print(test(arr)) 
# output: 
# no columns in array 
# None 

print(test(arr1)) 
# output: 
# 5