Блог

Как изменить порядок CSV?

Я пытаюсь преобразовать этот набор данных в годы, чтобы:

 country1 | price | year1
country2 | price | year1
country1 | price | year2
country2 | price | year2
  

становится:

          | year1 | year2 
country1 | price | price
country2 | price | price
  

https://raw.githubusercontent.com/TheEconomist/big-mac-data/master/source-data/big-mac-source-data.csv

Как я могу это сделать?

Попробуйте (я использовал данные из вашей ссылки и предположил, какие столбцы вам нужны):

 import pandas as pd

df = pd.read_csv('data.csv',
                 usecols=['name', 'local_price', 'date'],
                 index_col='name')
df.sort_values(by=['name', 'date'], inplace=True)
df = df.pivot(columns='date', values='local_price')
df.to_csv('data_out.csv')
  

Попробуйте df.pivot() :

 df.pivot(index='Country', columns='Year', values='Price')
  

Вы можете сделать это с помощью сводной таблицы pandas:

 import pandas as pd
pd.pivot_table(index='Country', columns='Year', values='Price')
  

Вывод:

   year    year1  year2
country
country1  price  price
country2  price  price