3D plot aspect ratio [matplotlib]

Question

I am using the code written here to make a plot like this one below.

The problem is, that I want to adjust an aspect ratio, namely, to stretch it along the z-axis, so that all the stacked images are more or less visible. Is there an easy way of doing that?

Related: http://stackoverflow.com/a/10328142/1813349, second approach. — Alex Shpilkin, Jul 01 '16 at 20:20
In my os x I'm not able to modify any files in python bin directory (even with `sudo` access). This is due to some strange protection (see [this](http://stackoverflow.com/questions/32659348/operation-not-permitted-when-on-root-el-capitan-rootless-disabled)). And at this point I'm too lazy to do make this work as your other solution worked : ) But probably for the future it would be good idea to do that. — hayk, Jul 01 '16 at 23:58

score 4 · Accepted Answer · answered Jul 01 '16 at 20:16

Looks like there is no “proper” way to do it, but we can try monkey-patching our way around the problem. Here’s the best kludge I could do:

from mpl_toolkits.mplot3d import proj3d

def make_get_proj(self, rx, ry, rz):
    '''
    Return a variation on :func:`~mpl_toolkit.mplot2d.axes3d.Axes3D.getproj` that
    makes the box aspect ratio equal to *rx:ry:rz*, using an axes object *self*.
    '''

    rm = max(rx, ry, rz)
    kx = rm / rx; ky = rm / ry; kz = rm / rz;

    # Copied directly from mpl_toolkit/mplot3d/axes3d.py. New or modified lines are
    # marked by ##
    def get_proj():
        relev, razim = np.pi * self.elev/180, np.pi * self.azim/180

        xmin, xmax = self.get_xlim3d()
        ymin, ymax = self.get_ylim3d()
        zmin, zmax = self.get_zlim3d()

        # transform to uniform world coordinates 0-1.0,0-1.0,0-1.0
        worldM = proj3d.world_transformation(xmin, xmax,
                                             ymin, ymax,
                                             zmin, zmax)

        # adjust the aspect ratio                          ##
        aspectM = proj3d.world_transformation(-kx + 1, kx, ##
                                              -ky + 1, ky, ##
                                              -kz + 1, kz) ##

        # look into the middle of the new coordinates
        R = np.array([0.5, 0.5, 0.5])

        xp = R[0] + np.cos(razim) * np.cos(relev) * self.dist
        yp = R[1] + np.sin(razim) * np.cos(relev) * self.dist
        zp = R[2] + np.sin(relev) * self.dist
        E = np.array((xp, yp, zp))

        self.eye = E
        self.vvec = R - E
        self.vvec = self.vvec / proj3d.mod(self.vvec)

        if abs(relev) > np.pi/2:
            # upside down
            V = np.array((0, 0, -1))
        else:
            V = np.array((0, 0, 1))
        zfront, zback = -self.dist, self.dist

        viewM = proj3d.view_transformation(E, R, V)
        perspM = proj3d.persp_transformation(zfront, zback)
        M0 = np.dot(viewM, np.dot(aspectM, worldM)) ##
        M = np.dot(perspM, M0)
        return M
    return get_proj

# and later in the code:
ax.get_proj = make_get_proj(ax, 1, 1, 2)
ax.set_aspect(1.0)

That worked! Thanks a lot. But it's kinda weird you can't do that on the go. :\ — hayk, Jul 01 '16 at 23:31

score 4 · Answer 2 · answered Oct 22 '20 at 17:21

4

As of matplotlib 3.3.0, Axes3D.set_box_aspect seems to be the recommended approach.

import matplotlib.pyplot as plt

ax = plt.axes(projection='3d')

N = 10  # some number > 1 that stretches z axis as you desire
ax.set_box_aspect((1, 1, N))  # xy aspect ratio is 1:1, but stretches z axis

answered Oct 22 '20 at 17:21

Matt Panzer

1,141
10
7

1

I spent 3 days to find the solution that you provided! This worked for me, it is simple and easy! – Jonathan Pacheco Jul 19 '22 at 20:27

score 0 · Answer 3 · answered Jul 01 '16 at 16:30

0

Have you tried stretching your canvas in the vertical direction? For instance,

plt.figure(figsize=(5,10))

or something similar should give you more space. I am not sure, though, if that will solve your problem.

answered Jul 01 '16 at 16:30

mommermi

982
10
18

1

Nope. This does not affect the aspect ration. There should be something like `ax.set_aspect()`, to affect the bounding box aspect. – hayk Jul 01 '16 at 16:39

disccip · Answer 4 · 2016-07-01T16:44:03.487

0

Maybe you can try

plt.figure.subplots_adjust(bottom = 0.05)

This will makes your figure have more space along the Z-axis, I guess that's what you want.

Or you want

ax.set_xlim(xmin,xmax)

edited Jul 01 '16 at 16:44

answered Jul 01 '16 at 16:38

disccip

573
3
5
15

`ax.set_zlim` simply squeezes the axes, but does not affect on the aspect ratio of the bounding box. – hayk Jul 01 '16 at 16:48
subplots_adjust doesn't work? Sorry I don't understand what you really want. – disccip Jul 01 '16 at 16:52

3D plot aspect ratio [matplotlib]

4 Answers4

Linked