SplitData.ipynb

{
 "metadata": {
  "name": ""
 },
 "nbformat": 3,
 "nbformat_minor": 0,
 "worksheets": [
  {
   "cells": [
    {
     "cell_type": "code",
     "collapsed": false,
     "input": [
      "import pandas as pd\n",
      "\n",
      "# \u8bbe\u5b9a\u8bad\u7ec3\u96c6\u7684\u6bd4\u4f8b\n",
      "train_ratio = 0.8\n",
      "\n",
      "df = pd.read_csv(\"data/season1/t_alibaba_data.csv\", encoding=\"gbk\")\n",
      "# \u5c06\u65f6\u95f4\u6233\u8f6c\u6362\u6210Timestamp\u683c\u5f0f\uff0c\u5e76\u628a\u8bb0\u5f55\u6309\u65f6\u95f4\u6233\u8fdb\u884c\u6392\u5e8f\n",
      "datetime = df['visit_datetime']\n",
      "def to_timestamp(dt_unicode):\n",
      "    return pd.Timestamp(\"2013-%i-%i\" % (int(list(dt_unicode)[0]), int(list(dt_unicode)[2])))\n",
      "to_ts_func = np.frompyfunc(to_timestamp, 1, 1)\n",
      "timestamp = to_ts_func(datetime)\n",
      "del df['visit_datetime']\n",
      "df['visit_datetime'] = timestamp\n",
      "df.sort(columns='visit_datetime', ascending=True, inplace=True)\n",
      "df['index'] = timestamp\n",
      "df = df.set_index('index')\n",
      "\n",
      "# \u8ba1\u7b97\u5206\u5272\u70b9\n",
      "train_start = pd.DatetimeIndex([df['visit_datetime'][0]], freq='1D')[0]\n",
      "test_end = pd.DatetimeIndex([df['visit_datetime'][len(df.index) - 1]], freq='1D')[0]\n",
      "time_interval = int((test_end - train_start).days * train_ratio)\n",
      "train_end = train_start + time_interval\n",
      "test_start = train_start + (time_interval + 1)\n",
      "\n",
      "train_df = df.ix[train_start:train_end].reset_index()\n",
      "test_df = df.ix[test_start:test_end].reset_index()\n",
      "\n",
      "# \u6062\u590d\u65f6\u95f4\u6233\u683c\u5f0f\n",
      "train_ts = train_df['visit_datetime']\n",
      "test_ts = test_df['visit_datetime']\n",
      "def to_date(time_stamp):\n",
      "    date_time = pd.Timestamp(time_stamp).to_datetime()\n",
      "    month = date_time.strftime(\"%m\u6708\").strip('0')\n",
      "    day = date_time.strftime(\"%d\u65e5\").strip('0')\n",
      "    return month + day\n",
      "to_date_func = np.frompyfunc(to_date, 1, 1)\n",
      "train_dt = to_date_func(train_ts)\n",
      "test_dt = to_date_func(test_ts)\n",
      "del train_df['visit_datetime']\n",
      "del test_df['visit_datetime']\n",
      "train_df['visit_datetime'] = train_dt\n",
      "test_df['visit_datetime'] = test_dt\n",
      "\n",
      "del train_df['index']\n",
      "del test_df['index']\n",
      "\n",
      "# \u5199\u5165\u6587\u4ef6\n",
      "train_df.to_csv(\"data/season1/train.csv\", encoding=\"gbk\", index=False)\n",
      "test_df.to_csv(\"data/season1/test.csv\", encoding=\"gbk\", index=False)"
     ],
     "language": "python",
     "metadata": {},
     "outputs": [
      {
       "html": [
        "<div style=\"max-height:1000px;max-width:1500px;overflow:auto;\">\n",
        "<table border=\"1\" class=\"dataframe\">\n",
        "  <thead>\n",
        "    <tr style=\"text-align: right;\">\n",
        "      <th></th>\n",
        "      <th>user_id</th>\n",
        "      <th>brand_id</th>\n",
        "      <th>type</th>\n",
        "      <th>visit_datetime</th>\n",
        "    </tr>\n",
        "  </thead>\n",
        "  <tbody>\n",
        "    <tr>\n",
        "      <th>0 </th>\n",
        "      <td> 10944750</td>\n",
        "      <td> 25687</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>1 </th>\n",
        "      <td> 10944750</td>\n",
        "      <td> 13450</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>2 </th>\n",
        "      <td> 10944750</td>\n",
        "      <td> 13450</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>3 </th>\n",
        "      <td> 10944750</td>\n",
        "      <td> 13450</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>4 </th>\n",
        "      <td> 10944750</td>\n",
        "      <td> 11080</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>5 </th>\n",
        "      <td> 12028500</td>\n",
        "      <td> 21392</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>6 </th>\n",
        "      <td> 12028500</td>\n",
        "      <td> 21392</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>7 </th>\n",
        "      <td> 12028500</td>\n",
        "      <td> 21392</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>8 </th>\n",
        "      <td> 12028500</td>\n",
        "      <td> 28059</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>9 </th>\n",
        "      <td> 12028500</td>\n",
        "      <td> 28059</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>10</th>\n",
        "      <td> 12028500</td>\n",
        "      <td> 15368</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>11</th>\n",
        "      <td> 12028500</td>\n",
        "      <td> 15368</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>12</th>\n",
        "      <td> 12028500</td>\n",
        "      <td> 11679</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>13</th>\n",
        "      <td> 12028500</td>\n",
        "      <td> 11679</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>14</th>\n",
        "      <td> 12028500</td>\n",
        "      <td> 11679</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>15</th>\n",
        "      <td> 12028500</td>\n",
        "      <td> 27794</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>16</th>\n",
        "      <td> 12028500</td>\n",
        "      <td> 24849</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>17</th>\n",
        "      <td> 12028500</td>\n",
        "      <td>  8696</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>18</th>\n",
        "      <td> 12028500</td>\n",
        "      <td>  8696</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>19</th>\n",
        "      <td> 12028500</td>\n",
        "      <td> 24551</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>20</th>\n",
        "      <td> 12028500</td>\n",
        "      <td> 12368</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>21</th>\n",
        "      <td> 12028500</td>\n",
        "      <td> 11684</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>22</th>\n",
        "      <td> 12028500</td>\n",
        "      <td> 18082</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>23</th>\n",
        "      <td> 12028500</td>\n",
        "      <td> 24088</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>24</th>\n",
        "      <td> 12028500</td>\n",
        "      <td> 27985</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>25</th>\n",
        "      <td> 12028500</td>\n",
        "      <td>  9213</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>26</th>\n",
        "      <td> 12028500</td>\n",
        "      <td>  9213</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>27</th>\n",
        "      <td> 12028500</td>\n",
        "      <td>  9213</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>28</th>\n",
        "      <td> 12028500</td>\n",
        "      <td>  9213</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>29</th>\n",
        "      <td> 12028500</td>\n",
        "      <td> 14533</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>30</th>\n",
        "      <td>  1206500</td>\n",
        "      <td>  5962</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>31</th>\n",
        "      <td> 12154500</td>\n",
        "      <td> 26394</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>32</th>\n",
        "      <td> 12154500</td>\n",
        "      <td> 12475</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>33</th>\n",
        "      <td> 12154500</td>\n",
        "      <td> 12475</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>34</th>\n",
        "      <td> 12154500</td>\n",
        "      <td> 12475</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>35</th>\n",
        "      <td> 12154500</td>\n",
        "      <td> 12475</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>36</th>\n",
        "      <td> 12154500</td>\n",
        "      <td> 12475</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>37</th>\n",
        "      <td> 12154500</td>\n",
        "      <td> 15302</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>38</th>\n",
        "      <td> 12154500</td>\n",
        "      <td>  4949</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>39</th>\n",
        "      <td> 12154500</td>\n",
        "      <td>  4949</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>40</th>\n",
        "      <td> 12154500</td>\n",
        "      <td>  4949</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>41</th>\n",
        "      <td> 12154500</td>\n",
        "      <td>  4949</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>42</th>\n",
        "      <td> 12154500</td>\n",
        "      <td>  4949</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>43</th>\n",
        "      <td> 12154500</td>\n",
        "      <td>  4949</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>44</th>\n",
        "      <td> 12154500</td>\n",
        "      <td>  4949</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>45</th>\n",
        "      <td> 12154500</td>\n",
        "      <td>  4949</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>46</th>\n",
        "      <td> 12154500</td>\n",
        "      <td>  4949</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>47</th>\n",
        "      <td> 12154500</td>\n",
        "      <td>  4949</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>48</th>\n",
        "      <td> 12154500</td>\n",
        "      <td>  4949</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>49</th>\n",
        "      <td> 12154500</td>\n",
        "      <td>  4949</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>50</th>\n",
        "      <td> 12154500</td>\n",
        "      <td>  4949</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>51</th>\n",
        "      <td> 12154500</td>\n",
        "      <td> 19574</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>52</th>\n",
        "      <td> 12154500</td>\n",
        "      <td> 19574</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>53</th>\n",
        "      <td> 12154500</td>\n",
        "      <td> 19574</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>54</th>\n",
        "      <td> 12154500</td>\n",
        "      <td> 19574</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>55</th>\n",
        "      <td> 12154500</td>\n",
        "      <td> 19574</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>56</th>\n",
        "      <td> 12154500</td>\n",
        "      <td> 20476</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>57</th>\n",
        "      <td> 12154500</td>\n",
        "      <td> 20476</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>58</th>\n",
        "      <td> 12154500</td>\n",
        "      <td> 11176</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th>59</th>\n",
        "      <td> 12154500</td>\n",
        "      <td> 11176</td>\n",
        "      <td> 0</td>\n",
        "      <td> 8\u67081\u65e5</td>\n",
        "    </tr>\n",
        "    <tr>\n",
        "      <th></th>\n",
        "      <td>...</td>\n",
        "      <td>...</td>\n",
        "      <td>...</td>\n",
        "      <td>...</td>\n",
        "    </tr>\n",
        "  </tbody>\n",
        "</table>\n",
        "<p>25158 rows \u00d7 4 columns</p>\n",
        "</div>"
       ],
       "output_type": "pyout",
       "prompt_number": 6,
       "text": [
        "     user_id  brand_id  type visit_datetime\n",
        "0   10944750     25687     0           8\u67081\u65e5\n",
        "1   10944750     13450     0           8\u67081\u65e5\n",
        "2   10944750     13450     0           8\u67081\u65e5\n",
        "3   10944750     13450     0           8\u67081\u65e5\n",
        "4   10944750     11080     0           8\u67081\u65e5\n",
        "5   12028500     21392     0           8\u67081\u65e5\n",
        "6   12028500     21392     0           8\u67081\u65e5\n",
        "7   12028500     21392     0           8\u67081\u65e5\n",
        "8   12028500     28059     0           8\u67081\u65e5\n",
        "9   12028500     28059     0           8\u67081\u65e5\n",
        "10  12028500     15368     0           8\u67081\u65e5\n",
        "11  12028500     15368     0           8\u67081\u65e5\n",
        "12  12028500     11679     0           8\u67081\u65e5\n",
        "13  12028500     11679     0           8\u67081\u65e5\n",
        "14  12028500     11679     0           8\u67081\u65e5\n",
        "15  12028500     27794     0           8\u67081\u65e5\n",
        "16  12028500     24849     0           8\u67081\u65e5\n",
        "17  12028500      8696     0           8\u67081\u65e5\n",
        "18  12028500      8696     0           8\u67081\u65e5\n",
        "19  12028500     24551     0           8\u67081\u65e5\n",
        "20  12028500     12368     0           8\u67081\u65e5\n",
        "21  12028500     11684     0           8\u67081\u65e5\n",
        "22  12028500     18082     0           8\u67081\u65e5\n",
        "23  12028500     24088     0           8\u67081\u65e5\n",
        "24  12028500     27985     0           8\u67081\u65e5\n",
        "25  12028500      9213     0           8\u67081\u65e5\n",
        "26  12028500      9213     0           8\u67081\u65e5\n",
        "27  12028500      9213     0           8\u67081\u65e5\n",
        "28  12028500      9213     0           8\u67081\u65e5\n",
        "29  12028500     14533     0           8\u67081\u65e5\n",
        "30   1206500      5962     0           8\u67081\u65e5\n",
        "31  12154500     26394     0           8\u67081\u65e5\n",
        "32  12154500     12475     0           8\u67081\u65e5\n",
        "33  12154500     12475     0           8\u67081\u65e5\n",
        "34  12154500     12475     0           8\u67081\u65e5\n",
        "35  12154500     12475     0           8\u67081\u65e5\n",
        "36  12154500     12475     0           8\u67081\u65e5\n",
        "37  12154500     15302     0           8\u67081\u65e5\n",
        "38  12154500      4949     0           8\u67081\u65e5\n",
        "39  12154500      4949     0           8\u67081\u65e5\n",
        "40  12154500      4949     0           8\u67081\u65e5\n",
        "41  12154500      4949     0           8\u67081\u65e5\n",
        "42  12154500      4949     0           8\u67081\u65e5\n",
        "43  12154500      4949     0           8\u67081\u65e5\n",
        "44  12154500      4949     0           8\u67081\u65e5\n",
        "45  12154500      4949     0           8\u67081\u65e5\n",
        "46  12154500      4949     0           8\u67081\u65e5\n",
        "47  12154500      4949     0           8\u67081\u65e5\n",
        "48  12154500      4949     0           8\u67081\u65e5\n",
        "49  12154500      4949     0           8\u67081\u65e5\n",
        "50  12154500      4949     0           8\u67081\u65e5\n",
        "51  12154500     19574     0           8\u67081\u65e5\n",
        "52  12154500     19574     0           8\u67081\u65e5\n",
        "53  12154500     19574     0           8\u67081\u65e5\n",
        "54  12154500     19574     0           8\u67081\u65e5\n",
        "55  12154500     19574     0           8\u67081\u65e5\n",
        "56  12154500     20476     0           8\u67081\u65e5\n",
        "57  12154500     20476     0           8\u67081\u65e5\n",
        "58  12154500     11176     0           8\u67081\u65e5\n",
        "59  12154500     11176     0           8\u67081\u65e5\n",
        "         ...       ...   ...            ...\n",
        "\n",
        "[25158 rows x 4 columns]"
       ]
      }
     ],
     "prompt_number": 6
    },
    {
     "cell_type": "code",
     "collapsed": false,
     "input": [
      "import pandas as pd\n",
      "\n",
      "train_ratio = 0.8\n",
      "file_name = \"train_linreg\"\n",
      "\n",
      "df = pd.read_csv(\"data/season1/%s.csv\" % file_name, parse_dates=['visit_datetime'], encoding=\"gbk\")\n",
      "columns = df.columns\n",
      "df = df.set_index('visit_datetime')\n",
      "\n",
      "partition_point = len(df.index) * train_ratio\n",
      "\n",
      "train_df = df.iloc[:partition_point].reset_index()\n",
      "test_df = df.iloc[partition_point:].reset_index()\n",
      "\n",
      "train_df = train_df.reindex_axis(columns, axis=1)\n",
      "test_df = test_df.reindex_axis(columns, axis=1)\n",
      "\n",
      "train_df.to_csv(\"data/season1/%s-0.csv\" % file_name, encoding=\"gbk\", index=False)\n",
      "test_df.to_csv(\"data/season1/%s-1.csv\" % file_name, encoding=\"gbk\", index=False)"
     ],
     "language": "python",
     "metadata": {},
     "outputs": [],
     "prompt_number": 35
    }
   ],
   "metadata": {}
  }
 ]
}