diff options
author | Jeff Zhang <zjffdu@apache.org> | 2015-10-26 09:25:19 +0100 |
---|---|---|
committer | Reynold Xin <rxin@databricks.com> | 2015-10-26 09:25:19 +0100 |
commit | 05c4bdb57947f44924b4fbdd8e4e2101f2f816f5 (patch) | |
tree | 87568238cf01d8d8d02e6fdf4113f8bcd9af8e35 /python/pyspark | |
parent | 07ced43424447699e47106de9ca2fa714756bdeb (diff) | |
download | spark-05c4bdb57947f44924b4fbdd8e4e2101f2f816f5.tar.gz spark-05c4bdb57947f44924b4fbdd8e4e2101f2f816f5.tar.bz2 spark-05c4bdb57947f44924b4fbdd8e4e2101f2f816f5.zip |
[SPARK-11279][PYSPARK] Add DataFrame#toDF in PySpark
Author: Jeff Zhang <zjffdu@apache.org>
Closes #9248 from zjffdu/SPARK-11279.
Diffstat (limited to 'python/pyspark')
-rw-r--r-- | python/pyspark/sql/dataframe.py | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/python/pyspark/sql/dataframe.py b/python/pyspark/sql/dataframe.py index 36fc6e0611..3baff81477 100644 --- a/python/pyspark/sql/dataframe.py +++ b/python/pyspark/sql/dataframe.py @@ -1266,6 +1266,18 @@ class DataFrame(object): raise TypeError("col should be a string or a Column") return DataFrame(jdf, self.sql_ctx) + @ignore_unicode_prefix + def toDF(self, *cols): + """Returns a new class:`DataFrame` that with new specified column names + + :param cols: list of new column names (string) + + >>> df.toDF('f1', 'f2').collect() + [Row(f1=2, f2=u'Alice'), Row(f1=5, f2=u'Bob')] + """ + jdf = self._jdf.toDF(self._jseq(cols)) + return DataFrame(jdf, self.sql_ctx) + @since(1.3) def toPandas(self): """Returns the contents of this :class:`DataFrame` as Pandas ``pandas.DataFrame``. |