Trying to add a 3d subplot to a matplotlib figure

Question

So I'm trying to create a figure that presents a 3d plot from data points, along with the plots 3 projections in 3 other subplots. I can add the subplots for the projections with no problems, but when I try to place the 3 dimensional plot into the figure things backfire.

here's my code:

def plotAll(data):
    fig = plt.figure()
    plot_3d = fig.add_subplot(221)
    ax = Axes3D(plot_3d)  
    for i,traj in enumerate(data.values()):
        ax.plot3D([traj[0][-1]],[traj[1][-1]],[traj[2][-1]],".",color=[0.91,0.39,0.046])    
    #plot_12v13 = fig.add_subplot(222)
    #plot_projections(data,0,1)
    #plot_13v14 = fig.add_subplot(223)
    #plot_projections(data,1,2)
    #plot_12v14 = fig.add_subplot(224)
    #plot_projections(data,0,2)
    #plt.plot()

which throws back: 'AxesSubplot' object has no attribute 'transFigure'

I'm using matplotlib 0.99.3, any help would be greatly appreciated, thanks!

Prakhar Sharma · Accepted Answer · 2022-03-10 18:08:37Z

24

If you would like to use plt.subplots instead of plt.subplot (see the difference here), then you can do something like this one:

import matplotlib.pyplot as plt
from matplotlib import cm # for a scatter plot
from mpl_toolkits.mplot3d import Axes3D

fig, ax = plt.subplots(1,2,figsize=(10,10),subplot_kw=dict(projection='3d'))

sc1 = ax[0].scatter(x,y,z, c = true, cmap=cm.jet)
ax[0].set_title('True solution')

sc2 = ax[1].scatter(x,y,z c = y_pred, cmap=cm.jet)
ax[1].set_title('Predicted Solution')

Well, I don't know how to set individual axes as 3D using plt.subplots. It would be helpful if someone could comment down.

answered Mar 10, 2022 at 18:08

Prakhar Sharma

8661 gold badge16 silver badges29 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

Marcel Flygare Over a year ago

Please note the additional use of subplot_kw! This might be the parameter "you" are looking for.

Xidus · Accepted Answer · 2020-06-20 09:12:55Z

21

I was searching for a way to create my 3D-plots with the nice fig, axes = plt.subplots(...) shortcut, but since I just browsed Matplotlib's mplot3d tutorial, I want to share a quote from the top of this site.

New in version 1.0.0: This approach is the preferred method of creating a 3D axes.
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
Note

Prior to version 1.0.0, the method of creating a 3D axes was different. For those using older versions of matplotlib, change ax = fig.add_subplot(111, projection='3d') to ax = Axes3D(fig).

So if you have to use the <1.0.0 version of Matplotlib, this should be taken into account.

edited Jun 20, 2020 at 9:12

community wiki

Xidus

3 Comments

SuperElectric Over a year ago

It should be emphasized that the line "from mpl_toolkits.mplot3d import Axes3D" is necessary, even though you never directly use Axes3D. Without that import, add_subplot(111, projection='3d') will fail, saying that the projection '3d' is not recognized. Using "import mpl_toolkits.mplot3d" instead also works.

Luciano Over a year ago

The method in the tutorial is to use fig.gca(projection="3d") (instead of add_subplot) which doesn't appear to work. But your method does. Maybe it's just that I'm using an out-of-date version, but they should probably update the examples!

kuzand Over a year ago

I think the from mpl_toolkits.mplot3d import Axes3D is not needed anymore.

David Zwicker · Accepted Answer · 2011-12-15 15:49:50Z

5

The preferred way of creating an 3D axis is giving the projection keyword:

def plotAll(data):
    fig = plt.figure()
    ax = fig.add_subplot(221, projection='3d')
    for i,traj in enumerate(data.values()):
        ax.plot3D([traj[0][-1]],[traj[1][-1]],[traj[2][-1]],".",color=[0.91,0.39,0.046])    
    plot_12v13 = fig.add_subplot(222)
    plot_projections(data,0,1)
    plot_13v14 = fig.add_subplot(223)
    plot_projections(data,1,2)
    plot_12v14 = fig.add_subplot(224)
    plot_projections(data,0,2)
    plt.plot()

Unfortunately, you didn't supply a working example with suitable data, so I couldn't test the code. Also, I would recommend updating to a newer version of matplotlib.

answered Dec 15, 2011 at 15:49

David Zwicker

24.4k8 gold badges68 silver badges80 bronze badges

Comments

domonnss · Accepted Answer · 2025-04-03 09:25:41Z

0

matlabplot version 3.7.2

I prefer to use OOP-style code

ax = fig.add_subplot(111, projection='3d') # not oo style code

If you want to use OOP-style code

fig, ax = plt.subplots(subplot_kw={"projection": "3d"}) 
# it works but I dont like subplot_kw

edited Apr 3 at 9:25

answered Apr 3 at 9:25

domonnss

11 bronze badge

Collectives™ on Stack Overflow

Trying to add a 3d subplot to a matplotlib figure

4 Answers 4

1 Comment

Note

3 Comments

Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

4 Answers 4

1 Comment

Note

3 Comments

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related