Passing an unspecified numpy array as an argument to a function

Question

I want to be able to define a function as having an argument of an unspecified array. For example,

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

Say that:

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

Then I want cols(x) to give output 3.

Note, the input must be a np.array, so please no workaround to that!

def cols(arr)? and why would you think this array gives you length of 3? — Guy
– Guy, Commented May 12, 2022 at 12:28
In you example, len(x) is equal to 2. If you want to get the size of the nested arrays (in other words the number of columns of x), you can use x.shape[1]. — leleogere
– leleogere, Commented May 12, 2022 at 12:35

Icenore · Accepted Answer · 2022-05-12 12:33:10Z

1

You don't have to specify what is your argument in 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

answered May 12, 2022 at 12:33

Icenore

966 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

William Over a year ago

Thank you, I'm quite new and this fixed it

Collectives™ on Stack Overflow

Passing an unspecified numpy array as an argument to a function

1 Answer 1

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related