python - 在数据框中组织收集的条目

Question

我有一个包含调查答案的数据框。每个人回答 3 个问题。不幸的是，数据框的每一行都对应一个问题的答案，而不是一个人的条目。

我怎样才能重新格式化它？

目前我有一个列设置为的数据框:

person_id、person_name、person_age、question、answer。

此数据框中的每个条目都是一个人对单个问题的回答。

我希望每个条目都是那个人的所有答案。所以列会更像

person_id，person_name，person_age，question1，question2，问题3

我没有太多代码或错误可以分享，因为我几乎不知道我应该如何解决这个问题，或者我应该如何使用该逻辑。

但不是像这样的数据框:

001___james___20___Question1____maybe
001___james___20___Question2____sure
001___james___20___Question3____no
003___anne___22___Question2____sure
003___anne___22___Question3____no

我希望它看起来像:

001___james___20___maybe___sure___no
003___anne___22___[BLANK]___sure___no

Answer 1

更简单的方法是使用 unstack 和 set index:

df = pd.DataFrame([
    [1,'James', 20, 'question 1', 'Yes'],
    [1,'James', 20, 'question 2', 'No'],
    [1,'James', 20, 'question 3', 'Maybe'],
    [2,'Elle', 20, 'question 1', 'No'],
    [2,'Elle', 20, 'question 2', 'Yes'],
    [2,'Elle', 20, 'question 3', 'Maybe']
], columns = ['person_id', 'person_name','person_age','question', 'answer'])

df = df.set_index(['person_id','person_name','person_age','question'])['answer'].unstack()

输出

question                         question 1 question 2 question 3
person_id person_name person_age                                 
1         James       20                Yes         No      Maybe
2         Elle        20                 No        Yes      Maybe