Цвет столбца гонки гистограммы Python - PullRequest
Новая
       22

Цвет столбца гонки гистограммы Python

0 голосов
/ 11 ноября 2019

Я создаю анимированную гистограмму в блокноте / Python Jupyter, адаптированную из этого примера . Я хотел бы назначить цвет для каждого столбца для каждого состояния. В настоящее время столбцу присваивается цвет, но при запуске гистограммы и изменении названий состояний в зависимости от количества самолетов порядок цветов остается прежним. Пример, который я использую, включает в себя этап группировки, который мне не нужен, поэтому я назначил цвета с помощью следующего: 

colors = ["#adb0ff", "#ffb3ff", "#90d595", "#e48381", "#aafbff", "#f7bb5f", "#eafb50"]

Я попытался добавить идентификаторы состояния и затем цвета, но получил "Invalid RGBAаргумент "ошибки, когда я назначаю цвета, используя код ниже. Состояния и цвета в приведенных ниже примерах сокращены.

Пример 1 

colors = dict(zip(
    ['AL', 'AK', 'AZ'],
    ["#adb0ff", "#ffb3ff", "#90d595"]
))
group = df.set_index('Aircraft').to_dict()

Пример 2 

colors =(
    ['AL', 'AK', 'AZ'],
     ['#adb0ff', '#ffb3ff', '#90d595']
)

Вот код, который я использую. 

fig, ax = plt.subplots(figsize=(15, 8))
dff = dff[::-1]
ax.barh(dff['State'], dff['Aircraft'], color=colors)
for i, (Aircraft, State) in enumerate(zip(dff['Aircraft'], dff['State'])):
    ax.text(Aircraft, i,     State,            ha='right')
    ax.text(Aircraft, i,     Aircraft, ha='left')
ax.text(1, 0.4, current_year, transform=ax.transAxes, size=46, ha='right')

Спасибо за вашу помощь.

...