Swipe to show menu

Working with Columns

When working with a DataFrame, you can access each column individually.

python

To clarify this syntax:

Start by writing the name of the DataFrame you're working with;
Next, place the column name you want to access inside square brackets . Remember to enclose the column name in quotation marks.

Alternatively, you can use dot notation to access a column if the column name:

Is a valid Python identifier (e.g., no spaces, special characters, or starting with a number);
Does not conflict with an existing pandas attribute or method name.

python


              12345678910111213
            
import pandas as pd

countries_data = {'country' : ['Thailand', 'Philippines', 'Monaco', 'Malta', 'Sweden', 'Paraguay', 'Latvia'],
          'continent' : ['Asia', 'Asia', 'Europe', 'Europe', 'Europe', 'South America', 'Europe'],
          'capital':['Bangkok', 'Manila', 'Monaco', 'Valletta', 'Stockholm', 'Asuncion', 'Riga']}
countries = pd.DataFrame(countries_data)

capitals = countries['capital']

# Second option
# capitals = countries.capital

print(capitals)

Executing this code will display just the column containing capital cities, rather than the entire DataFrame.

You can also access multiple columns like this:

python

Compared to accessing a single column, there is only one difference. This time, you'll need to put the list of column names inside an additional set of square brackets — meaning you'll use double square brackets.


              12345678
            
import pandas as pd

countries_data = {'country' : ['Thailand', 'Philippines', 'Monaco', 'Malta', 'Sweden', 'Paraguay', 'Latvia'],
          'continent' : ['Asia', 'Asia', 'Europe', 'Europe', 'Europe', 'South America', 'Europe'],
          'capital':['Bangkok', 'Manila', 'Monaco', 'Valletta', 'Stockholm', 'Asuncion', 'Riga']}
countries = pd.DataFrame(countries_data)
columns = countries[['country', 'capital']]
print(columns)

Task

Swipe to start coding

You are given a DataFrame named audi_cars.

Retrieve the data for the columns 'model', 'year', and 'price' and store the result in the columns variable.

Solution

Switch to desktop for real-world practiceContinue from where you are using one of the options below

Everything was clear?

Thanks for your feedback!

Section 1. Chapter 11

Ask AI

Ask anything or try one of the suggested questions to begin our chat

Working with Columns

When working with a DataFrame, you can access each column individually.

python

To clarify this syntax:

Start by writing the name of the DataFrame you're working with;
Next, place the column name you want to access inside square brackets . Remember to enclose the column name in quotation marks.

Alternatively, you can use dot notation to access a column if the column name:

Is a valid Python identifier (e.g., no spaces, special characters, or starting with a number);
Does not conflict with an existing pandas attribute or method name.

python


              12345678910111213
            
import pandas as pd

countries_data = {'country' : ['Thailand', 'Philippines', 'Monaco', 'Malta', 'Sweden', 'Paraguay', 'Latvia'],
          'continent' : ['Asia', 'Asia', 'Europe', 'Europe', 'Europe', 'South America', 'Europe'],
          'capital':['Bangkok', 'Manila', 'Monaco', 'Valletta', 'Stockholm', 'Asuncion', 'Riga']}
countries = pd.DataFrame(countries_data)

capitals = countries['capital']

# Second option
# capitals = countries.capital

print(capitals)

Executing this code will display just the column containing capital cities, rather than the entire DataFrame.

You can also access multiple columns like this:

python


              12345678
            
import pandas as pd

countries_data = {'country' : ['Thailand', 'Philippines', 'Monaco', 'Malta', 'Sweden', 'Paraguay', 'Latvia'],
          'continent' : ['Asia', 'Asia', 'Europe', 'Europe', 'Europe', 'South America', 'Europe'],
          'capital':['Bangkok', 'Manila', 'Monaco', 'Valletta', 'Stockholm', 'Asuncion', 'Riga']}
countries = pd.DataFrame(countries_data)
columns = countries[['country', 'capital']]
print(columns)

Task

Swipe to start coding

You are given a DataFrame named audi_cars.

Retrieve the data for the columns 'model', 'year', and 'price' and store the result in the columns variable.

Solution

Switch to desktop for real-world practiceContinue from where you are using one of the options below

Everything was clear?

Thanks for your feedback!

Section 1. Chapter 11

Switch to desktop for real-world practiceContinue from where you are using one of the options below