pandas 初识(六）-可视化

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt

df = pd.DataFrame({
    ‘color‘: [‘red‘,‘red‘,‘red‘,‘blue‘,‘blue‘,‘blue‘], 
    ‘x‘: [0,1,2,3,4,5],‘y‘: [0,1,2,9,16,25]
})

fig, ax = plt.subplots()

for key, group in df.groupby([‘color‘]):
    ax = group(ax=ax, kind=‘line‘, x=‘x‘, y=‘y‘, c=key, label=key)

plt.legend()
plt.show()

ax = None
for key, group in df.groupby([‘color‘]):
    ax = group(ax=ax, kind=‘line‘, x=‘x‘, y=‘y‘, c=key, label=key)

plt.legend()
plt.show()