3D Plots and Surface Plots

Duration: 5 min

This module delves into the creation of 3D plots and surface plots using Matplotlib, Seaborn, and Plotly in Python. These visualizations are crucial for understanding complex, multi-dimensional data, allowing for more intuitive insights and decision-making.

Creating 3D Plots with Matplotlib

Matplotlib's 3D plotting capabilities allow you to visualize data in a three-dimensional space. This can be particularly useful for datasets with multiple variables that interact in complex ways. By using the Axes3D class, you can create a 3D subplot and plot various types of 3D graphs, including scatter plots, line plots, and surface plots.

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

# Create a figure and a 3D Axes
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

# Generate data
x = np.linspace(-5, 5, 100)
y = np.linspace(-5, 5, 100)
x, y = np.meshgrid(x, y)
z = np.sin(np.sqrt(x**2 + y**2))

# Plot the surface
ax.plot_surface(x, y, z, cmap='viridis')

# Set labels and title
ax.set_xlabel('X Label')
ax.set_ylabel('Y Label')
ax.set_zlabel('Z Label')
plt.title('3D Surface Plot')

# Show plot
plt.show()

Try it in Google Colab:

A 3D surface plot will be displayed with X, Y, and Z axes labeled. The plot will show a surface colored according to the 'viridis' colormap, representing the sine of the square root of the sum of squares of X and Y.

Creating Interactive 3D Plots with Plotly

Plotly provides an excellent framework for creating interactive 3D plots. These plots can be embedded in web applications and allow users to rotate, zoom, and pan the plot to explore the data from different angles. This interactivity can significantly enhance the understanding of complex datasets.

import plotly.graph_objects as go
import numpy as np

# Generate data
x = np.outer(np.linspace(-2, 2, 30), np.ones(30))
y = x.copy().T
z = np.cos(x ** 2 + y ** 2)

# Create a 3D surface plot
fig = go.Figure(data=[go.Surface(x=x, y=y, z=z)])

# Update layout
fig.update_layout(title='Interactive 3D Surface Plot',
                  autosize=False,
                  width=800,
                  height=800,
                  margin=dict(l=65, r=50, b=65, t=90))

# Show plot
fig.show()

💡 Tip: When creating 3D plots, ensure your data is properly formatted as a meshgrid for surface plots. This will help in accurately representing the relationships between variables.