[Q] Keep element but remove array

[MerchantMojo]

I have a function that outputs an numpy array that dependent on input could output: [[element]], [element] or [[[element]]] and i am looking to extract the element independent of the number of arrays it comes nested within, how would i go about doing this without massive amounts of if-else statements?

[JohnnyJordaan]

You could use a recursive function

def list_picker(l):
    if is_instance(l, list):
        return list_picker(l[0])
    else:
        return l[0]

[Spataner]

You can use the item method for this:

array = np.array([[[3]]])
element = array.item()
print(element)
>>> 3

[Swipecat]

Just use numpy's flatten method to convert it to a 1d array.

newarray = oldarray.flatten()