ChatGPT - Reshape dataframe in Python