From 0fed2c3b03c3e1f783c414bd43d4944d628d2084 Mon Sep 17 00:00:00 2001 From: sg495 Date: Sun, 14 Nov 2021 02:12:19 +0000 Subject: [PATCH] Update state_visualization.py Closes #7263 --- qiskit/visualization/state_visualization.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/qiskit/visualization/state_visualization.py b/qiskit/visualization/state_visualization.py index 2189f400f40..df61487162d 100644 --- a/qiskit/visualization/state_visualization.py +++ b/qiskit/visualization/state_visualization.py @@ -250,7 +250,7 @@ def plot_bloch_multivector( Args: state (Statevector or DensityMatrix or ndarray): an N-qubit quantum state. title (str): a string that represents the plot title - figsize (tuple): Has no effect, here for compatibility only. + figsize (tuple): size of each individual Bloch sphere figure, in inches. reverse_bits (bool): If True, plots qubits following Qiskit's convention [Default:False]. Returns: @@ -289,7 +289,11 @@ def plot_bloch_multivector( _bloch_multivector_data(state)[::-1] if reverse_bits else _bloch_multivector_data(state) ) num = len(bloch_data) - width, height = plt.figaspect(1 / num) + if figsize is not None: + width, height = figsize + width *= num + else: + width, height = plt.figaspect(1 / num) fig = plt.figure(figsize=(width, height)) for i in range(num): pos = num - 1 - i if reverse_bits else i