построить последовательность чисел с разными цветами

Question

У меня есть случайный список 0 и 1 длиной> 300. Я хотел бы построить список с 1 зеленым и 0 красным, как показано на рисунке ниже * pi c. Каков наилучший способ сделать это в matplotlib?

Answer 1

Вы можете использовать таблицу matplotlib :

import matplotlib.pyplot as plt

data = [0,1,0,1,1,0]  # Setup data list
fig, ax = plt.subplots(figsize=(len(data)*0.5, 0.5))  # Setup figure
ax.axis("off")  # Just want table, no actual plot

# Create table, with our data array as the single row, and consuming the whole figure space
t = ax.table(cellText=[data], loc="center", cellLoc="center", bbox=[0,0,1,1])

# Iterate over cells to colour them based on value
for idx, cell in t.get_celld().items():
    if data[idx[1]] == 1:
        c = 'g'
    else:
        c = 'r'
    cell.set_edgecolor(c)
    cell.set_facecolor(c)

fig.show()