main.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 68,
   "metadata": {},
   "outputs": [],
   "source": [
    "import pandas as pd\n",
    "from sklearn.model_selection import train_test_split\n",
    "from random import randrange"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 69,
   "metadata": {},
   "outputs": [],
   "source": [
    "\n",
    "raw_shelves_df = pd.read_csv('data/shelves.csv')\n",
    "raw_shedules_df = pd.read_csv('data/shedules.csv')\n",
    "raw_products_df = pd.read_csv('data/products.csv')\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 70,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>id</th>\n",
       "      <th>store</th>\n",
       "      <th>schedule_name</th>\n",
       "      <th>shelves</th>\n",
       "      <th>day</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>5018</td>\n",
       "      <td>5018ДС_СПб_Полюстровский59</td>\n",
       "      <td>Индивидуальный график постеллажной. ДС 5018</td>\n",
       "      <td>A1;A10;A10;A10;A10;A10;A10;A10;A10;A11;A11;A11...</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>5018</td>\n",
       "      <td>5018ДС_СПб_Полюстровский59</td>\n",
       "      <td>Индивидуальный график постеллажной. ДС 5018</td>\n",
       "      <td>K1;K10;K11;K12;K13;K14;K15;K16;K17;K18;K19;K2;...</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>5018</td>\n",
       "      <td>5018ДС_СПб_Полюстровский59</td>\n",
       "      <td>Индивидуальный график постеллажной. ДС 5018</td>\n",
       "      <td>G13;O1;O2;O3;O4;O5;O6;O7;O8;Г1;Г10;Г11;Г12;Г2;...</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>5018</td>\n",
       "      <td>5018ДС_СПб_Полюстровский59</td>\n",
       "      <td>Индивидуальный график постеллажной. ДС 5018</td>\n",
       "      <td>Z1;Z10;Z12;Z13;Z14;Z15;Z16;Z17;Z18;Z2;Z3;Z4;Z5...</td>\n",
       "      <td>3</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>5018</td>\n",
       "      <td>5018ДС_СПб_Полюстровский59</td>\n",
       "      <td>Индивидуальный график постеллажной. ДС 5018</td>\n",
       "      <td>Z18;Z19;Z20;Z21;Z22;Z23;Z24;Z25;Z26;Z27;Z28;Z2...</td>\n",
       "      <td>4</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "     id                       store  \\\n",
       "0  5018  5018ДС_СПб_Полюстровский59   \n",
       "1  5018  5018ДС_СПб_Полюстровский59   \n",
       "2  5018  5018ДС_СПб_Полюстровский59   \n",
       "3  5018  5018ДС_СПб_Полюстровский59   \n",
       "4  5018  5018ДС_СПб_Полюстровский59   \n",
       "\n",
       "                                 schedule_name  \\\n",
       "0  Индивидуальный график постеллажной. ДС 5018   \n",
       "1  Индивидуальный график постеллажной. ДС 5018   \n",
       "2  Индивидуальный график постеллажной. ДС 5018   \n",
       "3  Индивидуальный график постеллажной. ДС 5018   \n",
       "4  Индивидуальный график постеллажной. ДС 5018   \n",
       "\n",
       "                                             shelves  day  \n",
       "0  A1;A10;A10;A10;A10;A10;A10;A10;A10;A11;A11;A11...    0  \n",
       "1  K1;K10;K11;K12;K13;K14;K15;K16;K17;K18;K19;K2;...    1  \n",
       "2  G13;O1;O2;O3;O4;O5;O6;O7;O8;Г1;Г10;Г11;Г12;Г2;...    2  \n",
       "3  Z1;Z10;Z12;Z13;Z14;Z15;Z16;Z17;Z18;Z2;Z3;Z4;Z5...    3  \n",
       "4  Z18;Z19;Z20;Z21;Z22;Z23;Z24;Z25;Z26;Z27;Z28;Z2...    4  "
      ]
     },
     "execution_count": 70,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "shedules_df = raw_shedules_df.rename(columns={\n",
    "    'N': 'id',\n",
    "    'name_TT': 'store',\n",
    "    'НаименованиеГрафика': 'schedule_name',\n",
    "    'Длительность (недель)': 'duration_weeks',\n",
    "    'Стеллажи': 'shelves',\n",
    "    'неделя 1': 'week_1',\n",
    "    'неделя 2': 'week_2',\n",
    "    'неделя 3': 'week_3',\n",
    "    'неделя 4': 'week_4',\n",
    "    'Кол-во дней': 'days_count',\n",
    "    'ЧастотаПостеллажная': 'shelf_frequency',\n",
    "    'ЧастотаГрафика': 'schedule_frequency'\n",
    "}, inplace=False)\n",
    "\n",
    "shedules_df = shedules_df[shedules_df['store'].str.contains(\"5018ДС_СПб_Полюстровский59\", case=False, na=False)]\n",
    "\n",
    "shedules_df = shedules_df.melt(\n",
    "    id_vars=['id', 'store', 'schedule_name', 'duration_weeks', 'shelves', 'days_count', 'shelf_frequency', 'schedule_frequency'],\n",
    "    value_vars=['week_1', 'week_2', 'week_3', 'week_4'],\n",
    "    var_name='week_number',\n",
    "    value_name='week_day'\n",
    ")\n",
    "\n",
    "shedules_df = shedules_df.dropna(subset=['week_day']).reset_index(drop=True)\n",
    "\n",
    "day_of_week_map = {'пн': 0, 'вт': 1, 'ср': 2, 'чт': 3, 'пт': 4, 'сб': 5, 'вс': 6}\n",
    "\n",
    "shedules_df['day_index'] = shedules_df['week_day'].map(day_of_week_map)\n",
    "shedules_df['day'] = shedules_df.apply(\n",
    "    lambda row: (int(row['week_number'][-1]) - 1) * 7 + row['day_index'], axis=1\n",
    ")\n",
    "\n",
    "\n",
    "shedules_df = shedules_df.drop(columns=['week_number', 'week_day', 'day_index', 'schedule_frequency', 'shelf_frequency', 'days_count', 'duration_weeks'])\n",
    "shedules_df = shedules_df.sort_values('day').reset_index(drop=True)\n",
    "\n",
    "shedules_df.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 71,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>id_tov</th>\n",
       "      <th>name</th>\n",
       "      <th>category</th>\n",
       "      <th>group</th>\n",
       "      <th>total_carts</th>\n",
       "      <th>system</th>\n",
       "      <th>per_month</th>\n",
       "      <th>avg_per_month</th>\n",
       "      <th>sum_per_month</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>23 074</td>\n",
       "      <td>Манго Египет</td>\n",
       "      <td>Овощи. Фрукты. Грибы. Зелень</td>\n",
       "      <td>Манго (все виды)</td>\n",
       "      <td>8352</td>\n",
       "      <td>кг</td>\n",
       "      <td>6972.48</td>\n",
       "      <td>0.83</td>\n",
       "      <td>2194135.41</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>731</td>\n",
       "      <td>Бананы</td>\n",
       "      <td>Овощи. Фрукты. Грибы. Зелень</td>\n",
       "      <td>Банан</td>\n",
       "      <td>6869</td>\n",
       "      <td>кг</td>\n",
       "      <td>6338.46</td>\n",
       "      <td>0.92</td>\n",
       "      <td>952972.7</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>647</td>\n",
       "      <td>Пакет-майка \"ВкусВилл\" малый</td>\n",
       "      <td>Нон-Фуд</td>\n",
       "      <td>Сумки, пакеты, упаковка</td>\n",
       "      <td>87843</td>\n",
       "      <td>шт</td>\n",
       "      <td>96923.00</td>\n",
       "      <td>1.10</td>\n",
       "      <td>756861.36</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>19 419</td>\n",
       "      <td>Бедро цыпленка-бройлера бескостное (филе бедра)</td>\n",
       "      <td>Мясо. Птица. П/ф охл. Яйца</td>\n",
       "      <td>Курица бескостное мясо</td>\n",
       "      <td>1504</td>\n",
       "      <td>кг</td>\n",
       "      <td>1438.22</td>\n",
       "      <td>0.96</td>\n",
       "      <td>723763.71</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>669</td>\n",
       "      <td>Арбуз</td>\n",
       "      <td>Овощи. Фрукты. Грибы. Зелень</td>\n",
       "      <td>Бахчевые</td>\n",
       "      <td>1621</td>\n",
       "      <td>кг</td>\n",
       "      <td>11265.38</td>\n",
       "      <td>6.95</td>\n",
       "      <td>683772.7</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "   id_tov                                             name  \\\n",
       "0  23 074                                     Манго Египет   \n",
       "1     731                                           Бананы   \n",
       "2     647                     Пакет-майка \"ВкусВилл\" малый   \n",
       "3  19 419  Бедро цыпленка-бройлера бескостное (филе бедра)   \n",
       "4     669                                            Арбуз   \n",
       "\n",
       "                       category                    group  total_carts system  \\\n",
       "0  Овощи. Фрукты. Грибы. Зелень         Манго (все виды)         8352     кг   \n",
       "1  Овощи. Фрукты. Грибы. Зелень                    Банан         6869     кг   \n",
       "2                       Нон-Фуд  Сумки, пакеты, упаковка        87843     шт   \n",
       "3    Мясо. Птица. П/ф охл. Яйца   Курица бескостное мясо         1504     кг   \n",
       "4  Овощи. Фрукты. Грибы. Зелень                 Бахчевые         1621     кг   \n",
       "\n",
       "   per_month  avg_per_month sum_per_month  \n",
       "0    6972.48           0.83    2194135.41  \n",
       "1    6338.46           0.92      952972.7  \n",
       "2   96923.00           1.10     756861.36  \n",
       "3    1438.22           0.96     723763.71  \n",
       "4   11265.38           6.95      683772.7  "
      ]
     },
     "execution_count": 71,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "products_df = raw_products_df.rename(columns={\n",
    "    'id_tov': 'id_tov',\n",
    "    'Наименование': 'name',\n",
    "    'Категория': 'category',\n",
    "    'Группа': 'group',\n",
    "    'Кол-во чеков за месяц': 'total_carts',\n",
    "    'Единица измерения': 'system',\n",
    "    'Кол-во за месяц': 'per_month',\n",
    "    'Среднее кол-во за месяц': 'avg_per_month',\n",
    "    'Сумма продаж за месяц': 'sum_per_month',\n",
    "}, inplace=False)\n",
    "\n",
    "# products_df = products_df[products_df['category'].str.contains(\"Овощи. Фрукты. Грибы. Зелень\", case=False, na=False)]\n",
    "\n",
    "products_df = products_df.reset_index(drop=True, inplace=False)\n",
    "\n",
    "def is_float(value):\n",
    "    try:\n",
    "        float(value)\n",
    "        return True\n",
    "    except ValueError:\n",
    "        return False\n",
    "    \n",
    "\n",
    "# Apply the filter to keep only rows where 'per_month' can be converted to float\n",
    "products_df = products_df[products_df['per_month'].apply(is_float)]\n",
    "\n",
    "# Convert the 'per_month' column to float after filtering\n",
    "products_df['per_month'] = products_df['per_month'].astype(float)\n",
    "\n",
    "products_df.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 72,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>name</th>\n",
       "      <th>price</th>\n",
       "      <th>shelve</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>Апельсины</td>\n",
       "      <td>215</td>\n",
       "      <td>A7</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>Апельсины Навелин</td>\n",
       "      <td>289</td>\n",
       "      <td>A3</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>Арбуз</td>\n",
       "      <td>67</td>\n",
       "      <td>A7</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>Баклажаны грунтовые</td>\n",
       "      <td>138</td>\n",
       "      <td>Fr1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>Бананы</td>\n",
       "      <td>158</td>\n",
       "      <td>A7</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "                  name  price shelve\n",
       "0            Апельсины    215     A7\n",
       "1    Апельсины Навелин    289     A3\n",
       "2                Арбуз     67     A7\n",
       "3  Баклажаны грунтовые    138    Fr1\n",
       "4               Бананы    158     A7"
      ]
     },
     "execution_count": 72,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "\n",
    "shelves_df = raw_shelves_df\n",
    "shelves_df['shelve'] = shelves_df['shelve'].apply(lambda x: ''.join(x.split('-')[:2]))\n",
    "\n",
    "shelves_df.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 73,
   "metadata": {},
   "outputs": [],
   "source": [
    "def determine_trigger(sku_id, name, amount, avg_amount_in_cart, day_index):\n",
    "  shelf_df = shelves_df[shelves_df['name'] == name]['shelve']\n",
    "  shelf_df = shelf_df.reset_index(drop=True)\n",
    "  \n",
    "  day_df = shedules_df[shedules_df['day'] == day_index]\n",
    "  day_df = day_df.reset_index(drop=True)\n",
    "  \n",
    "  if (not shelf_df.any() or not day_df['shelves'].any() or amount < avg_amount_in_cart): \n",
    "    return False\n",
    "  \n",
    "  return shelf_df[0] in day_df['shelves'][0]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 74,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>id_tov</th>\n",
       "      <th>name</th>\n",
       "      <th>category</th>\n",
       "      <th>group</th>\n",
       "      <th>total_carts</th>\n",
       "      <th>system</th>\n",
       "      <th>per_month</th>\n",
       "      <th>avg_per_month</th>\n",
       "      <th>sum_per_month</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>23 074</td>\n",
       "      <td>Манго Египет</td>\n",
       "      <td>Овощи. Фрукты. Грибы. Зелень</td>\n",
       "      <td>Манго (все виды)</td>\n",
       "      <td>8352</td>\n",
       "      <td>кг</td>\n",
       "      <td>6972.48</td>\n",
       "      <td>0.83</td>\n",
       "      <td>2194135.41</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>731</td>\n",
       "      <td>Бананы</td>\n",
       "      <td>Овощи. Фрукты. Грибы. Зелень</td>\n",
       "      <td>Банан</td>\n",
       "      <td>6869</td>\n",
       "      <td>кг</td>\n",
       "      <td>6338.46</td>\n",
       "      <td>0.92</td>\n",
       "      <td>952972.7</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>647</td>\n",
       "      <td>Пакет-майка \"ВкусВилл\" малый</td>\n",
       "      <td>Нон-Фуд</td>\n",
       "      <td>Сумки, пакеты, упаковка</td>\n",
       "      <td>87843</td>\n",
       "      <td>шт</td>\n",
       "      <td>96923.00</td>\n",
       "      <td>1.10</td>\n",
       "      <td>756861.36</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>19 419</td>\n",
       "      <td>Бедро цыпленка-бройлера бескостное (филе бедра)</td>\n",
       "      <td>Мясо. Птица. П/ф охл. Яйца</td>\n",
       "      <td>Курица бескостное мясо</td>\n",
       "      <td>1504</td>\n",
       "      <td>кг</td>\n",
       "      <td>1438.22</td>\n",
       "      <td>0.96</td>\n",
       "      <td>723763.71</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>669</td>\n",
       "      <td>Арбуз</td>\n",
       "      <td>Овощи. Фрукты. Грибы. Зелень</td>\n",
       "      <td>Бахчевые</td>\n",
       "      <td>1621</td>\n",
       "      <td>кг</td>\n",
       "      <td>11265.38</td>\n",
       "      <td>6.95</td>\n",
       "      <td>683772.7</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>488</td>\n",
       "      <td>Филе грудки цыпленка-бройлера</td>\n",
       "      <td>Мясо. Птица. П/ф охл. Яйца</td>\n",
       "      <td>Курица бескостное мясо</td>\n",
       "      <td>1314</td>\n",
       "      <td>кг</td>\n",
       "      <td>1241.24</td>\n",
       "      <td>0.94</td>\n",
       "      <td>616319.03</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>22 658</td>\n",
       "      <td>Яйцо куриное С0</td>\n",
       "      <td>Мясо. Птица. П/ф охл. Яйца</td>\n",
       "      <td>Яйца</td>\n",
       "      <td>2790</td>\n",
       "      <td>упак</td>\n",
       "      <td>3147.00</td>\n",
       "      <td>1.13</td>\n",
       "      <td>485488</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>16 306</td>\n",
       "      <td>Масло сливочное 82,5%, 200 г</td>\n",
       "      <td>Молочная Продукция</td>\n",
       "      <td>Масло</td>\n",
       "      <td>1497</td>\n",
       "      <td>шт</td>\n",
       "      <td>1714.00</td>\n",
       "      <td>1.14</td>\n",
       "      <td>425423.71</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>36 296</td>\n",
       "      <td>Молоко 2,5% в бутылке, 900 мл</td>\n",
       "      <td>Молочная Продукция</td>\n",
       "      <td>Молоко пастеризованное</td>\n",
       "      <td>3967</td>\n",
       "      <td>шт</td>\n",
       "      <td>5000.00</td>\n",
       "      <td>1.26</td>\n",
       "      <td>407427.92</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>64 336</td>\n",
       "      <td>Картофель молодой</td>\n",
       "      <td>Овощи. Фрукты. Грибы. Зелень</td>\n",
       "      <td>Картофель подгруппа</td>\n",
       "      <td>3508</td>\n",
       "      <td>кг</td>\n",
       "      <td>5922.67</td>\n",
       "      <td>1.69</td>\n",
       "      <td>389319.41</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>22 564</td>\n",
       "      <td>Томаты Махитос</td>\n",
       "      <td>Овощи. Фрукты. Грибы. Зелень</td>\n",
       "      <td>Томаты</td>\n",
       "      <td>1603</td>\n",
       "      <td>кг</td>\n",
       "      <td>1043.58</td>\n",
       "      <td>0.65</td>\n",
       "      <td>375737.35</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>21 270</td>\n",
       "      <td>Авокадо Хасс, 2 шт</td>\n",
       "      <td>Овощи. Фрукты. Грибы. Зелень</td>\n",
       "      <td>Авокадо</td>\n",
       "      <td>1538</td>\n",
       "      <td>шт</td>\n",
       "      <td>1523.00</td>\n",
       "      <td>0.99</td>\n",
       "      <td>330377.52</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>12</th>\n",
       "      <td>668</td>\n",
       "      <td>Дыня Торпеда</td>\n",
       "      <td>Овощи. Фрукты. Грибы. Зелень</td>\n",
       "      <td>Бахчевые</td>\n",
       "      <td>697</td>\n",
       "      <td>кг</td>\n",
       "      <td>2778.11</td>\n",
       "      <td>3.99</td>\n",
       "      <td>310745.27</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>13</th>\n",
       "      <td>20 969</td>\n",
       "      <td>Котлета куриная с картофельным пюре</td>\n",
       "      <td>Кулинария</td>\n",
       "      <td>Вторые блюда птица</td>\n",
       "      <td>1130</td>\n",
       "      <td>шт</td>\n",
       "      <td>1306.00</td>\n",
       "      <td>1.16</td>\n",
       "      <td>297558.93</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>14</th>\n",
       "      <td>665</td>\n",
       "      <td>Дыня Колхозница</td>\n",
       "      <td>Овощи. Фрукты. Грибы. Зелень</td>\n",
       "      <td>Бахчевые</td>\n",
       "      <td>1054</td>\n",
       "      <td>кг</td>\n",
       "      <td>1907.50</td>\n",
       "      <td>1.81</td>\n",
       "      <td>283642.22</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>15</th>\n",
       "      <td>1 067</td>\n",
       "      <td>Филе бедра индейки</td>\n",
       "      <td>Мясо. Птица. П/ф охл. Яйца</td>\n",
       "      <td>Индейка</td>\n",
       "      <td>395</td>\n",
       "      <td>кг</td>\n",
       "      <td>400.45</td>\n",
       "      <td>1.01</td>\n",
       "      <td>281604.11</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>16</th>\n",
       "      <td>19 434</td>\n",
       "      <td>Суп \"Куриный\" с домашней лапшой</td>\n",
       "      <td>Кулинария</td>\n",
       "      <td>Супы</td>\n",
       "      <td>1438</td>\n",
       "      <td>шт</td>\n",
       "      <td>1692.00</td>\n",
       "      <td>1.18</td>\n",
       "      <td>280639.88</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>17</th>\n",
       "      <td>72 834</td>\n",
       "      <td>Торт \"Белые ночи\" с сырным кремом</td>\n",
       "      <td>Торты, пирожные, десерты</td>\n",
       "      <td>Торты</td>\n",
       "      <td>491</td>\n",
       "      <td>шт</td>\n",
       "      <td>500.00</td>\n",
       "      <td>1.02</td>\n",
       "      <td>278495.1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>18</th>\n",
       "      <td>28 740</td>\n",
       "      <td>Огурцы корнишоны, 300 г</td>\n",
       "      <td>Овощи. Фрукты. Грибы. Зелень</td>\n",
       "      <td>Огурцы</td>\n",
       "      <td>1099</td>\n",
       "      <td>шт</td>\n",
       "      <td>1176.00</td>\n",
       "      <td>1.07</td>\n",
       "      <td>277633.75</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>19</th>\n",
       "      <td>18 825</td>\n",
       "      <td>Торт \"Маракуйя\"</td>\n",
       "      <td>Торты, пирожные, десерты</td>\n",
       "      <td>Торты</td>\n",
       "      <td>491</td>\n",
       "      <td>шт</td>\n",
       "      <td>518.00</td>\n",
       "      <td>1.05</td>\n",
       "      <td>267023.34</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "    id_tov                                             name  \\\n",
       "0   23 074                                     Манго Египет   \n",
       "1      731                                           Бананы   \n",
       "2      647                     Пакет-майка \"ВкусВилл\" малый   \n",
       "3   19 419  Бедро цыпленка-бройлера бескостное (филе бедра)   \n",
       "4      669                                            Арбуз   \n",
       "5      488                    Филе грудки цыпленка-бройлера   \n",
       "6   22 658                                  Яйцо куриное С0   \n",
       "7   16 306                     Масло сливочное 82,5%, 200 г   \n",
       "8   36 296                    Молоко 2,5% в бутылке, 900 мл   \n",
       "9   64 336                                Картофель молодой   \n",
       "10  22 564                                   Томаты Махитос   \n",
       "11  21 270                               Авокадо Хасс, 2 шт   \n",
       "12     668                                     Дыня Торпеда   \n",
       "13  20 969              Котлета куриная с картофельным пюре   \n",
       "14     665                                  Дыня Колхозница   \n",
       "15   1 067                               Филе бедра индейки   \n",
       "16  19 434                  Суп \"Куриный\" с домашней лапшой   \n",
       "17  72 834                Торт \"Белые ночи\" с сырным кремом   \n",
       "18  28 740                          Огурцы корнишоны, 300 г   \n",
       "19  18 825                                  Торт \"Маракуйя\"   \n",
       "\n",
       "                        category                    group  total_carts system  \\\n",
       "0   Овощи. Фрукты. Грибы. Зелень         Манго (все виды)         8352     кг   \n",
       "1   Овощи. Фрукты. Грибы. Зелень                    Банан         6869     кг   \n",
       "2                        Нон-Фуд  Сумки, пакеты, упаковка        87843     шт   \n",
       "3     Мясо. Птица. П/ф охл. Яйца   Курица бескостное мясо         1504     кг   \n",
       "4   Овощи. Фрукты. Грибы. Зелень                 Бахчевые         1621     кг   \n",
       "5     Мясо. Птица. П/ф охл. Яйца   Курица бескостное мясо         1314     кг   \n",
       "6     Мясо. Птица. П/ф охл. Яйца                     Яйца         2790   упак   \n",
       "7             Молочная Продукция                    Масло         1497     шт   \n",
       "8             Молочная Продукция   Молоко пастеризованное         3967     шт   \n",
       "9   Овощи. Фрукты. Грибы. Зелень      Картофель подгруппа         3508     кг   \n",
       "10  Овощи. Фрукты. Грибы. Зелень                   Томаты         1603     кг   \n",
       "11  Овощи. Фрукты. Грибы. Зелень                  Авокадо         1538     шт   \n",
       "12  Овощи. Фрукты. Грибы. Зелень                 Бахчевые          697     кг   \n",
       "13                     Кулинария       Вторые блюда птица         1130     шт   \n",
       "14  Овощи. Фрукты. Грибы. Зелень                 Бахчевые         1054     кг   \n",
       "15    Мясо. Птица. П/ф охл. Яйца                  Индейка          395     кг   \n",
       "16                     Кулинария                     Супы         1438     шт   \n",
       "17      Торты, пирожные, десерты                    Торты          491     шт   \n",
       "18  Овощи. Фрукты. Грибы. Зелень                   Огурцы         1099     шт   \n",
       "19      Торты, пирожные, десерты                    Торты          491     шт   \n",
       "\n",
       "    per_month  avg_per_month sum_per_month  \n",
       "0     6972.48           0.83    2194135.41  \n",
       "1     6338.46           0.92      952972.7  \n",
       "2    96923.00           1.10     756861.36  \n",
       "3     1438.22           0.96     723763.71  \n",
       "4    11265.38           6.95      683772.7  \n",
       "5     1241.24           0.94     616319.03  \n",
       "6     3147.00           1.13        485488  \n",
       "7     1714.00           1.14     425423.71  \n",
       "8     5000.00           1.26     407427.92  \n",
       "9     5922.67           1.69     389319.41  \n",
       "10    1043.58           0.65     375737.35  \n",
       "11    1523.00           0.99     330377.52  \n",
       "12    2778.11           3.99     310745.27  \n",
       "13    1306.00           1.16     297558.93  \n",
       "14    1907.50           1.81     283642.22  \n",
       "15     400.45           1.01     281604.11  \n",
       "16    1692.00           1.18     280639.88  \n",
       "17     500.00           1.02      278495.1  \n",
       "18    1176.00           1.07     277633.75  \n",
       "19     518.00           1.05     267023.34  "
      ]
     },
     "execution_count": 74,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# Now, for every product, for every day of the week, get the fucking trigger and sku_id\n",
    "\n",
    "training_data = []\n",
    "  \n",
    "for _, row in products_df.iterrows():\n",
    "  sku_id = row['id_tov']\n",
    "  name = row['name']\n",
    "  avg_per_month = row['avg_per_month']\n",
    "  amount = randrange((int(row['avg_per_month']) + 1) * 10)\n",
    "  \n",
    "  # Add 13 rows for each product with dt from 0 to 12\n",
    "  for dt in range(14):\n",
    "    training_row = {\n",
    "        'sku_id': sku_id,\n",
    "        'name': name,\n",
    "        'category': row['category'],  # Arbitrary value from products_df\n",
    "        'amount': amount,\n",
    "        'avg_cart': avg_per_month,\n",
    "        't_expiery': 100,  # Arbitrary value\n",
    "        'stocktaking_time': 30,  # Arbitrary value\n",
    "        'dt': dt,\n",
    "        'trigger': determine_trigger(sku_id, name, amount, avg_per_month, dt)\n",
    "    }\n",
    "    training_data.append(training_row)\n",
    "\n",
    "training_df = pd.DataFrame(training_data)\n",
    "\n",
    "training_df.to_csv('training_data.csv', index=False)\n",
    "\n",
    "\n",
    "products_df.head(20)"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.11.5"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}