BasePlot.add_item_with_z_offset: fix DeprecationWarning: Conversion of an array with ndim > 0 to a scalar

PierreRaybaut · PierreRaybaut · commit abe42cc70d71 · 2024-09-19T15:21:21.000+02:00
diff --git a/plotpy/plot/base.py b/plotpy/plot/base.py
@@ -1322,7 +1322,7 @@ def add_item_with_z_offset(self, item: itf.IBasePlotItem, zoffset: int) -> None:
         zlist = sorted(
             [_it.z() for _it in self.items if _it.z() >= zoffset] + [zoffset - 1]
         )
-        dzlist = np.argwhere(np.diff(zlist) > 1)
+        dzlist = np.argwhere(np.diff(zlist) > 1).flatten()
         if len(dzlist) == 0:
             z = max(zlist) + 1
         else:

Original file line number	Diff line number	Diff line change
`@@ -1322,7 +1322,7 @@ def add_item_with_z_offset(self, item: itf.IBasePlotItem, zoffset: int) -> None:`
`1322`	`1322`	`zlist = sorted(`
`1323`	`1323`	`[_it.z() for _it in self.items if _it.z() >= zoffset] + [zoffset - 1]`
`1324`	`1324`	`)`
`1325`		`- dzlist = np.argwhere(np.diff(zlist) > 1)`
	`1325`	`+ dzlist = np.argwhere(np.diff(zlist) > 1).flatten()`
`1326`	`1326`	`if len(dzlist) == 0:`
`1327`	`1327`	`z = max(zlist) + 1`
`1328`	`1328`	`else:`