python - 将重复的行标签转换为 Pandas 中的列标题

Question

我有一份这种格式的问卷

import pandas as pd
df = pd.DataFrame({'Question': ['Name', 'Age', 'Income','Name', 'Age', 'Income'], 
                   'Answer': ['Bob', 50, 42000, 'Michelle', 42, 62000]})

如您所见，重复出现相同的“问题”，我需要重新格式化，以便结果如下

df2 = pd.DataFrame({'Name': ['Bob', 'Michelle'], 
                   'Age': [ 50, 42],
                   'Income': [42000,62000]})

Answer 1

使用numpy.reshape:

print (pd.DataFrame(df["Answer"].to_numpy().reshape((2,-1)), columns=df["Question"][:3]))

或者转置和pd.concat:

s = df.set_index("Question").T

print (pd.concat([s.iloc[:, n:n+3] for n in range(0, len(s.columns), 3)]).reset_index(drop=True))

两者产生相同的结果:

Question      Name Age Income
0              Bob  50  42000
1         Michelle  42  62000