Pandas convert columns to rows

Question

I am a beginner in Data Science and I am trying to pivot this data frame using Pandas:

So it becomes something like this: (The labels should become the column and file paths the rows.)

I tried this code which gave me an error:

EDIT:

I have tried Marcel's suggestion, the output it gave is this:

The "label" column is a group or class of file paths. I want to convert it in such a way it fits this function: tf.Keras.preprocessing.image.flow_from_dataframe in categorical

Thanks in advance to all for helping me out.

Answer 1

I did not understand your question very well, but if you just want to convert columns to rows then you can do

train_df.T

wich means transpose

Answer 2

I think you are looking for something like this:

import pandas as pd

df = pd.DataFrame({
    'labels': ['a', 'a', 'a', 'b', 'b'],
    'pathes' : [1, 2, 3, 4, 5]
})

labels = df['labels'].unique()
new_cols = []
for label in labels:
    new_cols.append(df['pathes'].where(df['labels'] == label).dropna().reset_index(drop=True))
df_final = pd.concat(new_cols, axis=1)

print(df_final)

Answer 3

I've found what was wrong, I misunderstood y_col and x_col in tf.Keras.preprocessing.image.ImageDataGenerator.flow_from_dataframe . Thanks to all of you for your contributions. Your answers are all correct in different ways. Thanks again Marcel h and user16714199 !