pandas 分组 分类型 求占比

Question

数据和结果如下，需求就是python pandas按品种分组，求属于abc类型的金额合计，按品种求金额合计，最后求出占比。

注意：如果是通过筛选数据，分组求和，在拼接求占比，那么已经实现了，现在就是想知道有没有更高级的方法，比如使用groupby.apply或者pd.pivot_table等方式实现

数据：

df = pd.DataFrame({'品种':list('AAAABBBCCD'),
                   '类型':list('abcdccdadd'),
                   '金额':[1,1,1,1,1,1,1,1,1,1]
                  })

品种	类型	金额
A	a	1
A	b	1
A	c	1
A	d	1
B	c	1
B	c	1
B	d	1
C	a	1
C	d	1
D	d	1

结果：

品种	属于abc类型的金额汇总	按品种汇总金额	占比
A	3	4	75.00%
B	2	3	66.67%
C	1	2	50.00%
D	0	1	0.00%

Answer 1

import pandas as pd
import pandasql as pdsql

df = pd.DataFrame({'品种':list('AAAABBBCCD'),
                   '类型':list('abcdccdadd'),
                   '金额':[1,1,1,1,1,1,1,1,1,1]
                   })

sql = """
SELECT 品种,
SUM(CASE WHEN 类型 IN ('a','b','c') 
    THEN 金额 
    ELSE 0 
    END
    ) '属于abc类型的金额汇总',
    SUM(金额) '按品种汇总金额',
    PRINTF('%2.2f%%', 
    SUM(CASE WHEN 类型 IN ('a','b','c') 
    THEN 金额 
    ELSE 0 
    END)*1.0/SUM(金额)) '占比'
FROM df 
GROUP BY 品种"""

f = lambda x: pdsql.sqldf(x, globals())
res = f(sql)
print(res)

--result
  品种  属于abc类型的金额汇总  按品种汇总金额      占比
0  A             3        4  75.00%
1  B             2        3  66.67%
2  C             1        2  50.00%
3  D             0        1   0.00%