Skip to content

candlestick edgecolor #5160

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 3 commits into from
Closed

candlestick edgecolor #5160

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
ed0f7f1
candlestick edge color
lifetime42 Sep 30, 2015
10d798e
candlestick edge color
lifetime42 Oct 1, 2015
32503b3
candlestick body
lifetime42 Oct 1, 2015
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
48 changes: 34 additions & 14 deletions lib/matplotlib/finance.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -656,7 +656,7 @@ def _plot_day_summary(ax, quotes, ticksize=3,


def candlestick_ochl(ax, quotes, width=0.2, colorup='k', colordown='r',
alpha=1.0):
alpha=1.0, coloredge=None):

"""
Plot the time, open, close, high, low as a vertical line ranging
Expand All @@ -682,6 +682,8 @@ def candlestick_ochl(ax, quotes, width=0.2, colorup='k', colordown='r',
the color of the rectangle where close < open
alpha : float
the rectangle alpha level
coloredge : color (optional)
the color of the edge. default is color of the candlebody

Returns
-------
Expand All @@ -692,11 +694,11 @@ def candlestick_ochl(ax, quotes, width=0.2, colorup='k', colordown='r',
"""
return _candlestick(ax, quotes, width=width, colorup=colorup,
colordown=colordown,
alpha=alpha, ochl=True)
alpha=alpha, ochl=True, coloredge=coloredge)


def candlestick_ohlc(ax, quotes, width=0.2, colorup='k', colordown='r',
alpha=1.0):
alpha=1.0, coloredge=None):

"""
Plot the time, open, high, low, close as a vertical line ranging
Expand All @@ -722,6 +724,8 @@ def candlestick_ohlc(ax, quotes, width=0.2, colorup='k', colordown='r',
the color of the rectangle where close < open
alpha : float
the rectangle alpha level
coloredge : color (optional)
the color of the edge. default is color of the candlebody

Returns
-------
Expand All @@ -732,11 +736,11 @@ def candlestick_ohlc(ax, quotes, width=0.2, colorup='k', colordown='r',
"""
return _candlestick(ax, quotes, width=width, colorup=colorup,
colordown=colordown,
alpha=alpha, ochl=False)
alpha=alpha, ochl=False, coloredge=coloredge)


def _candlestick(ax, quotes, width=0.2, colorup='k', colordown='r',
alpha=1.0, ochl=True):
alpha=1.0, ochl=True, coloredge=None):

"""
Plot the time, open, high, low, close as a vertical line ranging
Expand All @@ -763,6 +767,8 @@ def _candlestick(ax, quotes, width=0.2, colorup='k', colordown='r',
the rectangle alpha level
ochl: bool
argument to select between ochl and ohlc ordering of quotes
coloredge : color (optional)
the color of the edge. default is color of the candlebody

Returns
-------
Expand All @@ -771,11 +777,18 @@ def _candlestick(ax, quotes, width=0.2, colorup='k', colordown='r',
added and patches is a list of the rectangle patches added

"""
if coloredge is None:
coloredgeup = colorup
coloredgedown = colordown
else:
coloredgeup = coloredge
coloredgedown = coloredge

OFFSET = width / 2.0

lines = []
patches = []

for q in quotes:
if ochl:
t, open, close, high, low = q[:5]
Expand All @@ -785,25 +798,32 @@ def _candlestick(ax, quotes, width=0.2, colorup='k', colordown='r',
if close >= open:
color = colorup
lower = open
height = close - open
height = close-open
vline = Line2D(xdata=(t, t), ydata=(low, high),
color=coloredgeup,
linewidth=0.5,
antialiased=True,
zorder=0)
else:
color = colordown
lower = close
height = open - close
height = open-close
vline = Line2D(xdata=(t, t), ydata=(low, high),
color=coloredgedown,
linewidth=0.5,
antialiased=True,
zorder=0)

vline = Line2D(
xdata=(t, t), ydata=(low, high),
color=color,
linewidth=0.5,
antialiased=True,
)
if coloredge is None:
edgecolor = color

rect = Rectangle(
xy=(t - OFFSET, lower),
width=width,
height=height,
facecolor=color,
edgecolor=color,
edgecolor=coloredge,
zorder=1,
)
rect.set_alpha(alpha)

Expand Down