use _bin_op in dataframe.py

apache · May 8, 2015 · 2a70526 · 2a70526
1 parent 6bf72bc
commit 2a70526
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/pyspark/sql/dataframe.py b/python/pyspark/sql/dataframe.py
@@ -1275,7 +1275,7 @@ def __init__(self, jc):
 
     # container operators
     __contains__ = _bin_op("contains")
-    __getitem__ = getItem
+    __getitem__ = _bin_op("getItem")
 
     # bitwise operators
     bitwiseOR = _bin_op("bitwiseOR")
@@ -1300,7 +1300,7 @@ def getItem(self, key):
         |   1| value|
         +----+------+
         """
-        return Column(self._jc.getItem(key))
+        return self[key]
 
     def getField(self, name):
         """An expression that gets a field by name in a StructField.
@@ -1320,7 +1320,7 @@ def getField(self, name):
         |  1|
         +---+
         """
-        return Column(self._jc.getField(name))
+        return self[name]
 
     def __getattr__(self, item):
         if item.startswith("__"):