Fix subset-cols issue with downcasting

alanmarazzi · Aug 12, 2019 · cafd377 · cafd377
1 parent e4ca714
commit cafd377
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/project.clj b/project.clj
@@ -1,4 +1,4 @@
-(defproject panthera "0.1-alpha.15"
+(defproject panthera "0.1-alpha.16"
   :description "Data Frames in Clojure (with Pandas) + NumPy"
   :url "https://github.com/alanmarazzi/panthera"
   :scm {:name "git" :url "https://github.com/alanmarazzi/panthera"}

diff --git a/src/panthera/pandas/generics.clj b/src/panthera/pandas/generics.clj
@@ -626,7 +626,10 @@
   ;2  6  8
   ```"
   [df & colnames]
-  (py/get-item df colnames))
+  (let [cols (if (= 1 (count colnames))
+               (first colnames)
+               (apply py/->py-list [colnames]))]
+    (py/get-item df cols)))
 
 (defn n-largest
   "Return the first `:n` values ordered by `:columns` in descending order.