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

为了创建一个pivot_table，我的代码在下面；

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

我收到一个KeyError:'rating'

我检查了其他答案，其中大多数建议使用.reset_index(),但没有用。 我不断收到同样的错误。

有什么建议可以解决这个问题吗？

提前致谢。

Answer 1

列名有问题：

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

您可以通过以下方式strip traling ' ：

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

Answer 2

您的列周围有额外的quotation mark 'xx' 。 所以如果你想保留它然后使用以下

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