将未指定的 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

将未指定的 numpy 数组作为参数传递给函数

Passing an unspecified numpy array as an argument to a function

python

arrays