pandas dataframe删除groupby中超过n行的组

import pandas as pd

# 读取数据到DataFrame
df = pd.read_csv('data.csv')

# 按照指定列进行分组，并计算每个组的行数
grouped = df.groupby('column_name').size()

# 设置过滤条件，筛选出超过n行的组
filtered_groups = grouped[grouped > n]

# 将超过n行的组的索引与原DataFrame进行匹配，得到需要删除的行的索引
rows_to_delete = df[df['column_name'].isin(filtered_groups.index)].index

# 删除需要删除的行
df.drop(rows_to_delete, inplace=True)

import pandas as pd

# 读取数据到DataFrame
df = pd.read_csv('data.csv')

# 按照指定列进行分组，并计算每个组的行数
grouped = df.groupby('column_name').size()

# 设置过滤条件，筛选出超过n行的组
filtered_groups = grouped[grouped > n]

# 将超过n行的组的索引与原DataFrame进行匹配，得到需要删除的行的索引
rows_to_delete = df[df['column_name'].isin(filtered_groups.index)].index

# 删除需要删除的行
df.drop(rows_to_delete, inplace=True)