Python 3.x 从pivot_表获取新列表？

我正在使用python 3和pandas。我在下面有一张透视表

它由以下命令生成：

    p= pandas.pivot_table(df, index=["assessment_attempt_id"], columns=["question_id"], values="text", aggfunc='first')

这张桌子基本上正是我想要的。现在我只需要Datatable中的第1、2、3、4列和assessment_trunt_id列，这样我就可以通过assessment_trunt_id将该数据与另一个现有数据表连接起来

通常，我会通过如下操作对数据进行子集划分：

    df1 = df[['a','b']]

但这会产生错误： KeyError:“['1''2']不在索引中” 这似乎是一个简单且已解决的问题，但我找不到答案。我还尝试了一个groupby变体，它产生了相同的输出，而且我无法提取我想要的列。我想我不能在多重索引上，但我不知道怎么做。多谢各位

    df1 = df[['a','b']]