如何在折线图下方从上一个值增加时以绿色填充体积条形图，在从先前值减少时以红色填充该条形图

Question

import os
from matplotlib.backends.backend_pdf import PdfPages
import  pandas as pd
import matplotlib.pyplot as plt
import datetime as dt
plt.rcParams.update({'figure.max_open_warning': 0})

pp = PdfPages('multipage.pdf')

pth = "D:/Technical_Data/"
for fle in os.listdir(pth):
    df = pd.read_csv(os.path.join(pth, fle),usecols=(0,4,5,10))
    if not df.empty:
       df=df.astype(float)
       days = pd.to_datetime(df['indx'].astype(str), format='%Y%m%d')
       value = df['Close']
       open_v = df['MA_50']
       fig = plt.figure()
       plt.plot_date(x=days, y=value,fmt="r-")
       plt.plot_date(x=days, y=open_v,fmt="g-")
       fig.autofmt_xdate()
       plt.xticks(2013,2014,2015,2016)
       plt.ylabel("Price")
       plt.grid(True)
       pp.savefig()
       volume = df['Volume']
       plt.clf()
pp.close()

我使用此代码在Y轴上填充两列，在X轴上填充日期值的折线图。 我想做的另一件事是在折线图下方添加音量条。 当音量从以前的值减小时，颜色为红色；增大时，颜色为绿色。

indx    open    High    Low Close   Volume
20141015    0.345   0.365   0.32    0.34    5540100
20141016    0.34    0.34    0.32    0.325   2044700
20141017    0.32    0.35    0.315   0.34    1818000
20141020    0.36    0.39    0.36    0.385   798900
20141021    0.39    0.41    0.39    0.395   2981500
20141022    0.415   0.43    0.39    0.39    2997900
20141023    0.4 0.4 0.37    0.375   458900
20141024    0.375   0.39    0.375   0.385   90000
20141027    0.38    0.38    0.355   0.38    562900
20141028    0.375   0.38    0.36    0.37    684300
20141029    0.375   0.38    0.365   0.37    339300

老实说，我尝试自己做。 与仅python相比，我发现matplotlib很难。 任何帮助表示赞赏

Answer 1

我已将您的代码减少到最低限度，以便仅绘制条形图； 我将它留给您，以整合到脚本的其余部分中。

这是如何绘制DataFrame中的条形图，然后将其更改为红色（如果其高度小于上一个条形图的话）：

import matplotlib.pyplot as plt
import numpy as np
import pandas as pd

df = pd.DataFrame({
            'indx':[20141015,20141016,20141017,20141020,20141021,20141022,20141023,20141024,20141027,20141028,20141029],
            'Volume':[5540100,2044700,1818000,798900,2981500,2997900,458900,90000,562900,684300,339300]
            }).astype(float)


fig,ax = plt.subplots(1)

days = pd.to_datetime(df['indx'].astype(str), format='%Y%m%d')
volume = df['Volume']

# Plot all bars with green facecolor 
bars = ax.bar(np.array(days),volume,facecolor='g')

# Loop over bars
for i, bar in enumerate(bars):
    if i == 0: 
        continue

    # Find bar heights
    h = bar.get_height()
    h0 = bars[i-1].get_height()

    # Change bar color to red if height less than previous bar
    if h < h0: bar.set_facecolor('r')

fig.autofmt_xdate()

plt.show()

基本上，您遍历ax.bar创建的Rectangle补丁，并对其进行查询以找到其高度。 然后，使用.set_facecolor()更改相关的条形图。

外观如下：

最后说明一点：因为尺度是价格和体积之间如此不同，你可能需要使用一个twinx轴绘制音量条。 有关更多信息，请参见此示例 。