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天，则从上市当天开始显示
        begin = 0
    # 2、画日K图
    fig = plt.figure(  # 添加fig对象
        figsize=(54, 28),  # 设置fig大小，长和宽，单位为英寸
        dpi=120)  # 每英寸的像素点数
    gs = fig.add_gridspec(1, 1)  # 在fig中添加一个一行一列的网格
    ax = fig.add_subplot(gs[0, 0])  # 在fig中添加网格的第一块，并返回一套坐标轴
    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)  # 点的大小

    ax.grid(axis='x', linestyle='-.')  # 添加x轴方向的网格
    ax.grid(axis='y', linestyle='-.')  # 添加y轴方向的网格
    ax.yaxis.tick_right()  # 将y轴的数值显示在右边
    ax.set_xlim(0, len(stock_data.index) - begin, 1)  # 设置x轴的范围
    ax.set_xticks(range(0, len(stock_data.index) - begin, 1))  # 设置x轴的刻度
    ax.axes.set_xticklabels(stock_data.week[begin:])

    for label in ax.xaxis.get_ticklabels():  # 设置x轴标签的字体大小
        label.set_fontsize(16)
    for label in ax.yaxis.get_ticklabels():  # 设置y轴标签的字体大小
        label.set_fontsize(40)

    # 3、输出日K图
    plt.savefig(dir_name + code + '.jpg')  # 保存图片


def main():
    code = '000001'
    paint_dayk(code)


if __name__ == '__main__':
    main()

生成的图形如下：

更多数据下载：股票数据，最近两百天，更新到了20220809-数据集文档类资源-CSDN下载