Related Articles

# How to Draw 3D Cube using Matplotlib in Python?

• Last Updated : 21 May, 2021

In this article, we will deal with the 3d plots of cubes using matplotlib and Numpy. Cubes are one of the most basic of 3D shapes. A cube is a 3-dimensional solid object bounded by 6 identical square faces. The cube has 6-faces, 12-edges, and 8-corners. All faces are squares of the same size. The total surface area of a cube is the sum of the area of the 6 identical squares.

Matplotlib comes with a wide variety of plots. Graphs help to understand trends, patterns to make correlations. Matplotlib was introduced for two-dimensional plotting. The 3d plot is enabled by importing the mplot3d toolkit., which comes with your standard Matplotlib. After importing, 3D plots can be created by passing the keyword projection=”3d” to any of the regular axes creation functions in Matplotlib.

### Modules Needed

• Matplotlib: It is a plotting library for Python programming it serves as a visualization utility library, Matplotlib is built on NumPy arrays, and designed to work with the broader SciPy stack.
• Numpy: It is a general-purpose array-processing package. It provides a high-performance multidimensional array and matrices along with a large collection of high-level mathematical functions.
• mpl_toolkits: It provides some basic 3d plotting (scatter, surf, line, mesh) tools. It is a collection of helper classes for displaying 3d axes in Matplotlib.

### Approach

Step 1: Import libraries.

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

Step 2: In this step, we are selecting the 3D axis of the dimension X =5, Y=5, Z=5, and in np.ones() we are passing the dimensions of the cube.

```# Create axis
axes = [5, 5, 5]

# Create Data
data = np.ones(axes)```

Step 3: In this step, we are selecting color opacity as alpha = 0.9 ( vary from 0.0 – 1.0 ). In the next step, we are passing the dimension of axes( i.e 5, 5, 5) + number of faces for the cube ( i.e 0-4 ) in np.empty() function after that we are passing color combination and opacity for each face of the cube.

```# controll Tranperency
alpha = 0.9

# control colour
colors = np.empty(axes + )

colors = [1, 0, 0, alpha] # red
colors = [0, 1, 0, alpha] # green
colors = [0, 0, 1, alpha] # blue
colors = [1, 1, 0, alpha] # yellow
colors = [1, 1, 1, alpha] # grey```

Step 4: In this step, we used figure() function of the matplotlib library which is used to create a new figure, After that, we used add_subplot() method to add an Axes to the figure as 3-Dimensional(i.e Projection = ‘3d’) part of a subplot arrangement. It has 3 arguments.

• The number of rows in the grid,
• The number of columns in the grid and,
• The position at which the new subplot must be placed.

It is to be noted that fig.add_subplot(1, 1, 1) is equivalent to fig.add_subplot(111).

```# Plot figure
fig = plt.figure()

Step 5: And in this last step Voxels is used for customizations of the size, position, and grid color. The proper syntax is provided above.

```# Voxels is used to customizations of the
# sizes, positions and colors.
ax.voxels(data, facecolors=colors, edgecolors='grey')```

Example 1: Simple cube of one color.

Here changing colors[ : ], means we are selecting all the array elements as one color (i.e red) and to remove the grid we deleted ‘edgecolor’ parameter from voxels method to have simply one color cube.

## Python3

 `# Import libraries``import` `matplotlib.pyplot as plt``from` `mpl_toolkits.mplot3d ``import` `Axes3D``import` `numpy as np`` ` ` ` `# Create axis``axes ``=` `[``5``, ``5``, ``5``]`` ` `# Create Data``data ``=` `np.ones(axes, dtype``=``np.``bool``)`` ` `# Controll Tranperency``alpha ``=` `0.9`` ` `# Control colour``colors ``=` `np.empty(axes ``+` `[``4``], dtype``=``np.float32)`` ` `colors[:] ``=` `[``1``, ``0``, ``0``, alpha]  ``# red`` ` `# Plot figure``fig ``=` `plt.figure()``ax ``=` `fig.add_subplot(``111``, projection``=``'3d'``)`` ` `# Voxels is used to customizations of the``# sizes, positions and colors.``ax.voxels(data, facecolors``=``colors)`

Output: Example 2:

Cube with Grid and different color

## Python3

 `# Import libraries``import` `matplotlib.pyplot as plt``from` `mpl_toolkits.mplot3d ``import` `Axes3D``import` `numpy as np`` ` ` ` `# Create axis``axes ``=` `[``5``, ``5``, ``5``]`` ` `# Create Data``data ``=` `np.ones(axes, dtype``=``np.``bool``)`` ` `# Controll Tranperency``alpha ``=` `0.9`` ` `# Control colour``colors ``=` `np.empty(axes ``+` `[``4``], dtype``=``np.float32)`` ` `colors[``0``] ``=` `[``1``, ``0``, ``0``, alpha]  ``# red``colors[``1``] ``=` `[``0``, ``1``, ``0``, alpha]  ``# green``colors[``2``] ``=` `[``0``, ``0``, ``1``, alpha]  ``# blue``colors[``3``] ``=` `[``1``, ``1``, ``0``, alpha]  ``# yellow``colors[``4``] ``=` `[``1``, ``1``, ``1``, alpha]  ``# grey`` ` `# Plot figure``fig ``=` `plt.figure()``ax ``=` `fig.add_subplot(``111``, projection``=``'3d'``)`` ` `# Voxels is used to customizations of``# the sizes, positions and colors.``ax.voxels(data, facecolors``=``colors, edgecolors``=``'grey'``)`

