Pandas Pivot_table KeyError

Question

Мои данные выглядят так, как показано ниже;

  'userID'  'songID'  'rating'
0         0      7171         5
1         0      8637         4
2         0     21966         4
3         0     35821         5
4         0     82446         5

Мой код приведен ниже для создания сводной таблицы;

ratings = pd.pivot_table(data, index = "userID", columns = "songID", values = 'rating')

Я получаю KeyError:'rating'

Я проверил другие ответы, большинство из них предлагают .reset_index(),, но это не сработало.Я продолжаю получать ту же ошибку.

Любое предложение, чтобы решить эту проблему?

Заранее спасибо.

Answer 1

Ваши столбцы имеют дополнительные quotation mark 'xx' вокруг.так что если вы хотите сохранить его, используйте следующую команду:

ratings = pd.pivot_table(df, index = "'userID'", columns = "'songID'", values = "'rating'")

Answer 2

Существует проблема в именах столбцов:

print (data.columns.tolist())
["'userID'", "'songID'", "'rating'"]

Вы можете strip набирать ' по:

ratings.columns = ratings.columns.str.strip("'")
ratings = pd.pivot_table(data, index = "userID", columns = "songID", values = 'rating')