Use flatten() on a list of 2d arrays to get a list of 1d arrays

Question

I have a list e of 2d arrays, which I want to flatten to get a list of 1d arrays

When I use on one element:

e[0].flatten()

it works. But when I want to transform every element with:

enew = [e[i].flatten() for i in e]

the failure gets raised:

IndexError: arrays used as indices must be of integer (or boolean) type

You are indexing the array e using elements in e as indices. So when you call e[i] you are using an array (i) as an index to a list (e). As the error message suggests, only integer and boolean arrays are acceptable as indices for lists. What you want is to call flatten() on each item individually (see the answer by @YTTY). — corvus
– corvus, Commented Aug 18, 2020 at 14:47

maggle · Accepted Answer · 2020-08-18 14:42:21Z

3

It should be i.flatten() not e[i].flattent() I guess

answered Aug 18, 2020 at 14:42

5274 silver badges12 bronze badges

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

Comments

Hadrian · Accepted Answer · 2020-08-18 15:02:31Z

0

i is an index so it should be an int not a list, so:

enew = [e[i].flatten() for i in len(e)]

answered Aug 18, 2020 at 14:56

9477 silver badges11 bronze badges