import tushare as ts
import time
import pandas as pd
import os
import matplotlib as mpl
import matplotlib.pyplot as plt
import matplotlib.finance as mpf
import datetime
from matplotlib.pylab import date2num
import sys

def getstockbasicinfo():
    filebasic = './output/stockbasic.csv'
    filetoday = './output/stocktoday.csv'
    if os.path.exists(filebasic):
        print("Get basic data from csv")
        dfstockbasic = pd.read_csv(filebasic, encoding="utf-8")
    else:
        print("Get basic data from internet")
        try:
            dfstockbasic = ts.get_stock_basics()
            dfstockbasic.to_csv(filebasic, encoding='utf-8')
        except Exception as e:
            print(e)
    if os.path.exists(filetoday):
        print("Get today data from csv")
        dfstocktoday = pd.read_csv(filetoday, encoding="utf-8")
    else:
        print("Get today data from internet")
        try:
            dfstocktoday = ts.get_today_all()
            dfstocktoday.to_csv(filetoday, encoding='utf-8')
        except Exception as e:
            print(e)

    dfstockbasic['tomarket'] = pd.to_numeric(dfstockbasic.timeToMarket)
    dflateststock = dfstockbasic[(dfstockbasic.tomarket >= 20151001)]
    dfmerge = pd.merge(dflateststock, dfstocktoday, on=['code', 'name'], how='left')
    dfmerge['totalnessets'] = (dfmerge.totals * dfmerge.trade)
    dfbigstock = dfmerge[dfmerge.totalnessets > 400]
    print(dfbigstock)
    codes = {}
    for index, row in dfbigstock.iterrows():
        codes['{:0>6}'.format(row['code'])] = row['totalnessets']
    getcandlecharts(codes)

def getcandlecharts(codes):
    chinese = mpl.font_manager.FontProperties(fname='C:\Windows\Fonts\simhei.ttf')
    plt.figure(figsize=(10, 8), facecolor='w')
    index = 1
    for code in codes:
        print(code)
        shyh = ts.get_k_data(code)
        mat_shyh = shyh.as_matrix()
        num_time = date_to_num(mat_shyh[:,0])
        mat_shyh[:,0]=num_time
        ax = plt.subplot(4, 4, index)
        mpf.candlestick_ochl(ax, mat_shyh,width=0.6, colorup="r", colordown="g", alpha=1.0)
        plt.grid(True)
        plt.xticks(rotation=30)
        plt.title('%s: %.2f' % (getstockinfo(code), codes[code]), fontproperties=chinese)
        plt.xlabel("Date")
        plt.ylabel("Price")
        ax.xaxis_date()
        index = index + 1
        if index == 16:
            break
    plt.suptitle('2015-10-01之后上市總市值超過400億人民幣的股票', fontproperties=chinese)
    plt.tight_layout(1.5)
    plt.subplots_adjust(top=0.92)
    plt.show()

if __name__ == "__main__":
        getstockbasicinfo()

圖片示例：

2018-05-23 18_20_35-Figure 1.png