Python中带有三个变量的R中的表函数结果

2024-04-25 04:58:21 发布

您现在位置:Python中文网/ 问答频道 /正文

假设我们有这个样本数据。你知道吗

| mem_id | main_title | sub_title |
-----------------------------------
|   1    |      1     |    1      |
|   10   |      3     |    2      |
|   3    |      3     |    2      |
|   45   |      1     |    2      |
|   162  |      2     |    2      |

... 

1)数据汇总

  • mem\u id:200人的唯一id
  • 主标题:3个独特的标签(1、2、3)
  • 子标题:6个唯一标签(1、2、3、4、5、6),每个主标题可以有一个子标题。你知道吗

可能有重复,就像一个mem\u id可以有多个case of(1:main,1:sub)

2)问题

我想用python生成R table函数result。你知道吗

R表函数的结果是这样的。 我可以从所有的主标题和副标题中做出各种可能的组合。 还可以通过mem\u id获取每个案例的计数

count.data <- table(data$mem_id, data$main_title, data$sub_title)
count.table <- as.data.frame(count.data)
===============================================
          mem_id   main_title   sub_title value
1            1         1              1     0
2            2         1              1     0
3            3         1              1     0
4            4         1              1     0
5            5         1              1     0
6            6         1              1     0
7            7         1              1     0
.
.
.

我试着用Python得到这个结果,下面的结果就是我目前得到的结果。你知道吗

cross_table1 = pd.melt(data, id_vars=['main_title ', 'sub_title'], value_vars='mem_id', value_name='mem_id')

==================================================
         main_title sub_title  variable   mem_id
1            1         1         mem_id     10
2            1         1         mem_id     10
3            3         1         mem_id     10
4            4         2         mem_id     10
5            1         4         mem_id     132
6            4         1         mem_id     65
7            4         3         mem_id     88
.
.
.
cross_table2 = cross_table1.pivot_table(index=['main_title ', 'sub_title', 'mem_id'], values='variable', aggfunc='count')

cross_table32.reset_index().sort_values('value')

==============================================
         main_title sub_title  mem_id    value
1            1         1         1         4
2            1         1         2         3
3            3         1         3         1
4            4         2         3         10
5            1         4         3         2
6            1         1         4         5
7            3         2         5         2
.
.
.

我认识到这只显示值(事例计数)列的正结果。你知道吗

我需要的是包含所有可能的主标题和子标题的组合,所以像1&1(main&sub)case必须有200行,count列中可能有零值。你知道吗

如果我能得到任何帮助或建议,我将不胜感激!! 谢谢:)


Tags: 数据函数id标题datatitlevaluemain
1条回答
网友
1楼 · 发布于 2024-04-25 04:58:21

在熊猫中,你可以用groupby+reindex

s=df.groupby(df.columns.tolist()).size()
idx=pd.MultiIndex.from_product(list(map(set,df.values.T)))
s=s.reindex(idx,fill_value=0)
s
Out[15]: 
162  1  1    0
        2    0
     2  1    0
        2    1
     3  1    0
        2    0
1    1  1    1
        2    0
     2  1    0
        2    0
     3  1    0
        2    0
10   1  1    0
        2    0
     2  1    0
        2    0
     3  1    0
        2    1
3    1  1    0
        2    0
     2  1    0
        2    0
     3  1    0
        2    1
45   1  1    0
        2    1
     2  1    0
        2    0
     3  1    0
        2    0
dtype: int64

相关问题 更多 >