Output: Example 3: Faced cube along Y – axes

In this example, we will add one more line of code view_init( ) to change the axis view as we want. The view_init() can be used to change the axes view programmatically. Here we are using elev=100 and  azim=0.

Syntax: view_init(elev, azim)

Parameters:

• ‘elev’ stores the elevation angle in the z plane.
• ‘azim’ stores the azimuth angle in the x,y plane.

## Python3

 `import` `matplotlib.pyplot as plt``from` `mpl_toolkits.mplot3d ``import` `Axes3D``import` `numpy as np`` ` `axes ``=` `[``5``, ``5``, ``5``]``data ``=` `np.ones(axes, dtype``=``np.``bool``)`` ` `colors ``=` `np.empty(axes ``+` `[``4``], dtype``=``np.float32)`` ` `# Controll Tranperency``alpha ``=` `.``7`` ` `# Control colors``colors[``0``] ``=` `[``1``, ``0``, ``0``, alpha]``colors[``1``] ``=` `[``0``, ``1``, ``0``, alpha]``colors[``2``] ``=` `[``0``, ``0``, ``1``, alpha]``colors[``3``] ``=` `[``1``, ``1``, ``0``, alpha]``colors[``4``] ``=` `[``0``, ``1``, ``1``, alpha]`` ` `# set all internal colors to ``# black with alpha=1``colors[``1``:``-``1``, ``1``:``-``1``, ``1``:``-``1``, ``0``:``3``] ``=` `0``colors[``1``:``-``1``, ``1``:``-``1``, ``1``:``-``1``, ``3``] ``=` `1`` ` `# Plot figure``fig ``=` `plt.figure()``ax ``=` `fig.add_subplot(``111``, projection``=``'3d'``)`` ` `# Control number of slice``data[``-``1``] ``=` `True``data[``-``2``] ``=` `False``data[``-``3``] ``=` `False``data[``-``4``] ``=` `False``data[``-``5``] ``=` `True`` ` `# Voxels is used to customizations of ``# the sizes, positions and colors.``ax.voxels(data, facecolors``=``colors, edgecolors``=``'pink'``)`` ` `# it can be used to change the axes view``ax.view_init(``100``, ``0``)`

Output: Example 4: Faced cube along X-axes

The view_init() can be used to change the axes view programmatically. Here we are using elev=100 and  azim=90.

## Python3

 `import` `matplotlib.pyplot as plt``from` `mpl_toolkits.mplot3d ``import` `Axes3D``import` `numpy as np`` ` `axes ``=` `[``5``, ``5``, ``5``]``data ``=` `np.ones(axes, dtype``=``np.``bool``)`` ` `colors ``=` `np.empty(axes ``+` `[``4``], dtype``=``np.float32)`` ` `# Controll Tranperency``alpha ``=` `.``7`` ` `# Control colors``colors[``0``] ``=` `[``1``, ``0``, ``0``, alpha]``colors[``1``] ``=` `[``0``, ``1``, ``0``, alpha]``colors[``2``] ``=` `[``0``, ``0``, ``1``, alpha]``colors[``3``] ``=` `[``1``, ``1``, ``0``, alpha]``colors[``4``] ``=` `[``0``, ``1``, ``1``, alpha]`` ` `# set all internal colors to ``# black with alpha=1``colors[``1``:``-``1``, ``1``:``-``1``, ``1``:``-``1``, ``0``:``3``] ``=` `0``colors[``1``:``-``1``, ``1``:``-``1``, ``1``:``-``1``, ``3``] ``=` `1`` ` `# Plot figure``fig ``=` `plt.figure()``ax ``=` `fig.add_subplot(``111``, projection``=``'3d'``)`` ` `# Control number of slice``data[``-``1``] ``=` `1``data[``-``2``] ``=` `False``data[``-``3``] ``=` `False``data[``-``4``] ``=` `False``data[``-``5``] ``=` `True`` ` `# Voxels is used to customizations ``# of the sizes, positions and colors.``ax.voxels(data, facecolors``=``colors, edgecolors``=``'pink'``)`` ` `# it can be used to change the axes view``ax.view_init(``100``, ``90``)`

Output: Attention geek! Strengthen your foundations with the Python Programming Foundation Course and learn the basics.

To begin with, your interview preparations Enhance your Data Structures concepts with the Python DS Course. And to begin with your Machine Learning Journey, join the Machine Learning – Basic Level Course

My Personal Notes arrow_drop_up