добавление соединительных линий в точечный график в matplotlib - PullRequest
0 голосов
/ 29 февраля 2020

У меня есть функция индивидуальной настройки маркера и цвета для каждой точки данных.

def mscatter(x,y,ax=None, m=None, **kw):
   import matplotlib.markers as mmarkers
   if not ax: ax=plt.gca()
   sc = ax.scatter(x,y,**kw)
   if (m is not None) and (len(m)==len(x)):
     paths = []
     for marker in m:
        if isinstance(marker, mmarkers.MarkerStyle):
            marker_obj = marker
        else:
            marker_obj = mmarkers.MarkerStyle(marker)
        path = marker_obj.get_path().transformed(
                    marker_obj.get_transform())
        paths.append(path)
    sc.set_paths(paths)
  return sc

Затем у меня есть некоторые данные для x, y и соответствующие цвета и маркеры для x, y.

col = np.where(this_data['ifAgentCorrect'],"g","r")
mrkr = np.where(this_data['ifCorrectDecisions'],"o","x")
x = range(len(this_data['ifCorrectDecisions']))
y = this_data['cumSum']

plt.figure(figsize=(20, 8))
scatter = mscatter(x, y, c=col, m=mrkr)
plt.show()

Все, что мне нужно сделать, это соединить точку рассеяния с линиями. Что обычно делается с использованием таких маркеров, как «o-», «x-». Но matplotlib.markers не имеет «o-» или «x-» в списке маркеров. поэтому я получаю ошибку ValueError: Unrecognized marker style o-. Любые предложения о том, как я могу сделать это sh? Спасибо.

...