Python - Conversion of list of arrays to 2D array
If I understood correctly what you're asking, you have a case where numpy did not convert array of arrays into 2d array. This can happen when your arrays are not of the same size. Example:
Automatic conversion to 2d array:
import numpy as npa = np.array([np.array([1,2,3]),np.array([2,3,4]),np.array([6,7,8])])print a
Output:
>>>[[1 2 3] [2 3 4] [6 7 8]]
No automatic conversion (look for the change in the second subarray):
import numpy as npb = np.array([np.array([1,2,3]),np.array([2,3,4,5]),np.array([6,7,8])])print b
Output:
>>>[array([1, 2, 3]) array([2, 3, 4, 5]) array([6, 7, 8])]
I found a couple of ways of converting an array of arrays to 2d array. In any case you need to get rid of subarrays which have different size. So you will need a mask to select only "good" subarrays. Then you can use this mask with list comprehensions to recreate array, like this:
import numpy as npa = np.array([np.array([1,2,3]),np.array([2,3,4,5]),np.array([6,7,8])])mask = np.array([True, False, True])c = np.array([element for (i,element) in enumerate(a) if mask[i]])print aprint c
Output:
>>>>[array([1, 2, 3]) array([2, 3, 4, 5]) array([6, 7, 8])]>>>>[[1 2 3] [6 7 8]]
Or you can delete "bad" subarrays and use vstack(), like this:
import numpy as npa = np.array([np.array([1,2,3]),np.array([2,3,4,5]),np.array([6,7,8])])mask = np.array([True, False, True])d = np.delete(a,np.where(mask==False))e = np.vstack(d)print aprint e
Output:
>>>>[array([1, 2, 3]) array([2, 3, 4, 5]) array([6, 7, 8])]>>>>[[1 2 3] [6 7 8]]
I believe second method would be faster for large arrays, but I haven't tested the timing.