У меня есть фрейм данных с большим количеством магазинов, и я пытаюсь удалить конкретный магазин с помощью этой функции:
def no_todos(df, n):
# remove shop 9
df = df.loc[df['shop_id'] != 9]
# since I also need a particular amount of observations from each original df this part
# of the function returns the amount of observations required
return df._selected_obj[df.cumcount(ascending = True) >= n].reset_index(drop = True)
# here I call the function
camisas_2prueba = no_todos(camisas1.groupby('shop_id'), 365).copy()
Однако я не могу удалить этот магазин, и вместо этого я получил эту ошибку :
Cannot access callable attribute 'loc' of 'DataFrameGroupBy' objects, try using the 'apply' method
Если кто-то знает, как решить эту проблему, я был бы очень признателен за помощь. Заранее спасибо.