From 2a7052611b1f0d66042a91d77c0048231b24c3d6 Mon Sep 17 00:00:00 2001
From: Wenchen Fan <cloud0fan@outlook.com>
Date: Wed, 29 Apr 2015 21:05:11 +0800
Subject: [PATCH] use _bin_op in dataframe.py

---
 python/pyspark/sql/dataframe.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/python/pyspark/sql/dataframe.py b/python/pyspark/sql/dataframe.py
index c32527ada73fe..fd10464adb908 100644
--- 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("__"):