Advertisement
Korotkodul

A. Ну что там с деньгами?

Feb 23rd, 2025
248
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 2.60 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.     sold_per_month = np.sum(income, axis = 1)
  16.     max_id = np.argmax(sold_per_month)
  17.     return months_list[max_id]
  18.  
  19.  
  20. from typing import Optional
  21. from enum import Enum
  22. from numbers import Real
  23.  
  24. class Strategies(Enum):
  25.     BY_GOOD = "BY_GOOD"   # придумайте значение enum'а
  26.     BY_MONTH = "BY_MONTH"  # придумайте значение enum'а
  27.  
  28.  
  29. def get_mean_profit(
  30.         amounts_of_sold_subscriptions: np.ndarray,
  31.         subscriptions_prices: np.ndarray,
  32.         strategy: Optional[Strategies] = None,
  33. ):
  34.     if amounts_of_sold_subscriptions.shape[1] != subscriptions_prices.shape[1]:
  35.         raise InconsistentDataError
  36.     income = amounts_of_sold_subscriptions * subscriptions_prices
  37.     income_per_month = np.sum(income, axis=1)
  38.     income_per_year = np.sum(income_per_month)
  39.     amounts_of_sold_subscriptions_per_month = np.sum(amounts_of_sold_subscriptions, axis = 1)
  40.     amounts_of_sold_subscriptions_per_year = np.sum(amounts_of_sold_subscriptions_per_month)
  41.     amounts_of_sold_subscriptions_per_service = np.sum(amounts_of_sold_subscriptions, axis = 0)
  42.     income_per_servise_per_year = np.sum(income, axis = 0)
  43.     res = None
  44.     if strategy == None:
  45.         res = income_per_year / amounts_of_sold_subscriptions_per_year
  46.     elif strategy == Strategies.BY_GOOD:
  47.         res = income_per_servise_per_year / amounts_of_sold_subscriptions_per_service
  48.     elif strategy == Strategies.BY_MONTH:
  49.         res = income_per_month / amounts_of_sold_subscriptions_per_month
  50.     return res
  51.  
  52.  
  53. import calendar
  54. months_list = np.array([calendar.month_name[i] for i in range(1, 13)])
  55.  
  56. def sort_month_names_by_profits(
  57.     amounts_of_sold_subscriptions: np.ndarray,
  58.     subscriptions_prices: np.ndarray,
  59.     ascending: bool = True,
  60. ) -> list[str]:
  61.     if amounts_of_sold_subscriptions.shape[1] != subscriptions_prices.shape[1]:
  62.         raise InconsistentDataError
  63.     income = amounts_of_sold_subscriptions * subscriptions_prices
  64.     income_per_month = np.sum(income, axis=1)
  65.     indices = np.argsort(income_per_month)
  66.     res = months_list[indices]
  67.     if not ascending:
  68.         res = res[::-1]
  69.     return res
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement