Matplotlib - Plot 3D with for loop

Question

I want to plot several 3D points with matplotlib. My coordinates are stored in 2D arrays because i got multiple cases and so i would like to plot all the cases in a same 3D plot with a "for loop" but when i do that, the results appeared on different plots...

As example :

from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
import numpy as np

X = np.array([[3,2,1],[4,5,6]])
Y = np.array([[1,2,1],[2,3,4]])
Z = np.array([[10,11,12],[13,12,16]])

for i in range(0,X.shape[0]):

    fig = plt.figure()
    ax = fig.add_subplot(111, projection='3d')

    ax.scatter(X[i,:], Y[i,:], Z[i,:], c='r', marker='o')

    ax.set_xlabel('Z')
    ax.set_ylabel('X')
    ax.set_zlabel('Y')

    plt.show()

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

2

You create a new figure each iteration and plot it each iteration. Also you always create the first suplot of a 1x1 subplot-grid.

You probably want a x.shape[0] x 1 grid or 1 x x.shape[0] grid:

from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
import numpy as np

X = np.array([[3,2,1],[4,5,6]])
Y = np.array([[1,2,1],[2,3,4]])
Z = np.array([[10,11,12],[13,12,16]])

# Create figure outside the loop
fig = plt.figure()

for i in range(0,X.shape[0]):
    
    # Add the i+1 subplot of the x.shape[0] x 1 grid
    ax = fig.add_subplot(X.shape[0], 1, i+1, projection='3d')

    ax.scatter(X[i,:], Y[i,:], Z[i,:], c='r', marker='o')

    ax.set_xlabel('Z')
    ax.set_ylabel('X')
    ax.set_zlabel('Y')
# Show it outside the loop
plt.show()

EDIT:

If you want to plot them all into the same plot use:

fig = plt.figure()
ax = fig.add_subplot(1, 1, 1, projection='3d')
ax.set_xlabel('Z')
ax.set_ylabel('X')
ax.set_zlabel('Y')

for i in range(0,X.shape[0]):
    # Only do the scatter inside the loop
    ax.scatter(X[i,:], Y[i,:], Z[i,:], c='r', marker='o')

plt.show()

edited Jun 20, 2020 at 9:12

CommunityBot

11 silver badge

answered Mar 8, 2016 at 10:37

MSeifert

154k41 gold badges356 silver badges377 bronze badges

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

4 Comments

user3601754 Over a year ago

Thanks for your help but i m trying to plot all the couples on a same grid.

MSeifert Over a year ago

@user3601754 - I've appended it at the end of the answer, check it out if that solves your problem.

user3601754 Over a year ago

and is it possible to use different markers or colors for each couple?

MSeifert Over a year ago

yes. The solution would be to define colors=['r', 'g', 'b'] and then change c=colors[i % 3]. For the markers the same applies.

Collectives™ on Stack Overflow

Matplotlib - Plot 3D with for loop

1 Answer 1

EDIT:

4 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

EDIT:

4 Comments

Your Answer

Sign up or log in

Post as a guest

Related