convert a dictionary in a numpy array

Question

I have a dictionary where each item is 1D python array and I would like to convert it into a 2D numpy array.

For example, my dictionary is

a = {"tom": np.random.uniform(size = 100), "bill": np.random.uniform(size=100)}

and I would like to store in a 2D numpy array with shape (100,2). Have you any suggestions?

There are various ways of combining a list or tuple of arrays. — hpaulj
– hpaulj, Commented Sep 12, 2021 at 20:20

Mad Physicist · Accepted Answer · 2021-09-12 20:33:04Z

3

You can stack the values:

np.stack(a.values(), axis=-1)

answered Sep 12, 2021 at 19:43

Mad Physicist

116k29 gold badges202 silver badges292 bronze badges

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

Comments

akuiper · Accepted Answer · 2021-09-12 19:43:24Z

2

You can column_stack the values:

np.column_stack(a.values()).shape
# (100, 2)

answered Sep 12, 2021 at 19:43

akuiper

216k33 gold badges362 silver badges379 bronze badges