向原始数据框添加虚拟列

Question

我有一个数据框看起来像这样：

             JOINED_CO GENDER EXEC_FULLNAME GVKEY年CONAME BECAMECEO重新加入LEFTOFC LEFTCO RELEFT原因页面
CO_PER_ROL                                                                                                                                     
5622 NaN MALE Ira A.Eichner 1004 1992 AAR CORP 19550101 NaN 19961001 19990531 NaN辞职79
5622 NaN MALE Ira A.Eichner 1004 1993 AAR CORP 19550101 NaN 19961001 19990531 NaN辞职79
5622 NaN MALE Ira A.Eichner 1004 1994 AAR CORP 19550101 NaN 19961001 19990531 NaN辞职79
5622 NaN MALE Ira A.Eichner 1004 1995 AAR CORP 19550101 NaN 19961001 19990531 NaN辞职79
5622 NaN MALE Ira A.Eichner 1004 1996 AAR CORP 19550101 NaN 19961001 19990531 NaN辞职79
5622 NaN MALE Ira A.Eichner 1004 1997 AAR CORP 19550101 NaN 19961001 19990531 NaN辞职79
5622 NaN MALE Ira A.Eichner 1004 1998 AAR CORP 19550101 NaN 19961001 19990531 NaN辞职79
5623 NaN MALE David P.Storch 1004 1992 AAR CORP 19961009 NaN NaN NaN NaN NaN 57
5623 NaN MALE David P.Storch 1004 1993 AAR CORP 19961009 NaN NaN NaN NaN NaN 57
5623 NaN MALE David P.Storch 1004 1994 AAR CORP 19961009 NaN NaN NaN NaN NaN 57
5623 NaN MALE David P.Storch 1004 1995 AAR CORP 19961009 NaN NaN NaN NaN NaN 57
5623 NaN MALE David P.Storch 1004 1996 AAR CORP 19961009 NaN NaN NaN NaN NaN 57

对于YEAR值，我想在原始数据帧中添加year列（1993,1994 …，2009）。如果YEAR中的值为1992，则1992列中的值应为1，否则为0。

我使用了一个非常愚蠢的for循环，但是由于我拥有大量数据集，它似乎可以永远运行。谁能帮助我，非常感谢！

Answer 1

In [77]: df = pd.concat([df, pd.get_dummies(df['YEAR'])], axis=1); df
Out[77]: 
      JOINED_CO GENDER    EXEC_FULLNAME  GVKEY  YEAR    CONAME  BECAMECEO  \
5622        NaN   MALE   Ira A. Eichner   1004  1992  AAR CORP   19550101   
5622        NaN   MALE   Ira A. Eichner   1004  1993  AAR CORP   19550101   
5622        NaN   MALE   Ira A. Eichner   1004  1994  AAR CORP   19550101   
5622        NaN   MALE   Ira A. Eichner   1004  1995  AAR CORP   19550101   
5622        NaN   MALE   Ira A. Eichner   1004  1996  AAR CORP   19550101   
5622        NaN   MALE   Ira A. Eichner   1004  1997  AAR CORP   19550101   
5622        NaN   MALE   Ira A. Eichner   1004  1998  AAR CORP   19550101   
5623        NaN   MALE  David P. Storch   1004  1992  AAR CORP   19961009   
5623        NaN   MALE  David P. Storch   1004  1993  AAR CORP   19961009   
5623        NaN   MALE  David P. Storch   1004  1994  AAR CORP   19961009   
5623        NaN   MALE  David P. Storch   1004  1995  AAR CORP   19961009   
5623        NaN   MALE  David P. Storch   1004  1996  AAR CORP   19961009

      REJOIN   LEFTOFC    LEFTCO  RELEFT    REASON  PAGE  1992  1993  1994  \
5622     NaN  19961001  19990531     NaN  RESIGNED    79     1     0     0   
5622     NaN  19961001  19990531     NaN  RESIGNED    79     0     1     0   
5622     NaN  19961001  19990531     NaN  RESIGNED    79     0     0     1   
5622     NaN  19961001  19990531     NaN  RESIGNED    79     0     0     0   
5622     NaN  19961001  19990531     NaN  RESIGNED    79     0     0     0   
5622     NaN  19961001  19990531     NaN  RESIGNED    79     0     0     0   
5622     NaN  19961001  19990531     NaN  RESIGNED    79     0     0     0   
5623     NaN       NaN       NaN     NaN       NaN    57     1     0     0   
5623     NaN       NaN       NaN     NaN       NaN    57     0     1     0   
5623     NaN       NaN       NaN     NaN       NaN    57     0     0     1   
5623     NaN       NaN       NaN     NaN       NaN    57     0     0     0   
5623     NaN       NaN       NaN     NaN       NaN    57     0     0     0

      1995  1996  1997  1998  
5622     0     0     0     0  
5622     0     0     0     0  
5622     0     0     0     0  
5622     1     0     0     0  
5622     0     1     0     0  
5622     0     0     1     0  
5622     0     0     0     1  
5623     0     0     0     0  
5623     0     0     0     0  
5623     0     0     0     0  
5623     1     0     0     0  
5623     0     1     0     0

如果您想删除该YEAR列，则可以通过进行后续操作del df['YEAR']。或者，在调用之前YEAR从中删除列：df``concat

df = pd.concat([df.drop('YEAR', axis=1), pd.get_dummies(df['YEAR'])], axis=1)