在更宽的数据帧中转换虚拟对象中的变量列表

Question

我已经导入了一个json文件，现在有了一个数据框，其中一列（代码）是一个列表

index year   gvkey    code
0    1998    15686    ['TAX', 'ENVR', 'HEALTH']
1    2005    15372    ['EDUC', 'TAX', 'HEALTH', 'JUST']
2    2001    27486    ['LAB', 'TAX', 'HEALTH']
3    2008    84967    ['HEALTH','LAB', 'JUST']

我想得到的东西如下：

index year   gvkey  TAX  ENVR HEALTH EDUC JUST LAB
0    1998    15686   1     1     1    0    0    0
1    2005    15372   1     0     1    0    1    0
2    2001    27486   1     0     1    0    1    0
3    2008    84967   0     0     1    0    1    1

在Pandas convert a column of list to dummies之后，我尝试了以下代码（其中df是我的数据帧）：

s = pd.Series(df["code"])
l = pd.get_dummies(s.apply(pd.Series).stack()).sum(level=0)

我正确地获得了数据的第二部分（变量TAX、ENVR、HEALTH、EDUC、JUST和LAB），但失去了第一部分（year和gvkey）

如何保持年份和gvkey变量

Answer 1

您可以通过以下方法执行此操作：

方法1：将列转换为数据帧并获取Dummie，然后groupbyon axis=1并获取max:

m = pd.get_dummies(pd.DataFrame(df['code'].tolist())).groupby(lambda x:
    x.split('_')[1],axis=1).max()
final1 = df.drop('code',1).assign(**m)

方法2：用|连接列列表并使用series.str.get_dummies

final2 = df.drop('code',1).assign(**df['code'].str.join('|').str.get_dummies())

方法3：使用concat的方法

s = pd.Series(df["code"])
l = pd.get_dummies(s.apply(pd.Series).stack()).max(level=0)
final3 = pd.concat((df.drop('code',1),l),axis=1)
#or final = df.drop('code',1).assign(**l)