Question

我的DataFrame由2列组成。一个带有病人的身份证，另一个带有病人的问题。我需要创建一个DataFrame，其中患者的所有问题都与相应患者的ID排成一行。现在，如果患者有问题，则此数据框将排成唯一的行。

PAT_MRN_ID  Problem                      
9641956     Headache
9641956     Stomach_ache  
8227510     Headache 
8165474     Chicken_pox
7860000     Stomach_ache

上面的示例需要喜欢：

 PAT_MRN_ID  Headache         Stomach_ache      Chicken_pox
 9641956      1                1                   0
 8227510      1                0                   0
 8165474      0                0                   1
 7860000      0                1                   0

最终，我想将DataFrame归类为上述示例。我尝试使用循环和聚合，但是不幸的是我的基本编程技能还不够。

Answer 1

使用pd.get_dummies。

src/@fuse

import pandas as pd
df = pd.DataFrame({"PAT_MRN_ID": [9641956, 9641956, 8227510, 8165474, 7860000], "Problem": ["Head", "Stomach", "Head", "Pox", "Stomach"]})
pd.get_dummies(df,columns=["Problem"]).groupby(df.index).sum()

Answer 2

将get_dummies与DataFrame.set_index一起使用，每个索引的最大值和DataFrame.reset_index：

df1 = (pd.get_dummies(df.set_index('PAT_MRN_ID')['Problem'], 
                    prefix='', prefix_sep='')
         .max(axis=0, level=0)
         .reset_index())
print (df)

PAT_MRN_ID Chicken_pox  Headache  Stomach_ache                                  
9641956               0         1             1
8227510               0         1             0
8165474               1         0             0
7860000               0         0             1

Answer 3

首先获取“问题”的假人，然后分组

import pandas as pd
df = pd.DataFrame({ "PAT_MRN_ID" : [9641956,9641956,8227510,8165474,7860000],
                    "Problem" : ["Headache","Stomach-Ache","Headache","Chicken-Pox","Stomach-Ache"]
                 })

    PAT_MRN_ID  Problem
0   9641956     Headache
1   9641956     Stomach-Ache
2   8227510     Headache
3   8165474     Chicken-Pox
4   7860000     Stomach-Ache


df=pd.get_dummies(df, columns=['Problem'],prefix='',prefix_sep='')
     .groupby(['PAT_MRN_ID'], as_index=False)
     .max()


    PAT_MRN_ID  Chicken-Pox Headache    Stomach-Ache
0   7860000     0           0           1
1   8165474     1           0           0
2   8227510     0           1           0
3   9641956     0           1           1

有没有一种汇总行而不汇总结果的方法？

3 个答案: