From 29c36f2bce3d963adcb54159e44c5dfcab916d47 Mon Sep 17 00:00:00 2001
From: Kunal Gosar <gosarkunal9@gmail.com>
Date: Mon, 23 Apr 2018 13:13:14 -0700
Subject: [PATCH] [DataFrame] Fix for __getitem__ string indexing (#1939)

* edge case fixes for __getitem__

* Enable None indexing
---
 python/ray/dataframe/dataframe.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/ray/dataframe/dataframe.py b/python/ray/dataframe/dataframe.py
index a069d1c24..3253cd0a1 100644
--- a/python/ray/dataframe/dataframe.py
+++ b/python/ray/dataframe/dataframe.py
@@ -3266,7 +3266,7 @@ class DataFrame(object):
                              columns=columns,
                              index=index)
         else:
-            columns = self.columns[key]
+            columns = self._col_metadata[key].index
 
             indices_for_rows = [self.columns.index(new_col)
                                 for new_col in columns]