Comment puis-je tracer un cadre de données pandas avec des couleurs de ligne différentes pour chaque colonne ?

Question

Le titre de la question explique tout. Je voudrais tracer un cadre de données. Disons, 7 lignes et 3 colonnes. Je voudrais tracer chaque colonne avec des couleurs aléatoires. Comment puis-je le faire ?

Answer 1

Comme vous ne disposez pas d'un énorme ensemble de données, vous pouvez créer un dictionnaire appelé color_dict et rechercher les couleurs à partir de celui-ci lors du traçage.

import pandas as pd

data = {
    'time0': [41, 28, 33, 34, 38, 31, 37],
    'time1': [48, 26, 39, 33, 58, 41, 43],
    'time2': [53, 30, 51, 37, 48, 49, 53]
}

df = pd.DataFrame(data=data)

import random

color_dict = {}
for idx in range(df.shape[1]):
    r = random.random()
    b = random.random()
    g = random.random()
    color = (r, g, b)
    color_dict[idx] = color 

colors = [color_dict.get(x) for x in range(df.shape[1])]

import matplotlib.pyplot as plt

for idx in range(df.shape[1]):
    plt.plot(df.iloc[:,idx], color=colors[idx])