Skip to content

Commit 39f3fc0

Browse files
committed
Match the outputs of scala api & python api
1 parent e4417c8 commit 39f3fc0

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

python/pyspark/sql/dataframe.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -203,7 +203,7 @@ def printSchema(self):
203203
|-- name: string (nullable = true)
204204
<BLANKLINE>
205205
"""
206-
self._jdf.printSchema()
206+
print(self._jdf.schema().treeString())
207207

208208
@since(1.3)
209209
def explain(self, extended=False):
@@ -212,6 +212,7 @@ def explain(self, extended=False):
212212
:param extended: boolean, default ``False``. If ``False``, prints only the physical plan.
213213
214214
>>> df.explain()
215+
== Physical Plan ==
215216
Scan PhysicalRDD[age#0,name#1]
216217
217218
>>> df.explain(True)
@@ -224,7 +225,10 @@ def explain(self, extended=False):
224225
== Physical Plan ==
225226
...
226227
"""
227-
self._jdf.explain(extended)
228+
if extended:
229+
print(self._jdf.queryExecution().toString())
230+
else:
231+
print(self._jdf.queryExecution().simpleString())
228232

229233
@since(1.3)
230234
def isLocal(self):
@@ -250,7 +254,7 @@ def show(self, n=20, truncate=True):
250254
| 5| Bob|
251255
+---+-----+
252256
"""
253-
self._jdf.show(n, truncate)
257+
print(self._jdf.showString(n, truncate))
254258

255259
def __repr__(self):
256260
return "DataFrame[%s]" % (", ".join("%s: %s" % c for c in self.dtypes))

0 commit comments

Comments
 (0)