Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Update the "Rotating a 3D plot" gallery example to show all 3 rotation axes #21737

Merged
merged 2 commits into from Dec 4, 2021

Conversation

scottshambaugh
Copy link
Contributor

@scottshambaugh scottshambaugh commented Nov 24, 2021

PR Summary

This PR builds off of #21426 and updates the rotate-axes3d-sgskip.py example to show rotation about each individual view axis, with annotations. Here is the resulting animation:

3d_rotations.mp4

PR Checklist

Tests and Styling

  • [N/A] Has pytest style unit tests (and pytest passes).
  • Is Flake 8 compliant (install flake8-docstrings and run flake8 --docstring-convention=all).

Documentation

  • [N/A] New features are documented, with examples if plot related.
  • [N/A] New features have an entry in doc/users/next_whats_new/ (follow instructions in README.rst there).
  • [N/A] API changes documented in doc/api/next_api_changes/ (follow instructions in README.rst there).
  • Documentation is sphinx and numpydoc compliant (the docs should build without error).

@scottshambaugh
Copy link
Contributor Author

scottshambaugh commented Nov 24, 2021

Note that the code to generate that .mp4 is slightly different, but produces the same result:

from mpl_toolkits.mplot3d import axes3d
import matplotlib.pyplot as plt
import matplotlib.animation as animation

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

# Grab some example data.
X, Y, Z = axes3d.get_test_data(0.05)

# Plot a basic wireframe.
ax.plot_wireframe(X, Y, Z, rstride=10, cstride=10)

# Set the axis labels
ax.set_xlabel('x')
ax.set_ylabel('y')
ax.set_zlabel('z')

# Define an update function to pass to FuncAnimation
def update_view(frame):
    # Normalize angle to the range [-180, 180] for display
    angle = (frame + 180) % 360 - 180

    # Cycle through rotating the elevation, azimuth, roll, then all
    elev = azim = roll = 0
    if frame <= 360:
        elev = angle
    elif frame <= 360*2:
        azim = angle
    elif frame <= 360*3:
        roll = angle
    else:
        elev = azim = roll = angle
    
    # Update the axis view and title
    ax.view_init(elev, azim, roll)
    plt.title('Elevation: %d°, Azimuth: %d°, Roll: %d°' % (elev, azim, roll))

# Plot the animation
anim = animation.FuncAnimation(fig, update_view, frames=360*4+1, interval=15)
anim.save("3d_rotations.mp4")

@QuLogic QuLogic added status: needs workflow approval For PRs from new contributors, from which GitHub blocks workflows by default. topic: mplot3d labels Nov 24, 2021
@scottshambaugh scottshambaugh marked this pull request as ready for review December 3, 2021 16:53
@QuLogic
Copy link
Member

QuLogic commented Dec 4, 2021

Please rebase to go over the committed version of #21426.

Rename gallery example for clarity - this animation does not do any rotating
@timhoffm timhoffm added Documentation and removed status: needs rebase status: needs workflow approval For PRs from new contributors, from which GitHub blocks workflows by default. labels Dec 4, 2021
@timhoffm timhoffm added this to the v3.6.0 milestone Dec 4, 2021
@timhoffm timhoffm merged commit be94172 into matplotlib:main Dec 4, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

3 participants