Merge pull request #7299 from eendebakpt/plot_dataset_fix

Fix bug in plot_data when a single parameter is selected

Author: jenshnielsen

src/qcodes/dataset/plotting.py

@@ -268,13 +268,15 @@ def plot_dataset(
 
         log.debug(f"Plotting data for {parameters}.")
 
+        indices_to_remove = []
         for i, data in enumerate(alldata):
             if len(data) == 2:  # 1D PLOTTING
                 if data[1]["name"] not in parameters:
-                    alldata.pop(i)
+                    indices_to_remove.append(i)
             elif len(data) == 3:  # 2D PLOTTING
                 if data[2]["name"] not in parameters:
-                    alldata.pop(i)
+                    indices_to_remove.append(i)
+        alldata = [d for (i, d) in enumerate(alldata) if i not in indices_to_remove]
 
     for data, ax, colorbar in zip(alldata, axeslist, colorbars):
         if len(data) == 2:  # 1D PLOTTING