Блог

Я хотел бы нарисовать точки разными цветами в соответствии с соответствующей строкой. Как назначить цвет строкам, начинающимся с C или D?

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

RV, sig, f = np.loadtxt('file', unpack=True, usecols=[2, 3, 4])
sp, yr = np.loadtxt('file', unpack=True, dtype='str', usecols=[0, 1])

fig, ax = plt.subplots(figsize=[10,7])

d = {'RV': RV, 'sig': sig, 'faze': f, 'yr': yr, 'sp': sp}
df = pd.DataFrame(data=d)

colors = {'AB':'#008f00', 'BB':'#e36500', 'C*': 'brown', 'D*': 'green'} # ????????????????

size_p=10
ax.scatter(df['faze'], df['RV'], c = df['sp'].map(colors), marker = "o", s=size_p, zorder = 1)

plt.show()