Advertisement
Korotkodul

A. Часть 1. Самый прибыльный месяц_DRAFT

Feb 22nd, 2025 (edited)
517
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 1.08 KB | None | 0 0
  1. import numpy as np
  2. import calendar
  3. months_list = [calendar.month_name[i] for i in range(1, 13)]
  4.  
  5. class InconsistentDataError(Exception):
  6.     pass
  7.  
  8. def get_most_profitable_month_name(
  9.     amounts_of_sold_subscriptions: np.ndarray,
  10.     subscriptions_prices: np.ndarray,
  11. ) -> str:
  12.     if amounts_of_sold_subscriptions.shape[1] != subscriptions_prices.shape[1]:
  13.         raise InconsistentDataError
  14.     income = amounts_of_sold_subscriptions * subscriptions_prices
  15.     print("income")
  16.     print(income)
  17.     sold_per_month = np.sum(income, axis = 1)
  18.     print("sold_per_month")
  19.     print(sold_per_month)
  20.     max_id = np.argmax(sold_per_month)
  21.     print("max_id", max_id)
  22.     return months_list[max_id]
  23.  
  24. amounts_of_sold_subscriptions = np.random.randint(0, 10, size=(12, 3))
  25. subscriptions_prices = np.random.randint(1, 10, size=(1, 3))
  26. print("amounts_of_sold_subscriptions")
  27. print(amounts_of_sold_subscriptions)
  28. print("subscriptions_prices")
  29. print(subscriptions_prices)
  30.  
  31. res = get_most_profitable_month_name(amounts_of_sold_subscriptions, subscriptions_prices)
  32. print("res", res)
  33.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement