如何用matlab画股市K线，修改内置candle函数

MATLAB里面自带了画K线的函数candle

自带函数效果预览图
原始candle的使用方法
- 语法
- 描述
- 输入参数
修改源代码
- 原始代码
- 修改部分
最终显示预览
修改后的完整代码

自带函数效果预览图

可以看出K线图有基本的框架，但是和我们平常的习惯不同，还需要修改其颜色，使得更直观些

原始candle的使用方法

语法

candle(Data)
candle(Data,Color)
h = candle(ax___)

描述

例子
candle(Data)从证券的一系列开盘价、最高价、最低价和收盘价中绘制蜡烛形图。如果收盘价大于开盘价,则主体(开盘价和收盘价之间的区域)未填充;否则主体被填充。
例子
candle(Data,Color)添加Color的可选参数.
例子
h = candle(ax___)为ax添加可选参数. 默认为gca，可选任意坐标轴

输入参数

Data=开盘价、最高价、最低价和收盘价数据
开盘价、最高价、最低价和收盘价的数据,指定为矩阵、表格或时间表。对于矩阵输入,Data 是存储在相应列中的开盘价、最高价、最低价和收盘价的M矩阵。
数据类型:double|table|timetable
color= (可选) 三个元素颜色矢量
图形窗口的背景颜色(默认) |颜色矢量[R G B] |字符串
三个元素颜色矢量,指定为[R G B]颜色矢量或指定颜色名称的字符串。默认颜色因图形窗口的背景颜色而异。
数据类型:double|string

修改源代码

原始代码

function varargout = candle(varargin)%--------------------------- Parsing/Validation --------------------------%
trynarginchk(1,Inf);[ax,args] = internal.finance.axesparser(varargin{:});if ~isempty(ax) && ~isscalar(ax)error(message('finance:internal:finance:axesparser:ScalarAxes'))endoutput = internal.finance.ftseriesInputParser(args, ...4,{'open','high','low','close'},{},{},{'Color'},{''},{@(x)1,@ischar},1);
catch MEthrowAsCaller(ME)
end[data,optional,dates,~] = output{:};
op = data(:,1);
hi = data(:,2);
lo = data(:,3);
cl = data(:,4);% Validation work will be left to child functions.
color = optional.Color;%------------------------------ Data Preparation -------------------------%% Need to pad all inputs with NaN's to leave spaces between day data
% Vertical High/Low lines data preparation.
numObs = length(hi(:));hiloVertical = [hi lo NaN(numObs, 1)]';
indexVertical = repmat(dates',3,1);% Boxes data preparation
if isdatetime(dates) && length(dates) > 1%If using datetimes, make the box width one half of the smallest%distance between datesinc = 1/4 * min(diff(dates));
elseinc = 0.25;
end
indexLeft = dates - inc;
indexRight = dates + inc;%------------------------------- Plot ------------------------------------%if isempty(ax)ax = gca;
end% Store NextPlot flag (and restore on cleanup):
next = get(ax,'NextPlot');
cleanupObj = onCleanup(@()set(ax,'NextPlot',next));backgroundColor = get(ax,'color');
if isempty(color)cls = get(ax, 'colororder');color = cls(1, :);
endh = gobjects(numObs+1,1); % Preallocate% Plot vertical lines
h(1) = plot(ax,indexVertical(:),hiloVertical(:),'Color',color,'AlignVertexCenters','on');set(ax,'NextPlot','add')% Plot filled boxes
colorSet = {backgroundColor,color};% Filled the boxes when opening price is greater than the closing price.
filledIndex = ones(numObs, 1);
filledIndex(op > cl) = 2;tryfor i = 1 : numObsh(i+1) = fill(ax, ...[indexLeft(i); indexLeft(i); indexRight(i); indexRight(i)], ...[op(i); cl(i); cl(i); op(i)],colorSet{filledIndex(i)},'Edgecolor',color, ...'AlignVertexCenters', 'on');end
catch MEthrowAsCaller(ME)
endswitch nextcase {'replace','replaceall'}grid(ax, 'on')case {'replacechildren','add'}  % Do not modify axes properties
endif nargout % Not equal to 0varargout = {h};
endend;

修改部分

1. 这里是输入参数的读取
2. 这里是单日数据的读取
由此可见，输入参数不管为多少个，只取前4个

1.这里是日期长度，后期画图plot需要的横坐标和纵坐标
2. 这里是画框需要的坐标点，后期画图fill需要的参数
3. 这里是坐标轴的设置，默认取当前axes

1. 这里是默认颜色的选取，可以不管
2. 这里是画直线的，也就是每天的最高价和最低价之间的细线，统一都是一种颜色，所以等会需要更
改，用两个plot分别画出不同颜色的细线
3. 这里是后面 fill 给框框填色的颜色选取，由前面的设置选取，需要更改
4. 这里是后面判断填色的依据

2处的修改

kk=indexVertical(:,find(op > cl));
kkk=hiloVertical(:,find(op > cl));
hh1=plot(ax,kk(:),kkk(:),'color',[0 150/255 0],'LineWidth',1,'AlignVertexCenters','on');
hold on;
kk=indexVertical(:,find(op <= cl));
kkk=hiloVertical(:,find(op <= cl));
hh2=plot(ax,kk(:),kkk(:),'r-','LineWidth',1,'AlignVertexCenters','on');
h = [hh1 hh2];

这里需要根据开盘价和收盘价设置响应的颜色，用两个 plot 得到相应的线，同时需要将句柄汇聚到h
3处的修改

colorSet = {[0 150/255 0],'red'};

这段代码的意思是跌了用深绿色，涨了用红色，深绿的RGB为[ 0 150 0]，但是matlab的参数必须小于1，得除以255.

这里使用啦填充颜色的，我们前面已经修改了填充的颜色，还需要更改边框的颜色，也就是Edgecolor，代码如下：

   for i = 1 : numObsh(i+2) = fill(ax, ...[indexLeft(i); indexLeft(i); indexRight(i); indexRight(i)], ...[op(i); cl(i); cl(i); op(i)],colorSet{filledIndex(i)},'Edgecolor',colorSet{filledIndex(i)}, ...'AlignVertexCenters', 'on');end

最终显示预览

好了，基本都已经修改完成了，代码也带着大家详细解读了，你可以根据自己的需求再更改其他的细节

修改后的完整代码

   function varargout = candle(varargin)
%CANDLE Candlestick chart.
%
% Syntax:
%
%   candle(Data)
%   candle(Data,Color)
%   candle(ax,___)
%
% Description:
%
%   CANDLE plots a candlestick chart from a series of opening, high, low,
%   and closing prices of a security. If the closing price is greater than the
%   opening price, the body (the region between the open and close price)
%   is unfilled; otherwise the body is filled.
%
% Input Argument:
%
%   Data    - A matrix, table, or timetable. For matrix input, Data is an
%             M-by-4 matrix of opening, high, low, and closing prices.
%             Timetables and tables with M rows contain variables named
%             'Open', 'High', 'Low', and 'Close' (case insensitive).
%
% Optional Argument:
%
%   ax      - Valid axis object. The plot will be created in the axes specified
%             by ax instead of in the current axes (gca). The option ax can
%             precede any of the input argument combinations.
%
%   Color   - Three element color vector, [R G B], or a string specifying the
%             color name. The default color differs depending on the background
%             color of the figure window. See COLORSPEC for additional details.
%
% Output Argument:
%
%   h                    - Graphic handle of the figure.
%
%   See also HIGHLOW, KAGI, LINEBREAK, POINTFIG, PRICEANDVOL, RENKO, VOLAREA.%  Copyright 1995-2018 The MathWorks, Inc.%--------------------------- Parsing/Validation --------------------------%
trynarginchk(1,Inf);[ax,args] = internal.finance.axesparser(varargin{:});if ~isempty(ax) && ~isscalar(ax)error(message('finance:internal:finance:axesparser:ScalarAxes'))endoutput = internal.finance.ftseriesInputParser(args, ...4,{'open','high','low','close'},{},{},{'Color'},{''},{@(x)1,@ischar},1);
catch MEthrowAsCaller(ME)
end[data,optional,dates,~] = output{:};
op = data(:,1);
hi = data(:,2);
lo = data(:,3);
cl = data(:,4);% Validation work will be left to child functions.
color = optional.Color;%------------------------------ Data Preparation -------------------------%% Need to pad all inputs with NaN's to leave spaces between day data
% Vertical High/Low lines data preparation.
numObs = length(hi(:));hiloVertical = [hi lo NaN(numObs, 1)]';
indexVertical = repmat(dates',3,1);% Boxes data preparation
if isdatetime(dates) && length(dates) > 1%If using datetimes, make the box width one half of the smallest%distance between datesinc = 1/4 * min(diff(dates));
elseinc = 0.25;
end
indexLeft = dates - inc;
indexRight = dates + inc;%------------------------------- Plot ------------------------------------%if isempty(ax)ax = gca;
end% Store NextPlot flag (and restore on cleanup):
next = get(ax,'NextPlot');
cleanupObj = onCleanup(@()set(ax,'NextPlot',next));backgroundColor = get(ax,'color');
if isempty(color)cls = get(ax, 'colororder');color = cls(1, :);
endh = gobjects(numObs+1,1); % Preallocate% Plot vertical lines
kk=indexVertical(:,find(op > cl));
kkk=hiloVertical(:,find(op > cl));
hh1=plot(ax,kk(:),kkk(:),'color',[0 150/255 0],'LineWidth',1,'AlignVertexCenters','on');
hold on;
kk=indexVertical(:,find(op <= cl));
kkk=hiloVertical(:,find(op <= cl));
hh2=plot(ax,kk(:),kkk(:),'r-','LineWidth',1,'AlignVertexCenters','on');
h = [hh1 hh2];set(ax,'NextPlot','add')% Plot filled boxes
colorSet = {[0 150/255 0],'red'};% Filled the boxes when opening price is greater than the closing price.
filledIndex = ones(numObs, 1);
filledIndex(op <= cl) = 2;tryfor i = 1 : numObsh(i+2) = fill(ax, ...[indexLeft(i); indexLeft(i); indexRight(i); indexRight(i)], ...[op(i); cl(i); cl(i); op(i)],colorSet{filledIndex(i)},'Edgecolor',colorSet{filledIndex(i)}, ...'AlignVertexCenters', 'on');end
catch MEthrowAsCaller(ME)
endswitch nextcase {'replace','replaceall'}grid(ax, 'on')case {'replacechildren','add'}  % Do not modify axes properties
endif nargout % Not equal to 0varargout = {h};
endend