Векторизация в python в следующем коде

Question

Я пытаюсь построить график Ганта. Я новичок в python и не знаком с передовыми концепциями программирования. При запуске кода я получаю показанную ошибку.

import numpy as np 
import matplotlib as mpl 
import pylab as plt 

arr=np.loadtxt('gantt.csv',dtype=float,delimiter=",") 

colormapd = { 
    1:"r", 
    2:"g", 
    3:"b", 
    4:"y", 
    5:"m", 
    6:"k", 
    7:"r", 
    8:"g", 
    9:"b", 
    0:"y", 
    } 

therange=range(500,2500) 

jobnum= arr[therange,0].astype(int) 
macnum= arr[therange,2].astype(int) 
procstart = arr[therange,3] 
procfinish = arr[therange,4] 


for i in range(500,2500): 
    plt.hlines(macnum[i],procstart[i],procfinish[i],colors = colormapd[1]) 
    plt.show() 

Ошибка:

IndexError: index 2000 is out of bounds for axis 0 with size 2000 

Я думаю, что мне нужно, чтобы иметь возможность векторизации (jobnum% 10) массив, используемый для доступа к словарю.

plt.hlines(macnum,procstart,procfinish,colors = colormapd[jobnum%10]) 

Я могу запустить:

plt.hlines(macnum,procstart,procfinish) 

Но я хочу, чтобы изменить цвет линий в соответствии с jobnumbers. У меня 2500 рабочих мест. Любой другой метод создания диаграммы Ганта, если лучше, можно было бы предложить.

Answer 1

Ваши массивы имеют индексы 0-1999, а не 500-2499. Поэтому, когда цикл for достигает 2000, индекс выходит за пределы массивов.

Чтобы установить отдельные цвета, перейдите в массив с цветами, соответствующими данным. Нравится так:

colors = np.array([colormapd[i%10] for i in therange]) 

for i in range(50,250): 
    plt.hlines(macnum,procstart,procfinish,colors=colors) 
    plt.show() 

Answer 2

Вы можете включить в свой вопрос полный трассировочный ответ, неясно, в какой строке ваш код не работает.

также вы можете опубликовать фрагмент вашего файла csv и часть массива.

Это, как говорится, , если у вас есть 2000 пунктов в файле CSV, то ваш индекс первого элемента будет 0 и последний 1999

ваш двухтысячный элемент в массиве действительно с индексом 1999, так как индексация массива начинается с 0.

также использование numpy.arange (старт, конец) нет python родственный плита. и помните, что диапазон и диапазон не включают конечную точку.

вот несколько примеров кода

macnum = np.arange(5,26) 
procstart = np.linspace(0.0,23.5,21) 
procend = procstart + .5 

print macnum, len(macnum) 
print procstart, len(procstart) 
print procend, len(procend) 

первые три аргумента в plt.hlines могут принимать массивы так просто передать свои массивы непосредственно функции.

plt.hlines(macnum, procstart, procend) 
plt.show() 

matplotlib имеет встроенные цветовые карты, но я не совсем уверен, как их использовать.