Advertisement
Korotkodul

A. Подкрутка

Apr 17th, 2025 (edited)
264
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.72 KB | None | 0 0
  1. import os
  2. import pandas as pd
  3. import kagglehub
  4.  
  5.  
  6. path = kagglehub.dataset_download("ersany/online-retail-dataset")
  7. df = pd.read_excel(os.path.join(path, "Online Retail.xlsx"))
  8.  
  9.  
  10. def change_price(group):
  11.     first_letter = group.name #A, B, C...
  12.  
  13.     if first_letter == 'A':
  14.         max_price = group['UnitPrice'].max()
  15.         group['UnitPrice'] = max_price
  16.     elif first_letter == 'B':
  17.         min_price = group['UnitPrice'].min()
  18.         group['UnitPrice'] = group['UnitPrice'] - (min_price / 2)
  19.     else:
  20.         group['UnitPrice'] = group['UnitPrice'] * 1.1
  21.  
  22.     return group
  23.  
  24.  
  25. def task1(df):
  26.     df = df.groupby(df['Country'].str[0], group_keys=False).apply(change_price)
  27.    
  28.     return df
  29.  
  30. df = task1(df)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement