Clockwise polar plot with 0 deg at the top
Asked Answered
B

4

19

How can I make a clockwise polar plot? Somebody ask a similar question here: How to make the angles in a matplotlib polar plot go clockwise with 0° at the top?, But I don't understand this:

import matplotlib.pyplot as plt
import numpy as np

fig = plt.figure()
ax = fig.add_subplot(111, polar=True)
ax.grid(True)

theta = np.arange(0,370,10)
theta = [i*np.pi/180.0 for i in theta]  # convert to radians

x = [3.00001,3,3,3,3,3,3,3,3,3,3,3,3,3,2.5,2,2,2,2,2,1.5,1.5,1,1.5,2,2,2.5,2.5,3,3,3,3,3,3,3,3,3]
ax.plot(theta, x)
plt.show()

EDIT:

import matplotlib.pyplot as plt
import numpy as np
from matplotlib.projections import PolarAxes, register_projection
from matplotlib.transforms import Affine2D, Bbox, IdentityTransform

class NorthPolarAxes(PolarAxes):
    '''
    A variant of PolarAxes where theta starts pointing north and goes
    clockwise.
    '''
    name = 'northpolar'

    class NorthPolarTransform(PolarAxes.PolarTransform):
        def transform(self, tr):
            xy   = np.zeros(tr.shape, np.float_)
            t    = tr[:, 0:1]
            r    = tr[:, 1:2]
            x    = xy[:, 0:1]
            y    = xy[:, 1:2]
            x[:] = r * np.sin(t)
            y[:] = r * np.cos(t)
            return xy

        transform_non_affine = transform

        def inverted(self):
            return NorthPolarAxes.InvertedNorthPolarTransform()

    class InvertedNorthPolarTransform(PolarAxes.InvertedPolarTransform):
        def transform(self, xy):
            x = xy[:, 0:1]
            y = xy[:, 1:]
            r = np.sqrt(x*x + y*y)

fig = plt.figure()
register_projection(NorthPolarAxes)
ax=plt.subplot(1, 1, 1, projection='northpolar')    
theta=np.linspace(0,2*np.pi,37)
x = [3.00001,3,3,3,3,3,3,3,3,3,3,3,3,3,2.5,2,2,2,2,
     2,1.5,1.5,1,1.5,2,2,2.5,2.5,3,3,3,3,3,3,3,3,3]
ax.plot(theta, x)
plt.show()

How to use register_projection(NorthPolarAxes) correctly?

Benedictbenedicta answered 5/10, 2011 at 15:54 Comment(0)
C
31

Add these lines:

ax.set_theta_direction(-1)
ax.set_theta_offset(pi/2.0)
Chromonema answered 28/8, 2013 at 11:21 Comment(0)
S
16
ax.set_theta_direction(-1)
ax.set_theta_zero_location('N')

is slightly more comprehensible.

Sculpture answered 4/10, 2014 at 14:8 Comment(0)
C
3

Edit: Please note that Pavel has provided a much better solution!


The SO question you linked to contains the answer. Here is a slightly modified version of ptomato's NorthPolarAxes class with theta=0 pointing East and increasing clockwise:

import matplotlib.pyplot as plt
import numpy as np
import matplotlib.projections as projections
import matplotlib.transforms as mtransforms

