时间戳

首页 > 解决方案 > 在 pandas 中运行多个具有函数的交叉表

python - 在 pandas 中运行多个具有函数的交叉表

问题描述

嗨，我正在尝试制作一些列联表。我想要它在一个函数中，这样我就可以将它用于各种列/数据框/组合等。

当前我有一个看起来像这样的数据框

df = pd.DataFrame(data={'group' : ['A','A','B','B','C','D'],
                        'class': ['g1','g2','g2','g3','g1','g2'],
                        'total' : ['0-10','20-30','0-10','30-40','50-60','20-30'],
                        'sub' : ['1-4', '5-9','10-14', '15-19','1-4','15-19'],
                        'n': [3,14,12,11,21,9]})

和一个看起来像这样的函数

def cts(tabs, df):
    out=[]
    for col in df.loc[:,df.columns != tabs]:
        a = pd.crosstab([df[tabs]], df[col])
        out.append(a)
    return(out)
cts('group', df)

它适用于一列与其余列的交叉表。但我想在分组中添加两个（或更多！）级别，例如

pd.crosstab([df['group'], df['class']], df['total'])

其中总计是针对组和类进行交叉制表的。

我认为函数中的 'tabs' var 应该是列名列表，但是当我尝试将其设为列表时，我得到错误是语法无效。我希望这是有道理的..谢谢！

标签： pythonpandascrosstab

解决方案

尝试：

def cts(tabs, df):
    out=[]
    cols = [col for col in df.columns if col not in tabs]
    for col in df.loc[:,cols]:
        a = pd.crosstab([df[tab] for tab in tabs], df[col])
        out.append(a)
    return(out)

推荐阅读