Accessing Specific Columns

How to get a certain column in a DataFrame? Put column name (as string, in quotes) between square brackets and place it right to variable with dataframe. Another way is to refer to column like to attribute - put column name (without quotes) right to dot symbol placed right to variable with dataframe (it works only for column that has only one word in name).


              123456789
            
# Importing library
import pandas as pd

# Reading csv file
df = pd.read_csv('https://codefinity-content-media.s3.eu-west-1.amazonaws.com/67798cef-5e7c-4fbc-af7d-ae96b4443c0a/audi.csv')
# Get 'year' column
print(df['year'])
# Get 'model' column
print(df.model)

If you want to get several columns at once, you need to pass list of columns you want to get. The second approach described above doesn't work here.


              1234567
            
# Importing library
import pandas as pd

# Reading csv file
df = pd.read_csv('https://codefinity-content-media.s3.eu-west-1.amazonaws.com/67798cef-5e7c-4fbc-af7d-ae96b4443c0a/audi.csv')
# Get columns 'model' and 'year'
print(df[['model', 'year']])

Pay attention to square brackets. Syntax df(['model', 'year']) will not work!

Tudo estava claro?

Seção 3. Capítulo 1