class EastPolarAxes(projections.PolarAxes):
    '''
    A variant of PolarAxes where theta starts pointing East and goes
    clockwise.
    https://mcmap.net/q/419163/-how-to-make-the-angles-in-a-polar-plot-go-clockwise-with-0-176-at-the-top#2433287
    https://mcmap.net/q/630768/-clockwise-polar-plot-with-0-deg-at-the-top#7664545    
    '''
    name = 'eastpolar'

    class EastPolarTransform(projections.PolarAxes.PolarTransform):
        """
        The base polar transform.  This handles projection *theta* and
        *r* into Cartesian coordinate space *x* and *y*, but does not
        perform the ultimate affine transformation into the correct
        position.
        """        
        def transform(self, tr):
            xy   = np.zeros(tr.shape, np.float_)
            t    = tr[:, 0:1]
            r    = tr[:, 1:2]
            x    = xy[:, 0:1]
            y    = xy[:, 1:2]
            x[:] = r * np.cos(-t)
            y[:] = r * np.sin(-t)
            return xy

        transform_non_affine = transform

        def inverted(self):
            return EastPolarAxes.InvertedEastPolarTransform()

    class InvertedEastPolarTransform(projections.PolarAxes.InvertedPolarTransform):
        """
        The inverse of the polar transform, mapping Cartesian
        coordinate space *x* and *y* back to *theta* and *r*.
        """        
        def transform(self, xy):
            x = xy[:, 0:1]
            y = xy[:, 1:]
            r = np.sqrt(x*x + y*y)
            theta = npy.arccos(x / r)
            theta = npy.where(y > 0, 2 * npy.pi - theta, theta)
            return np.concatenate((theta, r), 1)

        def inverted(self):
            return EastPolarAxes.EastPolarTransform()

    def _set_lim_and_transforms(self):
        projections.PolarAxes._set_lim_and_transforms(self)
        self.transProjection = self.EastPolarTransform()
        self.transData = (
            self.transScale + 
            self.transProjection + 
            (self.transProjectionAffine + self.transAxes))
        self._xaxis_transform = (
            self.transProjection +
            self.PolarAffine(mtransforms.IdentityTransform(), mtransforms.Bbox.unit()) +
            self.transAxes)
        self._xaxis_text1_transform = (
            self._theta_label1_position +
            self._xaxis_transform)
        self._yaxis_transform = (
            mtransforms.Affine2D().scale(np.pi * 2.0, 1.0) +
            self.transData)
        self._yaxis_text1_transform = (
            self._r_label1_position +
            mtransforms.Affine2D().scale(1.0 / 360.0, 1.0) +
            self._yaxis_transform)

def eastpolar_axes():
    projections.register_projection(EastPolarAxes)
    ax=plt.subplot(1, 1, 1, projection='eastpolar')    
    theta=np.linspace(0,2*np.pi,37)
    x = [3.00001,3,3,3,3,3,3,3,3,3,3,3,3,3,2.5,2,2,2,2,
         2,1.5,1.5,1,1.5,2,2,2.5,2.5,3,3,3,3,3,3,3,3,3]
    ax.plot(theta, x)
    plt.show()

eastpolar_axes()

enter image description here


The doc strings from matplotlib/projections/polar.py's PolarTransform and InvertedPolarTransform were added because I think they help explain what each component is doing. That guides you in changing the formulas.

To get clockwise behavior, you simply change t --> -t:

        x[:] = r * np.cos(-t)
        y[:] = r * np.sin(-t)

and in InvertedEastPolarTransform, we want to use 2 * npy.pi - theta when y > 0 (the upper half-plane) instead of when y < 0.

Cauterize answered 5/10, 2011 at 16:25 Comment(0)
O
0

Short answer

Create your subplot using this code:

kw = dict(projection = 'polar',     # <-- polar projection
          theta_offset = np.pi/2,   # <-- rotate theta
          theta_direction = -1)     # <-- theta clockwise

_, ax = plt.subplots(subplot_kw=kw)

Explanation

When using projection = 'polar', the subplot created uses a projection of class matplotlib.projections.polar.PolarAxes which accepts parameters theta_offset and theta_direction.

Alternatively you can create the polar subplot without parameters, and later call ax.set_theta_zero_location` and ax.set_theta_direction to offset theta axis and change its direction.

You may look at other methods like set_thetalim

Example

import numpy as np
import matplotlib.pyplot as plt

# Theta
theta_deg = np.linspace(0, 360, num=145)
theta_rad = np.radians(theta_deg)

# Rho
rho = abs(np.cos(theta_rad))

# Create polar subplot
kw = dict(projection = 'polar',
          theta_offset = np.pi/2,
          theta_direction = -1)
_, ax = plt.subplots(subplot_kw=kw)

# Plot rho against theta
ax.plot(theta_rad, rho)

enter image description here

Oakleil answered 17/6 at 12:16 Comment(0)

© 2022 - 2024 — McMap. All rights reserved.