Радарная диаграмма не центрирована в matplotlib

Я пытаюсь узнать, как работают радарные диаграммы в matplotlib. Я использую код из этого потока, но сюжет, который я создаю, не центрирован правильно, и отсутствует ось. Я пробовал с matplotlib 1.3.1, 1.4.1 и 1.5.1 на случай, если что-то изменилось в последних версиях.

введите здесь описание изображения

import numpy as np
import pylab as pl

class Radar(object):

    def __init__(self, fig, titles, labels, rect=None):
        if rect is None:
            rect = [0.05, 0.05, 0.95, 0.95]

        self.n = len(titles)
        self.angles = np.arange(90, 90+360, 360.0/self.n)
        self.axes = [fig.add_axes(rect, projection="polar", label="axes%d" % i) 
                         for i in range(self.n)]

        self.ax = self.axes[0]
        self.ax.set_thetagrids(self.angles, labels=titles, fontsize=14)

        for ax in self.axes[1:]:
            ax.patch.set_visible(False)
            ax.grid("off")
            ax.xaxis.set_visible(False)

        for ax, angle, label in zip(self.axes, self.angles, labels):
            ax.set_rgrids(range(1, 6), angle=angle, labels=label)
            ax.spines["polar"].set_visible(False)
            ax.set_ylim(0, 5)

    def plot(self, values, *args, **kw):
        angle = np.deg2rad(np.r_[self.angles, self.angles[0]])
        values = np.r_[values, values[0]]
        self.ax.plot(angle, values, *args, **kw)



fig = pl.figure(figsize=(6, 6))

titles = list("ABCDE")

labels = [
    list("abcde"), list("12345"), list("uvwxy"), 
    ["one", "two", "three", "four", "five"],
    list("jklmn")
]

radar = Radar(fig, titles, labels)
radar.plot([1, 3, 2, 5, 4],  "-", lw=2, color="b", alpha=0.4, label="first")
radar.plot([2.3, 2, 3, 3, 2],"-", lw=2, color="r", alpha=0.4, label="second")
radar.plot([3, 4, 3, 4, 2], "-", lw=2, color="g", alpha=0.4, label="third")
radar.ax.legend()

python matplotlib radar-chart
person FranGoitia    schedule 13.08.2016    source источник

Ответы (1)


arrow_upward
4
arrow_downward

У меня была такая же проблема, но я обнаружил, что проблема заключалась в следующем:

self.angles = np.arange(90, 90+360, 360.0/self.n)

Так что измените его на

self.angles = np.arange(0, 360, 360.0/self.n)

и поверните каждый ax с ax.set_theta_offset(np.deg2rad(90)) вместо этого.

Модифицированный класс выглядит так:

# Python 3.4
# matplotlib 1.5.3
class Radar(object):
    def __init__(self, fig, titles, labels, rotation=0, rect=None):
        if rect is None:
            rect = [0.05, 0.05, 0.95, 0.95]

        self.n = len(titles)
        self.angles = np.arange(0, 360, 360.0/self.n)
        self.axes = [fig.add_axes(rect, projection="polar", label="axes%d" % i) 
                         for i in range(self.n)]

        self.ax = self.axes[0]
        self.ax.set_thetagrids(self.angles, labels=titles, fontsize=14)

        for ax in self.axes[1:]:
            ax.patch.set_visible(False)
            ax.grid("off")
            ax.xaxis.set_visible(False)

        for ax, angle, label in zip(self.axes, self.angles, labels):
            ax.set_rgrids(range(1, 6), angle=angle, labels=label)
            ax.spines["polar"].set_visible(False)
            ax.set_ylim(0, 6)
            ax.set_theta_offset(np.deg2rad(rotation))

    def plot(self, values, *args, **kw):
        angle = np.deg2rad(np.r_[self.angles, self.angles[0]])
        values = np.r_[values, values[0]]
        self.ax.plot(angle, values, *args, **kw)

Примечание. Я добавил параметр rotation=0 к __init__ и применил поворот ко всем ax в последнем цикле.

Я знаю, что прошло некоторое время с тех пор, как вопрос был задан, но я предполагаю, что кто-то еще наткнется на это.

person Thomas Fauskanger    schedule 14.12.2016
comment
Я ценю ваши отзывы. Я предполагал, что хотя это был старый пост, кто-то придет позже. - person Thomas Fauskanger; 29.03.2017