Python - How to select a column from a Pandas DataFrame

PythonServer Side ProgrammingProgramming

To select a column from a DataFrame, just fetch it using square brackets. Mention the column to select in the brackets and that’s it, for example

dataFrame[‘ColumnName’]

At first, import the required library −

import pandas as pd

Now, create a DataFrame. We have two columns in it −

dataFrame = pd.DataFrame(
   {
      "Car": ['BMW', 'Lexus', 'Audi', 'Mustang', 'Bentley', 'Jaguar'],
      "Units": [100, 150, 110, 80, 110, 90]
   }
)

To select only a single column, mention the column name using the square bracket as shown below. Here, our column name is ‘Car’ −

dataFrame ['Car']

Example

Following is the code −

import pandas as pd

# Create DataFrame
dataFrame = pd.DataFrame(
   {
      "Car": ['BMW', 'Lexus', 'Audi', 'Mustang', 'Bentley', 'Jaguar'],
      "Units": [100, 150, 110, 80, 110, 90]
   }
)

print"DataFrame ...\n",dataFrame

# selecting a column
print"\nSelecting and displaying only a single column = \n",dataFrame ['Car']

Output

This will produce the following output −

DataFrame ...
       Car   Units
0      BMW     100
1    Lexus     150
2     Audi     110
3  Mustang      80
4  Bentley     110
5   Jaguar      90

Selecting and displaying only a single column =
0      BMW
1    Lexus
2     Audi
3  Mustang
4  Bentley
5   Jaguar
Name: Car, dtype: object
raja
Published on 13-Sep-2021 09:58:48
Advertisements