Merge pull request #245 from ToFuProject/Issue237_PLotArray

[#237] Fixed

Author: Didou09

datastock/_class1.py

@@ -470,7 +470,6 @@ def extract(
             return_keys: returns the value of keys
 
         """
-
         return _class1_compute._extract_instance(
             self,
             keys=keys,

datastock/_class1_compute.py

@@ -735,7 +735,6 @@ def _extract_instance(
     # -----------------
     # select
     # -----------------
-
     ldata, lref = _extract_select(
         coll=coll,
         keys=keys,
@@ -755,7 +754,6 @@ def _extract_instance(
         else:
             coll2 = coll.__class__()
 
-
     # ------------------------
     # Populate
     # ------------------------

datastock/_plot_as_array.py

@@ -91,15 +91,18 @@ def plot_as_array(
 
     # check key, inplace flag and extract sub-collection
     lk = [kk for kk in [keyX, keyY, keyZ, keyU] if kk is not None]
-    coll2, key = coll.extract(
+    coll2, keys = coll.extract(
         [key] + lk,
         inc_monot=False,
         inc_vectors=False,
         inc_allrefs=False,
         return_keys=True,
         inplace=inplace,
     )
-    key = [kk for kk in key if kk not in lk][0]
+    if key is None:
+        key = [kk for kk in key if kk not in lk]
+        lndim = [coll2.ddata[kk]['data'].ndim for kk in key]
+        key = key[np.argmax(lndim)]
     ndim = coll2.ddata[key]['data'].ndim
 
     # --------------
@@ -603,7 +606,6 @@ def _check(
         else:
             dvminmax2[k1]['min'] = dvminmax[kk]['min']
 
-
         if dvminmax is None or dvminmax.get(kk, {}).get('max') is None:
             dvminmax2[k1]['max'] = nanmax + margin
         else: