基于Numpy和Pandas的求dataframe某一列元素连续重复的次数：一个小trick

feihua1475

已于 2023-02-05 22:25:41 修改

阅读量1k

点赞数 2

文章标签： pandas numpy python

于 2022-07-27 13:30:34 首次发布

本文链接：https://blog.csdn.net/feihua1475/article/details/126012667

版权

问题：df是一个dataframe，其a列df['a'] = [1,1,1,...,-1,-1,-1,...,0,0,0,...,1,1,1]的多个连续重复元素组成如x1个1，x2个0，x3个-1，x4个1，....。当前想要获得一个list，list中每个元素对应连续重复次数[x1,x2,x3,x4,...,xn]。

import pandas as pd
import numpy as np

### 前面已经定义好了df，且df['a']中不包括nan值，否则用df['a'].dropna()替换df['a']
b = np.where(df['a'] != df['a'].shift(1))
np.diff(b)[0] # 所得为[x1,x2,...,xn-1]
# 计算 xn
len(df) - sum(np.diff(b)[0])