🎉 AI Engineers: Join our free webinar on how to improve your RAG performance. Secure your spot >

Aporia How to's

How to Select Multiple Columns in a DataFrame?

1 min read
select multiple columns pandas pyspark dataframe

In this short how-to article, we will learn how to select multiple columns in Pandas and PySpark DataFrames.

select multiple columns pandas pyspark dataframe

Pandas

We can select multiple columns by writing them in a list.

cols = ["f2", "f4"]
df[cols]

The iloc method can be used for selecting columns based on their indices. Consider you have a DataFrame with 30 columns and you want to select the first 10. You can perform this task as follows:

# Select the first 10 columns
df.iloc[:,:10]

# Select from the second to fifth
df.iloc[:,2:5]

PySpark

The select function can be used for selecting multiple columns from a PySpark DataFrame.

# first method
df.select("f1", "f2")

# second method
df.select(df.f1, df.f2)

This question was also being asked as:

  • How to choose specific columns in a DataFrame?

People have also asked for:

Green Background

Control All your GenAI Apps in minutes