find_growing_stock.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 138,
   "metadata": {},
   "outputs": [],
   "source": [
    "import tushare as ts\n",
    "import pandas as pd\n",
    "#http://tushare.org/reference.html#id3\n",
    "\n",
    "\"\"\"\n",
    "https://xueqiu.com/2785562330/55607659\n",
    "1、好行业\n",
    "    环保、医疗保健、互联网+（信息安全）、新材料\n",
    "2、好企业\n",
    "3、好价格\n",
    "    一个重要的买入观察及时指标就是(周）均线，尽可能在均线收拢是买入\n",
    "\"\"\"\n",
    "def prepare(year, quarter):\n",
    "    # get data\n",
    "    df_b0 = ts.get_stock_basics()\n",
    "    df_prof0 = ts.get_profit_data(year,quarter)\n",
    "    df_grow0 = ts.get_growth_data(year,quarter)\n",
    "    df_deb0 = ts.get_debtpaying_data(year, quarter)\n",
    "    # merge feature\n",
    "    df_b0['mktcap'] = df_b0.apply(lambda row: row.pe * row.esp * row.totals, \n",
    "                                  axis=1)\n",
    "    df_b = df_b0[['name', 'industry', 'pe', 'esp', 'mktcap', 'gpr','totals']]\n",
    "    df_grow = df_grow0.set_index('code')[['nav', 'mbrg', 'nprg', 'epsg']]\n",
    "    df_prof = df_prof0.set_index('code')[['roe', 'business_income', 'bips']]\n",
    "    df_deb = df_deb0.set_index('code')[['sheqratio']]\n",
    "    df = df_b.join(df_prof).join(df_grow).join(df_deb)\n",
    "    df['peg'] = df.apply(lambda row: row.pe / max(1e-6, row.epsg), axis=1)\n",
    "    df['minr'] = df.apply(lambda row: row.totals > 0 and row.bips/(\n",
    "        row.business_income/100/row.totals) or 0, axis=1)\n",
    "    df.drop(['bips','business_income'], axis = 1, inplace=True)\n",
    "    return df\n",
    "\n",
    "def feature(df, names):\n",
    "    if isinstance(df['name'][0], unicode):\n",
    "        names = [s.decode('utf8') for s in names]\n",
    "    return df[df['name'].isin(names)]\n",
    "\n",
    "def filter_kv(df, dic, more = True):\n",
    "    for k,v in dic.iteritems():\n",
    "        if more:\n",
    "            df = df[df[k] >= v]\n",
    "        else:\n",
    "            df = df[df[k] <= v]\n",
    "    return df\n",
    "\n",
    "def good_stock(df):\n",
    "    \"\"\"\n",
    "（1）市盈率增长率比 peg ＜1 \n",
    "         peg= 市盈率（PE）/每股收益增长率(epsg)\n",
    "（2）营业收入增长率 mbrg＞30% \n",
    "（3）净利润增长率 nprg ＞30%\n",
    "（4）净资产收益率 roe ＞15%\n",
    "（5）净资产增长率 nav > 8%\n",
    "（6）毛利率 gpr＞50%\n",
    " (7）资产负债率 1 - sheqratio(股东权益率) ＜ 60%  \n",
    "（8）市值尽可能小 mktcap\n",
    "(9) 主营收入占比 > 80% minr\n",
    "    \"\"\"\n",
    "    df = filter_kv(df, {'roe':15, 'gpr': 50, 'nprg':30, \n",
    "                        'mbrg': 30, 'nav': 8, 'sheqratio': 40, 'minr': 0.8}, True)\n",
    "    df = filter_kv(df, {'mktcap':1e3, 'peg': 1}, False)\n",
    "    return df"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 143,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[Getting data:]##########[Getting data:]#######[Getting data:]########"
     ]
    }
   ],
   "source": [
    "year, quarter = 2017,4\n",
    "df = prepare(year, quarter)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 144,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>name</th>\n",
       "      <th>industry</th>\n",
       "      <th>pe</th>\n",
       "      <th>esp</th>\n",
       "      <th>mktcap</th>\n",
       "      <th>gpr</th>\n",
       "      <th>totals</th>\n",
       "      <th>roe</th>\n",
       "      <th>nav</th>\n",
       "      <th>mbrg</th>\n",
       "      <th>nprg</th>\n",
       "      <th>epsg</th>\n",
       "      <th>sheqratio</th>\n",
       "      <th>peg</th>\n",
       "      <th>minr</th>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>code</th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>002677</th>\n",
       "      <td>浙江美大</td>\n",
       "      <td>家用电器</td>\n",
       "      <td>36.60</td>\n",
       "      <td>0.480</td>\n",
       "      <td>113.489280</td>\n",
       "      <td>53.94</td>\n",
       "      <td>6.46</td>\n",
       "      <td>23.32</td>\n",
       "      <td>13.6709</td>\n",
       "      <td>54.1812</td>\n",
       "      <td>50.5960</td>\n",
       "      <td>50.0000</td>\n",
       "      <td>78.0823</td>\n",
       "      <td>0.732000</td>\n",
       "      <td>0.999880</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>300146</th>\n",
       "      <td>汤臣倍健</td>\n",
       "      <td>食品</td>\n",
       "      <td>32.46</td>\n",
       "      <td>0.522</td>\n",
       "      <td>248.909123</td>\n",
       "      <td>67.08</td>\n",
       "      <td>14.69</td>\n",
       "      <td>15.02</td>\n",
       "      <td>9.1048</td>\n",
       "      <td>34.7182</td>\n",
       "      <td>51.0487</td>\n",
       "      <td>40.5405</td>\n",
       "      <td>83.9432</td>\n",
       "      <td>0.800681</td>\n",
       "      <td>0.999231</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>300740</th>\n",
       "      <td>御家汇</td>\n",
       "      <td>日用化工</td>\n",
       "      <td>53.29</td>\n",
       "      <td>0.991</td>\n",
       "      <td>84.496624</td>\n",
       "      <td>52.63</td>\n",
       "      <td>1.60</td>\n",
       "      <td>29.35</td>\n",
       "      <td>44.4849</td>\n",
       "      <td>40.6107</td>\n",
       "      <td>117.8902</td>\n",
       "      <td>112.9032</td>\n",
       "      <td>63.0505</td>\n",
       "      <td>0.471997</td>\n",
       "      <td>1.333324</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>600338</th>\n",
       "      <td>西藏珠峰</td>\n",
       "      <td>铅锌</td>\n",
       "      <td>20.63</td>\n",
       "      <td>1.705</td>\n",
       "      <td>229.687199</td>\n",
       "      <td>72.55</td>\n",
       "      <td>6.53</td>\n",
       "      <td>65.35</td>\n",
       "      <td>31.2005</td>\n",
       "      <td>67.0234</td>\n",
       "      <td>71.9849</td>\n",
       "      <td>71.2593</td>\n",
       "      <td>52.6689</td>\n",
       "      <td>0.289506</td>\n",
       "      <td>0.999967</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>600516</th>\n",
       "      <td>方大炭素</td>\n",
       "      <td>矿物制品</td>\n",
       "      <td>13.14</td>\n",
       "      <td>2.110</td>\n",
       "      <td>496.007406</td>\n",
       "      <td>76.59</td>\n",
       "      <td>17.89</td>\n",
       "      <td>37.31</td>\n",
       "      <td>68.4229</td>\n",
       "      <td>248.6204</td>\n",
       "      <td>13132.3664</td>\n",
       "      <td>5175.0000</td>\n",
       "      <td>73.0582</td>\n",
       "      <td>0.002539</td>\n",
       "      <td>1.000112</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "        name industry     pe    esp      mktcap    gpr  totals    roe  \\\n",
       "code                                                                    \n",
       "002677  浙江美大     家用电器  36.60  0.480  113.489280  53.94    6.46  23.32   \n",
       "300146  汤臣倍健       食品  32.46  0.522  248.909123  67.08   14.69  15.02   \n",
       "300740   御家汇     日用化工  53.29  0.991   84.496624  52.63    1.60  29.35   \n",
       "600338  西藏珠峰       铅锌  20.63  1.705  229.687199  72.55    6.53  65.35   \n",
       "600516  方大炭素     矿物制品  13.14  2.110  496.007406  76.59   17.89  37.31   \n",
       "\n",
       "            nav      mbrg        nprg       epsg sheqratio       peg      minr  \n",
       "code                                                                            \n",
       "002677  13.6709   54.1812     50.5960    50.0000   78.0823  0.732000  0.999880  \n",
       "300146   9.1048   34.7182     51.0487    40.5405   83.9432  0.800681  0.999231  \n",
       "300740  44.4849   40.6107    117.8902   112.9032   63.0505  0.471997  1.333324  \n",
       "600338  31.2005   67.0234     71.9849    71.2593   52.6689  0.289506  0.999967  \n",
       "600516  68.4229  248.6204  13132.3664  5175.0000   73.0582  0.002539  1.000112  "
      ]
     },
     "execution_count": 144,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "good_stock(df)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 142,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>name</th>\n",
       "      <th>industry</th>\n",
       "      <th>pe</th>\n",
       "      <th>esp</th>\n",
       "      <th>mktcap</th>\n",
       "      <th>gpr</th>\n",
       "      <th>totals</th>\n",
       "      <th>roe</th>\n",
       "      <th>nav</th>\n",
       "      <th>mbrg</th>\n",
       "      <th>nprg</th>\n",
       "      <th>epsg</th>\n",
       "      <th>sheqratio</th>\n",
       "      <th>peg</th>\n",
       "      <th>minr</th>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>code</th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>000961</th>\n",
       "      <td>中南建设</td>\n",
       "      <td>建筑施工</td>\n",
       "      <td>48.78</td>\n",
       "      <td>0.109</td>\n",
       "      <td>197.261442</td>\n",
       "      <td>17.87</td>\n",
       "      <td>37.10</td>\n",
       "      <td>2.91</td>\n",
       "      <td>-5.0284</td>\n",
       "      <td>-0.9725</td>\n",
       "      <td>9.1220</td>\n",
       "      <td>-23.0932</td>\n",
       "      <td>10.9991</td>\n",
       "      <td>4.878000e+07</td>\n",
       "      <td>1.000052</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>600436</th>\n",
       "      <td>片仔癀</td>\n",
       "      <td>中成药</td>\n",
       "      <td>56.46</td>\n",
       "      <td>1.118</td>\n",
       "      <td>380.627348</td>\n",
       "      <td>43.21</td>\n",
       "      <td>6.03</td>\n",
       "      <td>16.71</td>\n",
       "      <td>17.3580</td>\n",
       "      <td>68.8339</td>\n",
       "      <td>35.0511</td>\n",
       "      <td>40.0000</td>\n",
       "      <td>77.9313</td>\n",
       "      <td>1.411500e+00</td>\n",
       "      <td>0.999465</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>600519</th>\n",
       "      <td>贵州茅台</td>\n",
       "      <td>白酒</td>\n",
       "      <td>34.98</td>\n",
       "      <td>15.908</td>\n",
       "      <td>6989.160710</td>\n",
       "      <td>89.93</td>\n",
       "      <td>12.56</td>\n",
       "      <td>23.68</td>\n",
       "      <td>23.9850</td>\n",
       "      <td>59.3972</td>\n",
       "      <td>60.2488</td>\n",
       "      <td>60.3831</td>\n",
       "      <td>69.1664</td>\n",
       "      <td>5.793012e-01</td>\n",
       "      <td>0.999842</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>600779</th>\n",
       "      <td>水井坊</td>\n",
       "      <td>白酒</td>\n",
       "      <td>69.26</td>\n",
       "      <td>0.498</td>\n",
       "      <td>168.663337</td>\n",
       "      <td>78.88</td>\n",
       "      <td>4.89</td>\n",
       "      <td>16.35</td>\n",
       "      <td>6.7673</td>\n",
       "      <td>85.1825</td>\n",
       "      <td>63.2229</td>\n",
       "      <td>63.2045</td>\n",
       "      <td>60.1699</td>\n",
       "      <td>1.095808e+00</td>\n",
       "      <td>1.000910</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>600809</th>\n",
       "      <td>山西汾酒</td>\n",
       "      <td>白酒</td>\n",
       "      <td>46.39</td>\n",
       "      <td>0.930</td>\n",
       "      <td>373.615782</td>\n",
       "      <td>69.39</td>\n",
       "      <td>8.66</td>\n",
       "      <td>15.80</td>\n",
       "      <td>11.3296</td>\n",
       "      <td>42.7965</td>\n",
       "      <td>81.7039</td>\n",
       "      <td>78.5262</td>\n",
       "      <td>60.6077</td>\n",
       "      <td>5.907582e-01</td>\n",
       "      <td>1.000159</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>603799</th>\n",
       "      <td>华友钴业</td>\n",
       "      <td>小金属</td>\n",
       "      <td>49.69</td>\n",
       "      <td>1.852</td>\n",
       "      <td>545.713468</td>\n",
       "      <td>33.25</td>\n",
       "      <td>5.93</td>\n",
       "      <td>20.76</td>\n",
       "      <td>117.5545</td>\n",
       "      <td>76.1888</td>\n",
       "      <td>41327.9603</td>\n",
       "      <td>9150.0000</td>\n",
       "      <td>38.9175</td>\n",
       "      <td>5.430601e-03</td>\n",
       "      <td>1.000542</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>603986</th>\n",
       "      <td>兆易创新</td>\n",
       "      <td>半导体</td>\n",
       "      <td>86.68</td>\n",
       "      <td>1.700</td>\n",
       "      <td>299.132680</td>\n",
       "      <td>38.57</td>\n",
       "      <td>2.03</td>\n",
       "      <td>21.16</td>\n",
       "      <td>29.3085</td>\n",
       "      <td>44.6865</td>\n",
       "      <td>132.6599</td>\n",
       "      <td>-8.7473</td>\n",
       "      <td>75.1758</td>\n",
       "      <td>8.668000e+07</td>\n",
       "      <td>1.001505</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "        name industry     pe     esp       mktcap    gpr  totals    roe  \\\n",
       "code                                                                      \n",
       "000961  中南建设     建筑施工  48.78   0.109   197.261442  17.87   37.10   2.91   \n",
       "600436   片仔癀      中成药  56.46   1.118   380.627348  43.21    6.03  16.71   \n",
       "600519  贵州茅台       白酒  34.98  15.908  6989.160710  89.93   12.56  23.68   \n",
       "600779   水井坊       白酒  69.26   0.498   168.663337  78.88    4.89  16.35   \n",
       "600809  山西汾酒       白酒  46.39   0.930   373.615782  69.39    8.66  15.80   \n",
       "603799  华友钴业      小金属  49.69   1.852   545.713468  33.25    5.93  20.76   \n",
       "603986  兆易创新      半导体  86.68   1.700   299.132680  38.57    2.03  21.16   \n",
       "\n",
       "             nav     mbrg        nprg       epsg  sheqratio           peg  \\\n",
       "code                                                                        \n",
       "000961   -5.0284  -0.9725      9.1220   -23.0932    10.9991  4.878000e+07   \n",
       "600436   17.3580  68.8339     35.0511    40.0000    77.9313  1.411500e+00   \n",
       "600519   23.9850  59.3972     60.2488    60.3831    69.1664  5.793012e-01   \n",
       "600779    6.7673  85.1825     63.2229    63.2045    60.1699  1.095808e+00   \n",
       "600809   11.3296  42.7965     81.7039    78.5262    60.6077  5.907582e-01   \n",
       "603799  117.5545  76.1888  41327.9603  9150.0000    38.9175  5.430601e-03   \n",
       "603986   29.3085  44.6865    132.6599    -8.7473    75.1758  8.668000e+07   \n",
       "\n",
       "            minr  \n",
       "code              \n",
       "000961  1.000052  \n",
       "600436  0.999465  \n",
       "600519  0.999842  \n",
       "600779  1.000910  \n",
       "600809  1.000159  \n",
       "603799  1.000542  \n",
       "603986  1.001505  "
      ]
     },
     "execution_count": 142,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "my_stocks = ('华友钴业', '兆易创新', '中南建设', '水井坊', '贵州茅台', '山西汾酒','片仔癀')\n",
    "feature(df, my_stocks)\n",
    "# epsg < 0 季度不靠谱\n",
    "# 酒类靠谱\n",
    "# g类，周期性，得环比去年"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 43,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[Getting data:]####"
     ]
    },
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>name</th>\n",
       "      <th>type</th>\n",
       "      <th>report_date</th>\n",
       "      <th>pre_eps</th>\n",
       "      <th>range</th>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>code</th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>300509</th>\n",
       "      <td>新美星</td>\n",
       "      <td>预升</td>\n",
       "      <td>2018-03-21</td>\n",
       "      <td>0.1500</td>\n",
       "      <td>-15.78%~1.07%</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2709</th>\n",
       "      <td>天赐材料</td>\n",
       "      <td>预增</td>\n",
       "      <td>2018-03-21</td>\n",
       "      <td>0.2000</td>\n",
       "      <td>530%~560%</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>300457</th>\n",
       "      <td>赢合科技</td>\n",
       "      <td>预增</td>\n",
       "      <td>2018-03-21</td>\n",
       "      <td>0.3100</td>\n",
       "      <td>50%~65%</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2449</th>\n",
       "      <td>国星光电</td>\n",
       "      <td>预增</td>\n",
       "      <td>2018-03-21</td>\n",
       "      <td>0.1088</td>\n",
       "      <td>30%~60%</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>300052</th>\n",
       "      <td>中青宝</td>\n",
       "      <td>预增</td>\n",
       "      <td>2018-03-21</td>\n",
       "      <td>0.0100</td>\n",
       "      <td>140%~170%</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2404</th>\n",
       "      <td>嘉欣丝绸</td>\n",
       "      <td>预增</td>\n",
       "      <td>2018-03-21</td>\n",
       "      <td>0.0400</td>\n",
       "      <td>50%~80%</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>300436</th>\n",
       "      <td>广生堂</td>\n",
       "      <td>预降</td>\n",
       "      <td>2018-03-21</td>\n",
       "      <td>0.1438</td>\n",
       "      <td>-50%~-20%</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>300571</th>\n",
       "      <td>平治信息</td>\n",
       "      <td>预增</td>\n",
       "      <td>2018-03-20</td>\n",
       "      <td>0.5500</td>\n",
       "      <td>91.97%~119.4%</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>300446</th>\n",
       "      <td>乐凯新材</td>\n",
       "      <td>预升</td>\n",
       "      <td>2018-03-20</td>\n",
       "      <td>0.2484</td>\n",
       "      <td>-10%~10%</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>300563</th>\n",
       "      <td>神宇股份</td>\n",
       "      <td>预升</td>\n",
       "      <td>2018-03-20</td>\n",
       "      <td>0.1746</td>\n",
       "      <td>0%~10%</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "        name type report_date  pre_eps          range\n",
       "code                                                 \n",
       "300509   新美星   预升  2018-03-21   0.1500  -15.78%~1.07%\n",
       "2709    天赐材料   预增  2018-03-21   0.2000      530%~560%\n",
       "300457  赢合科技   预增  2018-03-21   0.3100        50%~65%\n",
       "2449    国星光电   预增  2018-03-21   0.1088        30%~60%\n",
       "300052   中青宝   预增  2018-03-21   0.0100      140%~170%\n",
       "2404    嘉欣丝绸   预增  2018-03-21   0.0400        50%~80%\n",
       "300436   广生堂   预降  2018-03-21   0.1438      -50%~-20%\n",
       "300571  平治信息   预增  2018-03-20   0.5500  91.97%~119.4%\n",
       "300446  乐凯新材   预升  2018-03-20   0.2484       -10%~10%\n",
       "300563  神宇股份   预升  2018-03-20   0.1746         0%~10%"
      ]
     },
     "execution_count": 43,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df_fu = ts.forecast_data(2018,1)\n",
    "feature(df_fu.set_index('code'),  my_stocks)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 105,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[Getting data:]##########################################################"
     ]
    },
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>code</th>\n",
       "      <th>name</th>\n",
       "      <th>arturnover</th>\n",
       "      <th>arturndays</th>\n",
       "      <th>inventory_turnover</th>\n",
       "      <th>inventory_days</th>\n",
       "      <th>currentasset_turnover</th>\n",
       "      <th>currentasset_days</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>2080</th>\n",
       "      <td>603986</td>\n",
       "      <td>兆易创新</td>\n",
       "      <td>13.9736</td>\n",
       "      <td>19.3222</td>\n",
       "      <td>2.0504</td>\n",
       "      <td>131.6816</td>\n",
       "      <td>0.9706</td>\n",
       "      <td>278.1784</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2503</th>\n",
       "      <td>603799</td>\n",
       "      <td>华友钴业</td>\n",
       "      <td>8.0790</td>\n",
       "      <td>33.4200</td>\n",
       "      <td>1.3927</td>\n",
       "      <td>193.8680</td>\n",
       "      <td>0.8759</td>\n",
       "      <td>308.2544</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2547</th>\n",
       "      <td>600436</td>\n",
       "      <td>片仔癀</td>\n",
       "      <td>6.7875</td>\n",
       "      <td>39.7790</td>\n",
       "      <td>1.3259</td>\n",
       "      <td>203.6353</td>\n",
       "      <td>0.7344</td>\n",
       "      <td>367.6471</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2948</th>\n",
       "      <td>600809</td>\n",
       "      <td>山西汾酒</td>\n",
       "      <td>78.6900</td>\n",
       "      <td>3.4312</td>\n",
       "      <td>0.7623</td>\n",
       "      <td>354.1913</td>\n",
       "      <td>0.9086</td>\n",
       "      <td>297.1605</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3178</th>\n",
       "      <td>600779</td>\n",
       "      <td>水井坊</td>\n",
       "      <td>56.0520</td>\n",
       "      <td>4.8170</td>\n",
       "      <td>0.3731</td>\n",
       "      <td>723.6666</td>\n",
       "      <td>0.8634</td>\n",
       "      <td>312.7172</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3246</th>\n",
       "      <td>600519</td>\n",
       "      <td>贵州茅台</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.2060</td>\n",
       "      <td>1310.6796</td>\n",
       "      <td>0.4354</td>\n",
       "      <td>620.1194</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3253</th>\n",
       "      <td>000961</td>\n",
       "      <td>中南建设</td>\n",
       "      <td>2.8553</td>\n",
       "      <td>94.5610</td>\n",
       "      <td>0.1832</td>\n",
       "      <td>1473.7991</td>\n",
       "      <td>0.1504</td>\n",
       "      <td>1795.2128</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "        code  name  arturnover  arturndays  inventory_turnover  \\\n",
       "2080  603986  兆易创新     13.9736     19.3222              2.0504   \n",
       "2503  603799  华友钴业      8.0790     33.4200              1.3927   \n",
       "2547  600436   片仔癀      6.7875     39.7790              1.3259   \n",
       "2948  600809  山西汾酒     78.6900      3.4312              0.7623   \n",
       "3178  600779   水井坊     56.0520      4.8170              0.3731   \n",
       "3246  600519  贵州茅台         NaN         NaN              0.2060   \n",
       "3253  000961  中南建设      2.8553     94.5610              0.1832   \n",
       "\n",
       "      inventory_days  currentasset_turnover  currentasset_days  \n",
       "2080        131.6816                 0.9706           278.1784  \n",
       "2503        193.8680                 0.8759           308.2544  \n",
       "2547        203.6353                 0.7344           367.6471  \n",
       "2948        354.1913                 0.9086           297.1605  \n",
       "3178        723.6666                 0.8634           312.7172  \n",
       "3246       1310.6796                 0.4354           620.1194  \n",
       "3253       1473.7991                 0.1504          1795.2128  "
      ]
     },
     "execution_count": 105,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# 运营指标\n",
    "df_y = ts.get_operation_data(2017,3)\n",
    "feature(df_y,[s.decode('utf-8') for s in my_stocks])"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 2",
   "language": "python",
   "name": "python2"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 2
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython2",
   "version": "2.7.14"
  },
  "toc": {
   "nav_menu": {},
   "number_sections": true,
   "sideBar": true,
   "skip_h1_title": false,
   "toc_cell": false,
   "toc_position": {},
   "toc_section_display": "block",
   "toc_window_display": false
  }
 },
 "nbformat": 4,
 "nbformat_minor": 1
}