將未指定的 numpy 數組作為參數傳遞給函數

Question

我希望能夠將函數定義為具有未指定數組的參數。 例如，

import numpy as np
def cols(np.array([]): 
    return len(np.array([])

比如說：

x=np.array([[1,2,3],[4,5,6]])

然后我希望cols(x)給出輸出 3。

請注意，輸入必須是 np.array，因此請不要解決此問題！

Answer 1

您不必在 python 中指定您的參數是什么

import numpy as np

def cols(arr):
    return arr.shape[1]

number_of_cols = cols(np.array([[1, 2, 3], [4, 5, 6]]))
print(number_of_cols) # will print 3