How to delete columns in PySpark dataframe ?

In this article, we are going to delete columns in Pyspark dataframe. To do this we will be using the drop() function. This function can be used to remove values from the dataframe.

Syntax: dataframe.drop(‘column name’)

Python code to create student dataframe with three columns:

Output:

Actual data in dataframe
+----------+------------+-------+
|student ID|student NAME|college|
+----------+------------+-------+
|         1|      sravan| vignan|
|         2|      ojaswi|   vvit|
|         3|      rohith|   vvit|
|         4|     sridevi| vignan|
|         1|      sravan| vignan|
|         5|     gnanesh|    iit|
+----------+------------+-------+

Example 1: Python program to delete a single column.

Here we are going to delete ‘student ID’ from the dataframe, for this we will use drop().

Output:

+------------+-------+
|student NAME|college|
+------------+-------+
|      sravan| vignan|
|      ojaswi|   vvit|
|      rohith|   vvit|
|     sridevi| vignan|
|      sravan| vignan|
|     gnanesh|    iit|
+------------+-------+

Example 2: Delete multiple columns

Here we will delete multiple columns in a dataframe just passing multiple columns inside the drop() function.

Output:

+-------+
|college|
+-------+
| vignan|
|   vvit|
|   vvit|
| vignan|
| vignan|
|    iit|
+-------+

Example 3: Delete all columns

Here we will delete all the columns in dataframe.

Output:

++
||
++
||
||
||
||
||
||
++

gottumukkalabobby

Follow

Improve

Article Tags :

• Python
• Python-Pyspark

Practice Tags :

• python