How to select multiple columns in a pandas dataframe

Python is a great language for doing data analysis, primarily because of the fantastic ecosystem of data-centric Python packages. Pandas is one of those packages and makes importing and analyzing data much easier.

Let’s discuss all different ways of selecting multiple columns in a pandas DataFrame.

Method #1: Basic Method



Given a dictionary which contains Employee entity as keys and list of those entity as values.

filter_none

edit
close

play_arrow

link
brightness_4
code

# Import pandas package
import pandas as pd
  
# Define a dictionary containing employee data
data = {'Name':['Jai', 'Princi', 'Gaurav', 'Anuj'],
        'Age':[27, 24, 22, 32],
        'Address':['Delhi', 'Kanpur', 'Allahabad', 'Kannauj'],
        'Qualification':['Msc', 'MA', 'MCA', 'Phd']}
  
# Convert the dictionary into DataFrame 
df = pd.DataFrame(data)
  
# select two columns
df[['Name', 'Qualification']]

chevron_right






                        filter_none










Output:




Select Second to fourth column.













                                    filter_none




                                    edit

                                    close




                                    play_arrow




                                    link

                                    brightness_4

                                    code
                                




















# Import pandas package 


import pandas as pd 


  


# Define a dictionary containing employee data 


data = {'Name':['Jai', 'Princi', 'Gaurav', 'Anuj'], 


        'Age':[27, 24, 22, 32], 


        'Address':['Delhi', 'Kanpur', 'Allahabad', 'Kannauj'], 


        'Qualification':['Msc', 'MA', 'MCA', 'Phd']} 


  


# Convert the dictionary into DataFrame  


df = pd.DataFrame(data) 


  


# select all rows  


# and second to fourth column 


df[df.columns[1:4]] 



















                        chevron_right







                        filter_none










Output:




 


Method #2: Using loc[]


Example 1: Select two columns













                                    filter_none




                                    edit

                                    close




                                    play_arrow




                                    link

                                    brightness_4

                                    code
                                




















# Import pandas package 


import pandas as pd 


  


# Define a dictionary containing employee data 


data = {'Name':['Jai', 'Princi', 'Gaurav', 'Anuj'], 


        'Age':[27, 24, 22, 32], 


        'Address':['Delhi', 'Kanpur', 'Allahabad', 'Kannauj'], 


        'Qualification':['Msc', 'MA', 'MCA', 'Phd']} 


  


# Convert the dictionary into DataFrame  


df = pd.DataFrame(data) 


  


# select three rows and two columns 


df.loc[1:3, ['Name', 'Qualification']] 



















                        chevron_right







                        filter_none










Output:




Example 2: Select one to another columns. In our case we select column name “Name” to “Address”.













                                    filter_none




                                    edit

                                    close




                                    play_arrow




                                    link

                                    brightness_4

                                    code
                                




















# Import pandas package 


import pandas as pd 


  


# Define a dictionary containing employee data 


data = {'Name':['Jai', 'Princi', 'Gaurav', 'Anuj'], 


        'Age':[27, 24, 22, 32], 


        'Address':['Delhi', 'Kanpur', 'Allahabad', 'Kannauj'], 


        'Qualification':['Msc', 'MA', 'MCA', 'Phd']} 


  


# Convert the dictionary into DataFrame  


df = pd.DataFrame(data) 


  


# select two rows and  


# column "name" to "Address" 


# Means total three columns 


df.loc[0:1, 'Name':'Address'] 



















                        chevron_right







                        filter_none










Output:




        
          
          
          
        

            

Example 3: First filtering rows and selecting columns by label format and then Select all columns.













                                    filter_none




                                    edit

                                    close




                                    play_arrow




                                    link

                                    brightness_4

                                    code
                                




















# Import pandas package 


import pandas as pd 


  


# Define a dictionary containing employee data 


data = {'Name':['Jai', 'Princi', 'Gaurav', 'Anuj'], 


        'Age':[27, 24, 22, 32], 


        'Address':['Delhi', 'Kanpur', 'Allahabad', 'Kannauj'], 


        'Qualification':['Msc', 'MA', 'MCA', 'Phd'] 


       } 


# Convert the dictionary into DataFrame  


df = pd.DataFrame(data) 


  


# .loc DataFrame method 


# filtering rows and selecting columns by label 


# format 


# df.loc[rows, columns] 


# row 1, all columns 


df.loc[0, :] 



















                        chevron_right







                        filter_none










Output:




 


Method #3: Using iloc[]


Example 1:  Select first two column.













                                    filter_none




                                    edit

                                    close




                                    play_arrow




                                    link

                                    brightness_4

                                    code
                                




















# Import pandas package 


import pandas as pd 


  


# Define a dictionary containing employee data 


data = {'Name':['Jai', 'Princi', 'Gaurav', 'Anuj'], 


        'Age':[27, 24, 22, 32], 


        'Address':['Delhi', 'Kanpur', 'Allahabad', 'Kannauj'], 


        'Qualification':['Msc', 'MA', 'MCA', 'Phd']} 


  


# Convert the dictionary into DataFrame  


df = pd.DataFrame(data) 


  


# Remember that Python does not 


# slice inclusive of the ending index. 


# select all rows  


# select first two column 


df.iloc[:, 0:2]  



















                        chevron_right







                        filter_none










Output:




Example 2: Select all or some columns, one to another using .iloc.













                                    filter_none




                                    edit

                                    close




                                    play_arrow




                                    link

                                    brightness_4

                                    code
                                




















# Import pandas package 


import pandas as pd 


  


# Define a dictionary containing employee data 


data = {'Name':['Jai', 'Princi', 'Gaurav', 'Anuj'], 


        'Age':[27, 24, 22, 32], 


        'Address':['Delhi', 'Kanpur', 'Allahabad', 'Kannauj'], 


        'Qualification':['Msc', 'MA', 'MCA', 'Phd']} 


  


# Convert the dictionary into DataFrame  


df = pd.DataFrame(data) 


  


# iloc[row slicing, column slicing] 


df.iloc [0:2, 1:3] 



















                        chevron_right







                        filter_none










Output:




 


Method #4: Using .ix


Select all or some columns, one to another using .ix.













                                    filter_none




                                    edit

                                    close




                                    play_arrow




                                    link

                                    brightness_4

                                    code
                                




















# Import pandas package 


import pandas as pd 


  


# Define a dictionary containing employee data 


data = {'Name':['Jai', 'Princi', 'Gaurav', 'Anuj'], 


        'Age':[27, 24, 22, 32], 


        'Address':['Delhi', 'Kanpur', 'Allahabad', 'Kannauj'], 


        'Qualification':['Msc', 'MA', 'MCA', 'Phd']} 


  


# Convert the dictionary into DataFrame  


df = pd.DataFrame(data) 


  


# select all rows and 0 to 2 columns  


print(df.ix[:, 0:2]) 



















                        chevron_right







                        filter_none










Output:





          
          
          
          

            

    

        My Personal Notes
        arrow_drop_up
    

    

        
        

            
            
        

    


Recommended Posts:
Rajput-Ji
Strategy Path planning and Destination matters in success No need to worry about in between temporary failures
If you like GeeksforGeeks and would like to contribute, you can also write an article using contribute.geeksforgeeks.org or mail your article to contribute@geeksforgeeks.org. See your article appearing on the GeeksforGeeks main page and help other Geeks.
Please Improve this article if you find anything incorrect by clicking on  the "Improve Article" button below.