pandas实现sum()over()、count()over()等窗口函数

Alex.liu

于 2020-02-10 10:05:08 发布

阅读量2.2k

点赞数

CC 4.0 BY-SA版权

分类专栏： pandas

本文链接：https://blog.csdn.net/longwei92/article/details/104244323

pandas 专栏收录该内容

15 篇文章

订阅专栏

pandas利用transform实现窗口函数

array =  [
    ['a', 2, 29],
    ['a', 1, 18],
    ['a', 0, 18],
    ['b', 2, 25],
    ['b', 4, 12],
    ['c', 6, 21],
]
data = pd.DataFrame(array, columns=["name", "class", "score"])

# 去重计数：nunique
data['nunique'] = data.groupby('name')['name'].transform('nunique')

# 求和：sum
data['sum'] = data.groupby('name')['score'].('sum')

# 均值：mean
data['mean'] = data.groupby('name')['score'].transform('mean')

# 其余统计函数类似