How to flatten an object array in numpy?

Question

I have an array

np.array([[[ 1,  2], [ 3, -4]],[[-1,  0]]], dtype=object)

and I want to flatten it to get something like:

array([1,2,3,-4,-1,0], dtype=int32)

I tried Flatten numpy array but it raises a Value error

To be clear, my array is always an object array consists of multiple 2D and 1D arrays

@hpaulj It is an object array consisting of 1d and 2d arrays — Shree Singhi
– Shree Singhi, Commented Jul 5, 2019 at 11:35
In newer versions, the print shows which of the elements are arrays, eg. np.array([np.array([1,2,3]), ...]) Your print is ambiguous, but I'm guessing (2,) containing a (2,2) and (1,2). If so then vstack should make a (3,2). — hpaulj
– hpaulj, Commented Jul 5, 2019 at 11:45

hpaulj · Accepted Answer · 2019-07-05 16:08:10Z

2

In [333]: arr = np.array([[[ 1,  2], [ 3, -4]],[[-1,  0]]], dtype=object)                                       
In [334]: arr                                                                                                   
Out[334]: array([list([[1, 2], [3, -4]]), list([[-1, 0]])], dtype=object)
In [335]: arr.shape                                                                                             
Out[335]: (2,)
In [336]: np.vstack(arr)                                                                                        
Out[336]: 
array([[ 1,  2],
       [ 3, -4],
       [-1,  0]])
In [337]: np.vstack(arr).ravel()                                                                                
Out[337]: array([ 1,  2,  3, -4, -1,  0])

answered Jul 5, 2019 at 16:08

hpaulj

233k14 gold badges260 silver badges392 bronze badges

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

1 Comment

j314erre Apr 4 at 4:08

brilliant, works

Florian H · Accepted Answer · 2019-07-05 11:14:23Z

0

If you dont know the depth of your nestet array you can do it like this:

l = np.array([[[ 1,  2],[ 3, -4]], [-1,  0]])

from collections.abc import Iterable

def flatten(l):
    returnlist=[]
    for elem in l:
        if isinstance(elem, Iterable):
            returnlist.extend(flatten(elem))
        else:
            returnlist.append(elem)

    return returnlist

np.array(flatten(l))

If its 2 dimensional you can go like that post suggest How to make a flat list out of list of lists:

flat_list = [item for sublist in l for item in sublist]

or just use numpys flatten.

By the way, your example is not two dimensional because of those double brackets, thats also why flatten() does not work:

np.array([[[ 1, 2], [ 3, -4 ]],[[-1, 0]]], dtype=object)

answered Jul 5, 2019 at 11:14

Florian H

3,0722 gold badges16 silver badges27 bronze badges

2 Comments

Shree Singhi Over a year ago

Is there any simpler/faster method ?

Florian H Over a year ago

Better data to start with would make it easier. You have a mix of a numpy array and lists here. But to answer the question: There might be, but at the moment i know for no faster method even though it probably exists.

monkut · Accepted Answer · 2019-07-06 13:01:04Z

0

It's nested, but you can use sum() to do this:

nested_list_values = [[[ 1,  2], [ 3, -4]],[[-1,  0]]]
sum(sum(, []), [])

[1, 2, 3, -4, -1, 0]

Or, with itertools.chain() if that feels more natural:

from itertools import chain

nested_list_values = [[[ 1,  2], [ 3, -4]],[[-1,  0]]]
list(chain(*chain(*nested_list_values)))

[1, 2, 3, -4, -1, 0]

answered Jul 6, 2019 at 13:01

monkut

44.1k26 gold badges133 silver badges161 bronze badges

Collectives™ on Stack Overflow

How to flatten an object array in numpy?

3 Answers 3

1 Comment

2 Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

1 Comment

2 Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related