首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >多列范畴变量(2)

多列范畴变量(2)
EN

Stack Overflow用户
提问于 2018-06-18 07:26:45
回答 2查看 134关注 0票数 2

我有一个带有分类变量段的数据

代码语言:javascript
复制
ID  Segment Var1 Var2 Var3
1   AAA     1    1    1
2   BBB     1    0    1
3   BBB     1    1    1
4   AAA     0    1    1
5   CCC     0    1    1
6   AAA     0    0    1
7   AAA     0    1    1
8   AAA     1    0    1
9   BBB     1    0    1
10  CCC     1    0    1

我想为每个变量将列段转换为3类,如下所示:

代码语言:javascript
复制
ID Var1_AAA Var1_BBB Var1_CCC Var2_AAA Var2_BBB Var2_CC Var3_AAA Var3_BBB  Var3_CCC
1  1        null     null     1         null    null    1        null     null
2  null     1        null     null      0       null    null     1        null
3  null     1        null     null      1       null    null     1        null
4  0        null     null     1         null    null    1        null     null
5  null     null     0        null      null    1       null     null     1
6  0        null     null     0         null    null    1        null     null
7  0        null     null     1         null    null    1        null     null
8  1        null     null     0         null    null    1        null     null
9  null     1        null     null      0       null    null     1        null
10 null     null     1        null      null    0       null     null     1

你能帮我一下吗?

EN

回答 2

Stack Overflow用户

回答已采纳

发布于 2018-06-18 07:33:56

meltpivot_table

代码语言:javascript
复制
v = df.melt(['ID', 'Segment'])
v = v.pivot_table(index='ID', 
                  columns=['Segment', 'variable'], 
                  values='value', 
                  fill_value='null')
v.columns = v.columns.map('{0[1]}_{0[0]}'.format)

代码语言:javascript
复制
print(v)

   Var1_AAA Var2_AAA Var3_AAA Var1_BBB Var2_BBB Var3_BBB Var1_CCC Var2_CCC  \
ID                                                                           
1         1        1        1     null     null     null     null     null   
2      null     null     null        1        0        1     null     null   
3      null     null     null        1        1        1     null     null   
4         0        1        1     null     null     null     null     null   
5      null     null     null     null     null     null        0        1   
6         0        0        1     null     null     null     null     null   
7         0        1        1     null     null     null     null     null   
8         1        0        1     null     null     null     null     null   
9      null     null     null        1        0        1     null     null   
10     null     null     null     null     null     null        1        0   

   Var3_CCC  
ID           
1      null  
2      null  
3      null  
4      null  
5         1  
6      null  
7      null  
8      null  
9      null  
10        1  

我建议省略fill_value='null',代之以fill_value=0或完全删除,因为将字符串与数字数据混在一起会影响性能。但是,如果您只想保存结果而不需要更多的话,这是可以的。

票数 3
EN

Stack Overflow用户

发布于 2018-06-18 07:44:06

另一种方法是,

使用unstack格式和数据格式列名

代码语言:javascript
复制
df= df.set_index(['ID','Segment']).unstack(fill_value='null')
df.columns = ['_'.join(val ) for val in zip(df.columns.get_level_values(0),df.columns.get_level_values(1))] 

或者用更优雅的方式格式化列,

代码语言:javascript
复制
df.columns = df.columns.map('{0[0]}_{0[1]}'.format)
print df

输出:

代码语言:javascript
复制
   Var1_AAA Var1_BBB Var1_CCC Var2_AAA Var2_BBB Var2_CCC Var3_AAA Var3_BBB  \
ID                                                                           
1         1     null     null        1     null     null        1     null   
2      null        1     null     null        0     null     null        1   
3      null        1     null     null        1     null     null        1   
4         0     null     null        1     null     null        1     null   
5      null     null        0     null     null        1     null     null   
6         0     null     null        0     null     null        1     null   
7         0     null     null        1     null     null        1     null   
8         1     null     null        0     null     null        1     null   
9      null        1     null     null        0     null     null        1   
10     null     null        1     null     null        0     null     null   

   Var3_CCC  
ID           
1      null  
2      null  
3      null  
4      null  
5         1  
6      null  
7      null  
8      null  
9      null  
10        1 
票数 1
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/50904510

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档