How can I make my dataframe into a numpy array by column rather than by row?

Question

Consider that I have this example dataframe:

d = {'Gender': [1,1,0,1,0], 'Employed': [1,0,0,1,1]}

I would like this to be an array of this form:

[[1 1 0 1 0][1 0 0 1 1]]

When I run

d[['Gender', 'Employed']].to_numpy()

I get an array of the form [[1 1][1 0][0 0][1 1][0 1]].

user7864386 · Accepted Answer · 2021-12-13 11:49:25Z

1

You can transpose the DataFrame, get the underlying numpy array with values method and convert to a list with tolist method:

out = pd.DataFrame(d).T.values.tolist()

Output:

[[1, 1, 0, 1, 0],
 [1, 0, 0, 1, 1]]

answered Dec 13, 2021 at 11:49

user7864386

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

Comments

LukasNeugebauer · Accepted Answer · 2021-12-13 11:48:37Z

1

Just transpose it.

d[['Gender', 'Employed']].values.T

answered Dec 13, 2021 at 11:48

LukasNeugebauer

1,3478 silver badges10 bronze badges

Tamil Selvan · Accepted Answer · 2021-12-13 11:48:07Z

0

This can get your expected output

import pandas as pd
df = pd.DataFrame({'Gender': [1,1,0,1,0], 'Employed': [1,0,0,1,1]})
print([df[column].to_list() for column in df])

Output:

[[1, 1, 0, 1, 0], [1, 0, 0, 1, 1]]

Let me know if you expect more from here.

answered Dec 13, 2021 at 11:48

Tamil Selvan

1,7391 gold badge15 silver badges28 bronze badges