Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

How to turn a Pandas column into array and transpose it?

I have a Pandas dataframe called 'training_set' that resembles the screenshot below:

enter image description here

I try to turn the 'label' column into array and transpose it. I tried doing Y_train=np.asarray(training_set['label']) but what I got is a horizontal array that resembles the screenshot below, which is not what I want.

enter image description here

I want the array to display vertically just like the screenshot below (The screenshot has 2 variables per row. My desired output should only contain 1 variable, the 'label', per row.)

enter image description here

Any suggestion or help would be greatly appreciated!

like image 235
Stanleyrr Avatar asked Mar 28 '18 23:03

Stanleyrr


2 Answers

One way:

Y_train = training_set['label'].values[:, None]
like image 159
jpp Avatar answered Oct 30 '22 09:10

jpp


pandas >= 0.24

Use DataFrame.to_numpy(), the new Right Way to extract a numpy array:

training_set[['label']].to_numpy()

pandas < 0.24

Slice out your column as a single columned DataFrame (using [[...]]), not as a Series:

Y_train = np.asarray(training_set[['label']])

Or,

Y_train = training_set[['label']].values
like image 33
cs95 Avatar answered Oct 30 '22 09:10

cs95