fix .query

LinkedEarth · Jun 1, 2024 · 4647fcd · 4647fcd
1 parent 8423565
commit 4647fcd
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/pyleoclim/core/geoseries.py b/pyleoclim/core/geoseries.py
@@ -457,7 +457,7 @@ def map_neighbors(self, mgs, radius=3000, projection='Orthographic', proj_defaul
             from pylipd.utils.dataset import load_dir
             lipd = load_dir(name='Pages2k')
             df = lipd.get_timeseries_essentials()
-            dfs = df.query("archiveType in ('tree','documents','coral','lake sediment') and paleoData_variableName not in ('year')")
+            dfs = df.query("archiveType in ('Wood','Documents','Coral','Lake sediment') and paleoData_variableName not in ('year')")
             # place in a MultipleGeoSeries object
             ts_list = []
             for _, row in dfs.iterrows():

diff --git a/pyleoclim/core/multiplegeoseries.py b/pyleoclim/core/multiplegeoseries.py
@@ -57,7 +57,7 @@ class MultipleGeoSeries(MultipleSeries):
         from pylipd.utils.dataset import load_dir
         lipd = load_dir(name='Pages2k')
         df = lipd.get_timeseries_essentials()
-        dfs = df.query("archiveType in ('tree','documents','coral','lake sediment')")
+        dfs = df.query("archiveType in ('Wood','Documents','Coral','Lake sediment')")
         # place in a MultipleGeoSeries object
         ts_list = []
         for _, row in dfs.iterrows():
@@ -238,7 +238,7 @@ def map(self, marker='archiveType', hue='archiveType', size=None, cmap=None,
             from pylipd.utils.dataset import load_dir
             lipd = load_dir(name='Pages2k')
             df = lipd.get_timeseries_essentials()
-            dfs = df.query("archiveType in ('tree','documents','coral','lake sediment','borehole')")
+            dfs = df.query("archiveType in ('Wood','Documents','Coral','Lake sediment','Borehole')")
             # place in a MultipleGeoSeries object
             ts_list = []
             for _, row in dfs.iterrows():
@@ -355,7 +355,7 @@ def pca(self, weights=None,missing='fill-em',tol_em=5e-03, max_em_iter=100,**pca
             lipd = load_dir(name='Pages2k') # this loads a small subset of the PAGES 2k database
             lipd_euro = lipd.filter_by_geo_bbox(-20,20,40,80)
             df = lipd_euro.get_timeseries_essentials()
-            dfs = df.query("archiveType in ('tree') & paleoData_variableName not in ('year')")
+            dfs = df.query("archiveType in ('Wood') & paleoData_variableName not in ('year')")
             # place in a MultipleGeoSeries object
             ts_list = []
             for _, row in dfs.iterrows():
@@ -605,7 +605,7 @@ def time_geo_plot(self, figsize=[10, 3],
             from pylipd.utils.dataset import load_dir
             lipd = load_dir(name='Pages2k')
             df = lipd.get_timeseries_essentials()
-            dfs = df.query("archiveType in ('tree','documents','coral','lake sediment')")
+            dfs = df.query("archiveType in ('Wood','Documents','Coral','Lake sediment')")
             # place in a MultipleGeoSeries object
             ts_list = []
             for _, row in dfs.iloc[:5].iterrows():