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("__"):