Question

我有一个pandas数据帧如下：

user_id product_id order_number
1       1          1
1       1          2
1       1          3
1       2          1
1       2          5
2       1          1
2       1          3
2       1          4
2       1          5
3       1          1
3       1          2
3       1          6

我想查询此df表示最长条纹（没有跳过order_number）和最后一条条纹（自上次订单编号以来）。

理想的结果如下：

user_id product_id longest_streak last_streak
1       1          3              3
1       2          0              0
2       1          3              3
3       1          2              0

我很感激对此有任何见解。

Answer 1

我仍然不太确定你如何定义last_streak，但是，假设不重复用户和产品的相同组合，以下计算最长的条纹：

import itertools

def extract_streaks(data):
   streaks = [len(list(rows)) for d,rows in itertools.groupby(data) if d==1.0]
   return max(streaks) + 1 if streaks else 0

df['diffs'] = df.order_number.diff()
df.groupby(['user_id', 'product_id'])['diffs'].apply(extract_streaks)
#user_id  product_id
#1        1             3
#         2             0
#2        1             3

Answer 2

你可以尝试

s=df.assign(key=1).set_index(['user_id','product_id','order_number']).key.unstack()  s=s.notnull().astype(int).diff(axis=1).fillna(0).ne(0).cumsum(axis=1).mask(s.isnull())    
s=s.apply(pd.value_counts,1)
s=s.mask(s==1,0)    
pd.concat([s.max(1),s.ffill(axis=1).iloc[:,-1]],1)
Out[974]: 
                    0.0  2.0
user_id product_id          
1       1           3.0  3.0
        2           0.0  0.0
2       1           3.0  3.0

Answer 3

使用循环和defaultdict

a = defaultdict(lambda:None)
longest = defaultdict(int)
current = defaultdict(int)
for i, j, k in df.itertuples(index=False):
    if a[(i, j)] == k - 1:
        current[(i, j)] += 1 if current[(i, j)] else 2
        longest[(i, j)] = max(longest[(i, j)], current[(i, j)])
    else:
        current[(i, j)] = 0
        longest[(i, j)] |= 0
    a[(i, j)] = k

pd.concat(
    [pd.Series(d) for d in [longest, current]],
    axis=1, keys=['longest_streak', 'last_streak']
).rename_axis(['user_id', 'product_id']).reset_index()

   user_id  product_id  longest_streak  last_streak
0        1           1               3            3
1        1           2               0            0
2        2           1               3            3
3        3           1               2            0

熊猫 - 检查列中的数字是否在行中

3 个答案: