получить новый список из pivot_table? - PullRequest
Новая
       13

получить новый список из pivot_table?

1 голос
/ 18 марта 2019

Я использую Python 3 и панд.У меня есть эта сводная таблица ниже. 

    >>> print(p.head())
    question_id               1       2                       3                          4        ...                       26              27              28              29
    assessment_attempt_id
                                                                         ...                                                                         
243908                 21-24  Female  4th year undergraduate                      White       ...                 Disagree        Disagree           Agree           Agree
290934                 25-29    Male    Prefer Not to Answer  Black or African American       ...                  Neutral         Neutral         Neutral         Neutral
312457                 18-20  Female  1st year undergraduate                      White       ...           Strongly Agree  Strongly Agree  Strongly Agree  Strongly Agree
312766                 18-20  Female  2nd year undergraduate       Hispanic or Latina/o       ...                    Agree           Agree           Agree           Agree
312786                 21-24  Female  4th year undergraduate  Black or African American       ...        Strongly Disagree           Agree           Agree           Agree

Она создается из этой команды: 

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

Таблица в основном именно то, что я хочу.Теперь мне нужны столбцы 1, 2, 3, 4 и столбецvaluation_attempt_id в Datatable, чтобы я мог объединить эти данные с помощью analysis_attempt_id с другой существующей таблицей данных.

Обычно я бы помещал данные в подмножество, делая что-то вроде этого: 

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

, но это выдает ошибку: KeyError: "['1' '2'] not in index" Кажется,Как это должно быть простой и решаемой проблемой, но я не могу найти ответ.Я также попробовал групповую вариацию, которая выдает тот же результат, а также я не смог извлечь нужные столбцы.Я полагаю, что не могу на мульти-индекс, но я не знаю как.Спасибо.

...