Skip to content

Commit 5dfc6b1

Browse files
j1642j1642
committed
DOC: Add 3D plots to plot_types gallery
1 parent 4965500 commit 5dfc6b1

File tree

7 files changed

+162
-0
lines changed

7 files changed

+162
-0
lines changed

‎doc/sphinxext/gallery_order.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@
2828
'../plot_types/arrays',
2929
'../plot_types/stats',
3030
'../plot_types/unstructured',
31+
'../plot_types/3D',
3132
]
3233

3334

‎plot_types/3D/README.rst

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
.. _3D_plots:
2+
3+
3D
4+
--
5+
6+
3D plot types.

‎plot_types/3D/scatter3d.py

Lines changed: 36 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,36 @@
1+
"""
2+
==============
3+
3D scatterplot
4+
==============
5+
6+
Demonstration of a basic scatterplot in 3D.
7+
"""
8+
import matplotlib.pyplot as plt
9+
import numpy as np
10+
11+
plt.style.use('_mpl-gallery')
12+
13+
np.random.seed(19680801)
14+
15+
16+
def randrange(n, vmin, vmax):
17+
"""
18+
Helper function to make an array of random numbers having shape (n, )
19+
with each number distributed Uniform(vmin, vmax).
20+
"""
21+
return (vmax - vmin)*np.random.rand(n) + vmin
22+
23+
fig = plt.figure()
24+
ax = fig.add_subplot(projection='3d')
25+
26+
n = 100
27+
28+
# Make data
29+
xs = randrange(n, 23, 32)
30+
ys = randrange(n, 0, 100)
31+
zs = randrange(n, -50, -25)
32+
33+
# Plot
34+
ax.scatter(xs, ys, zs)
35+
36+
plt.show()

‎plot_types/3D/surface3d.py

Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,33 @@
1+
"""
2+
=====================
3+
3D surface (colormap)
4+
=====================
5+
6+
Demonstrates plotting a 3D surface colored with the coolwarm colormap.
7+
The surface is made opaque by using ``antialiased=False``.
8+
9+
Also demonstrates using the `.LinearLocator` and custom formatting for the
10+
z axis tick labels.
11+
"""
12+
import matplotlib.pyplot as plt
13+
from matplotlib import cm
14+
import numpy as np
15+
16+
plt.style.use('_mpl-gallery')
17+
18+
fig, ax = plt.subplots(subplot_kw={"projection": "3d"})
19+
20+
# Make data
21+
X = np.arange(-5, 5, 0.25)
22+
Y = np.arange(-5, 5, 0.25)
23+
X, Y = np.meshgrid(X, Y)
24+
R = np.sqrt(X**2 + Y**2)
25+
Z = np.sin(R)
26+
27+
# Plot the surface
28+
surf = ax.plot_surface(X, Y, Z, cmap=cm.coolwarm,
29+
linewidth=0, antialiased=False)
30+
31+
fig.colorbar(surf, shrink=0.5, aspect=5)
32+
33+
plt.show()

‎plot_types/3D/trisurf3d.py

Lines changed: 30 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,30 @@
1+
"""
2+
======================
3+
Triangular 3D surfaces
4+
======================
5+
6+
Plot a 3D surface with a triangular mesh.
7+
"""
8+
import matplotlib.pyplot as plt
9+
import numpy as np
10+
11+
plt.style.use('_mpl-gallery')
12+
13+
n_radii = 8
14+
n_angles = 36
15+
16+
# Make radii and angles spaces
17+
radii = np.linspace(0.125, 1.0, n_radii)
18+
angles = np.linspace(0, 2*np.pi, n_angles, endpoint=False)[..., np.newaxis]
19+
20+
# Convert polar (radii, angles) coords to cartesian (x, y) coords.
21+
x = np.append(0, (radii*np.cos(angles)).flatten())
22+
y = np.append(0, (radii*np.sin(angles)).flatten())
23+
z = np.sin(-x*y)
24+
25+
# Plot
26+
ax = plt.figure().add_subplot(projection='3d')
27+
28+
ax.plot_trisurf(x, y, z, linewidth=0.2, antialiased=True)
29+
30+
plt.show()

‎plot_types/3D/voxels.py

Lines changed: 34 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,34 @@
1+
"""
2+
==========================
3+
3D voxel / volumetric plot
4+
==========================
5+
6+
Demonstrates plotting 3D volumetric objects with `.Axes3D.voxels`.
7+
"""
8+
import matplotlib.pyplot as plt
9+
import numpy as np
10+
11+
plt.style.use('_mpl-gallery')
12+
13+
# Prepare some coordinates
14+
x, y, z = np.indices((8, 8, 8))
15+
16+
# Draw cuboids in the top left and bottom right corners, and a link between
17+
# them
18+
cube1 = (x < 3) & (y < 3) & (z < 3)
19+
cube2 = (x >= 5) & (y >= 5) & (z >= 5)
20+
link = abs(x - y) + abs(y - z) + abs(z - x) <= 2
21+
22+
# Combine the objects into a single boolean array
23+
voxelarray = cube1 | cube2 | link
24+
25+
colors = np.empty(voxelarray.shape, dtype=object)
26+
colors[link] = 'red'
27+
colors[cube1] = 'blue'
28+
colors[cube2] = 'green'
29+
30+
# Plot
31+
ax = plt.figure().add_subplot(projection='3d')
32+
ax.voxels(voxelarray, facecolors=colors, edgecolor='k')
33+
34+
plt.show()

‎plot_types/3D/wire3d.py

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,22 @@
1+
"""
2+
=================
3+
3D wireframe plot
4+
=================
5+
6+
A very basic demonstration of a wireframe plot.
7+
"""
8+
from mpl_toolkits.mplot3d import axes3d
9+
import matplotlib.pyplot as plt
10+
11+
plt.style.use('_mpl-gallery')
12+
13+
fig = plt.figure()
14+
ax = fig.add_subplot(projection='3d')
15+
16+
# Make data
17+
X, Y, Z = axes3d.get_test_data(0.05)
18+
19+
# Plot
20+
ax.plot_wireframe(X, Y, Z, rstride=10, cstride=10)
21+
22+
plt.show()

0 commit comments

Comments
 (0)
pFad - Phonifier reborn

Pfad - The Proxy pFad of © 2024 Garber Painting. All rights reserved.

Note: This service is not intended for secure transactions such as banking, social media, email, or purchasing. Use at your own risk. We assume no liability whatsoever for broken pages.


Alternative Proxies:

Alternative Proxy

pFad Proxy

pFad v3 Proxy

pFad v4 Proxy