Python绘制股票日K图（八）绘制大资金折线图

大资金代表了大势力（比如机构）对股票的态度，也是一个比较重要的指标。

绘制折线图的代码如下：

ax.plot(
        np.arange(0, len(stock_data.index)-begin),
        stock_data.mainBuy[begin:],  # 大资金流入的数据
        c='red',  # 折线颜色
        marker='o',  # 折线上点的形状
        markersize=2.6)  # 点的大小
    ax.plot(
        np.arange(0, len(stock_data.index)-begin),
        stock_data.mainSell[begin:],  # 大资金流出的数据
        c='green',  # 折线颜色
        marker='o',  # 折线上点的形状
        markersize=2.6)  # 点的大小

完整的代码如下：

# 绘制大资金折线图
import pandas
import numpy as np
import matplotlib.pyplot as plt


dir_name = 'D:\\gupiao\\'


def paint_dayk(code):
    # 1、获取数据
    stock_data = pandas.read_csv(dir_name + code + '.txt')  # 读取数据
    begin = len(stock_data) - 120  # 取最近120天的数据
    if begin < 0:  # 如果上市未满120天，则从上市当天开始显示