Matplotlib仅在堆叠条形图上每2条中绘制1条

时间:2020-08-13 15:55:36

标签: python numpy matplotlib plot graph

我正在尝试从loglList下绘制堆积的条形图。 'up''down'用于确定颜色,整数是条形表示的百分比。数组可以更改大小,所以这就是为什么我有for循环的原因。但是,当我运行代码时,仅绘制出2条中的1条,其余的是它们之间的空白。尽管有if声明,但它们的颜色也不正确。 我想知道是否有人对此有何见解?谢谢

import numpy as np
import matplotlib.pyplot as plt

loglist = [['up', 0.8519966683633334], ['down', 0.09023182638471149], ['up', 4.120008329091666], ['down', 0.09023182638471149], ['up', 0.3441534403775855], ['down', 0.089075008097728], ['up', 0.3447318495210772], ['down', 1.650779695525427], ['up', 11.537527185229743], ['down', 22.519781592707417], ['up', 14.05245014113183], ['down', 4.601823145620286], ['up', 3.8163435287585026], ['down', 2.34313544028504], ['up', 7.137568830688075], ['down', 0.7415205219564112], ['up', 3.7324742029522002], ['down', 21.91534403775855]]

width = 0.3
barDict = []

for index, plotLog in enumerate(logList):
    if(plotLog[0] == "up"):
        colour = "g"
    elif(plotLog[0] == "down"):
        colour = "r"

    bottom = 0

    for i in range(index):
        bottom = bottom + logList[i][1]


    barDict[index] = plt.bar(1, plotLog[1], width, bottom=bottom, color=colour)

plt.yticks(np.arange(0, 101, 10))
plt.show()

2 个答案:

答案 0 :(得分:0)

您的代码似乎有多种错别字,因此很难猜测其真正意图。这是最少的重写:

import numpy as np
import matplotlib.pyplot as plt

logList = [['up', 0.8519966683633334], ['down', 0.09023182638471149], ['up', 4.120008329091666], ['down', 0.09023182638471149], ['up', 0.3441534403775855], ['down', 0.089075008097728], ['up', 0.3447318495210772], ['down', 1.650779695525427], ['up', 11.537527185229743], ['down', 22.519781592707417], ['up', 14.05245014113183], ['down', 4.601823145620286], ['up', 3.8163435287585026], ['down', 2.34313544028504], ['up', 7.137568830688075], ['down', 0.7415205219564112], ['up', 3.7324742029522002], ['down', 21.91534403775855]]

width = 0.3
bottom = 0
for index, plotLog in enumerate(logList):
    if (plotLog[0] == "up"):
        colour = "g"
    elif (plotLog[0] == "down"):
        colour = "r"
    plt.bar(1, plotLog[1], width, bottom=bottom, color=colour)
    bottom += plotLog[1]
plt.yticks(np.arange(0, 101, 10))
plt.xticks([])
plt.show()

example plot

或者,您可以使用以下方法绘制相同的图:

width = 0.3
vals = np.array([plotLog[1] for plotLog in logList])
colours = ['g' if plotLog[0] == "up" else 'r' for plotLog in logList]
bars = plt.bar(np.ones_like(vals), vals, bottom=np.append(0, np.cumsum(vals[:-1])), color=colours, width=width)

答案 1 :(得分:0)

几个错误。试试:

import numpy as np
import matplotlib.pyplot as plt

logList = [['up', 0.8519966683633334], ['down', 0.09023182638471149], ['up', 4.120008329091666], ['down', 0.09023182638471149], ['up', 0.3441534403775855], ['down', 0.089075008097728], ['up', 0.3447318495210772], ['down', 1.650779695525427], ['up', 11.537527185229743], ['down', 22.519781592707417], ['up', 14.05245014113183], ['down', 4.601823145620286], ['up', 3.8163435287585026], ['down', 2.34313544028504], ['up', 7.137568830688075], ['down', 0.7415205219564112], ['up', 3.7324742029522002], ['down', 21.91534403775855]]

width = 0.3
barDict = []

for index, plotLog in enumerate(logList):
    if(plotLog[0] == "up"):
        colour = "g"
    elif(plotLog[0] == "down"):
        colour = "r"

    bottom = 0

    for i in range(index):
        bottom = bottom + logList[i][1]

    print(plotLog, plotLog[0])
    barDict.append(plt.bar(1, plotLog[1], width, bottom=bottom, color=colour))

plt.yticks(np.arange(0, 101, 10))
plt.show()