qml.drawer.tape_mpl

tape_mpl(tape, wire_order=None, show_all_wires=False, decimals=None, **kwargs)[source]

Produces a matplotlib graphic from a tape.

Parameters

tape (QuantumTape) – the operations and measurements to draw

Keyword Arguments
  • wire_order (Sequence[Any]) – the order (from top to bottom) to print the wires of the circuit

  • show_all_wires (bool) – If True, all wires, including empty wires, are printed.

  • decimals (int) – How many decimal points to include when formatting operation parameters. Default None will omit parameters from operation labels.

  • fontsize (float or str) – fontsize for text. Valid strings are {'xx-small', 'x-small', 'small', 'medium', large', 'x-large', 'xx-large'}. Default is 14.

  • wire_options (dict) – matplotlib formatting options for the wire lines

  • label_options (dict) – matplotlib formatting options for the wire labels

  • active_wire_notches (bool) – whether or not to add notches indicating active wires. Defaults to True.

Returns

The key elements for matplotlib’s object oriented interface.

Return type

matplotlib.figure.Figure, matplotlib.axes._axes.Axes

Example:

with qml.tape.QuantumTape() as tape:
    qml.QFT(wires=(0,1,2,3))
    qml.IsingXX(1.234, wires=(0,2))
    qml.Toffoli(wires=(0,1,2))
    qml.CSWAP(wires=(0,2,3))
    qml.RX(1.2345, wires=0)
    qml.CRZ(1.2345, wires=(3,0))
    qml.expval(qml.PauliZ(0))

fig, ax = tape_mpl(tape)
fig.show()
../../_images/default.png

Decimals:

The keyword decimals controls how many decimal points to include when labelling the operations. The default value None omits parameters for brevity.

with qml.tape.QuantumTape() as tape2:
    qml.RX(1.23456, wires=0)
    qml.Rot(1.2345,2.3456, 3.456, wires=0)
    qml.expval(qml.PauliZ(0))

fig, ax = tape_mpl(tape2, decimals=2)
../../_images/decimals1.png

Wires:

The keywords wire_order and show_all_wires control the location of wires from top to bottom.

fig, ax = tape_mpl(tape, wire_order=[3,2,1,0])
../../_images/wire_order1.png

If a wire is in wire_order, but not in the tape, it will be omitted by default. Only by selecting show_all_wires=True will empty wires be diplayed.

fig, ax = tape_mpl(tape, wire_order=["aux"], show_all_wires=True)
../../_images/show_all_wires1.png

Integration with matplotlib:

This function returns matplotlib figure and axes objects. Using these objects, users can perform further customization of the graphic.

fig, ax = tape_mpl(tape)
fig.suptitle("My Circuit", fontsize="xx-large")

options = {'facecolor': "white", 'edgecolor': "#f57e7e", "linewidth": 6, "zorder": -1}
box1 = plt.Rectangle((-0.5, -0.5), width=3.0, height=4.0, **options)
ax.add_patch(box1)

ax.annotate("CSWAP", xy=(3, 2.5), xycoords='data', xytext=(3.8,1.5), textcoords='data',
            arrowprops={'facecolor': 'black'}, fontsize=14)
../../_images/postprocessing1.png

Formatting:

PennyLane has inbuilt styles for controlling the appearance of the circuit drawings. All available styles can be determined by evaluating qml.drawer.available_styles(). Any available string can then be passed to qml.drawer.use_style.

qml.drawer.use_style('black_white')
fig, ax = tape_mpl(tape)
../../_images/black_white_style2.png

You can also control the appearance with matplotlib’s provided tools, see the matplotlib docs . For example, we can customize plt.rcParams:

plt.rcParams['patch.facecolor'] = 'mistyrose'
plt.rcParams['patch.edgecolor'] = 'maroon'
plt.rcParams['text.color'] = 'maroon'
plt.rcParams['font.weight'] = 'bold'
plt.rcParams['patch.linewidth'] = 4
plt.rcParams['patch.force_edgecolor'] = True
plt.rcParams['lines.color'] = 'indigo'
plt.rcParams['lines.linewidth'] = 5
plt.rcParams['figure.facecolor'] = 'ghostwhite'

fig, ax = tape_mpl(tape)
../../_images/rcparams1.png

The wires and wire labels can be manually formatted by passing in dictionaries of keyword-value pairs of matplotlib options. wire_options accepts options for lines, and label_options accepts text options.

fig, ax = tape_mpl(tape, wire_options={'color':'black', 'linewidth': 5},
            label_options={'size': 20})
../../_images/wires_labels1.png