按特定列中的值频率限制 DataFrame 行

Question

基本上我有一个基本的 dataframe，在这个 dataframe 中有一个“流媒体服务”列。 我想将结果限制为每个服务提供商的前 5 条记录。 换句话说，我想将这个 dataframe 从可能的数千条节目记录限制到每个流媒体服务的最后 5 条。

import pandas as pd
import numpy as np

data = {'Show Name': ['GameOfThrones', 'StrangerThings', 'Casual', ...], 
        'Streaming Service': ['HBO', 'Netflix', 'Hulu']}
df1 = pd.DataFrame(data)

这样做的最佳方法是什么？

Answer 1

df1.groupby('Streaming Service').head(5)

Answer 2

我最终想出了自己的解决方案。 问题过于复杂：

service_dfs = []

for c in df['Streaming Service'].unique():
    df_c = df.loc[df[ 'Streaming Service'] == c].tail(100)
    service_dfs.append(df_c)
df = pd.concat(service_dfs)