绘制柱状图的代码如下:
ax.bar(
np.arange(0, len(stock_data.index)-begin),
stock_data.change[begin:],
color=['g' if stock_data.open[x] > stock_data.close[x] else 'r' for x in range(begin, len(stock_data.index))])
柱的颜色根据当天开盘价和收盘价的大小来定,开盘价大于收盘价则为绿柱,否则为红柱。
其他设置代码同之前,完整代码如下:
# 仅制作单股票的换手率柱状图
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(