Matplotlib adding legend based on existing color series

Question

I plotted some data using scatter plot and specified it as such:

plt.scatter(rna.data[\'x\'], rna.data[\'y\'], s=size,
                    c=rna.data[\'colors         


        

Answer 1

Altair can be a great choice here.

Continuous classes

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

df = pd.DataFrame(40*np.random.randn(10, 3), columns=['A', 'B','C'])

from altair import *
Chart(df).mark_circle().encode(x='A',y='B', color='C').configure_cell(width=200, height=150)

Discrete classes

df = pd.DataFrame(10*np.random.randn(40, 2), columns=['A', 'B'])
df['C'] = np.random.choice(['alpha','beta','gamma','delta'], size=40)

from altair import *
Chart(df).mark_circle().encode(x='A',y='B', color='C').configure_cell(width=200, height=150)