How to rearrange a CSV?
I’m trying to rearrange this data set into years so that:
JavaScript
x
5
1
country1 | price | year1
2
country2 | price | year1
3
country1 | price | year2
4
country2 | price | year2
5
becomes:
JavaScript
1
4
1
| year1 | year2
2
country1 | price | price
3
country2 | price | price
4
How can I do that?
Advertisement
Answer
Try (I used the data from your link and assumed which columns you need):
JavaScript
1
9
1
import pandas as pd
2
3
df = pd.read_csv('data.csv',
4
usecols=['name', 'local_price', 'date'],
5
index_col='name')
6
df.sort_values(by=['name', 'date'], inplace=True)
7
df = df.pivot(columns='date', values='local_price')
8
df.to_csv('data_out.csv')
9