Tables (pandas)#

Last updated: 2024-03-10 14:50:15

Introduction#

Tables are a fundamental data structure, common to many software interfaces, such as spreadsheet programs, relational databases, statistical software, etc. In general, a table is a two-dimensional structure where:

  • Columns represent variables

  • Rows represent observations

Accordingly, the data in a given table column is usually of the same type. For example, a text column may be used to store product names, and a numeric column may be used for storing product prices. Conversely, the data in a given row may comprise mixed types, such as the product name (text) and the price (numeric) of a specific product.

In this chapter we introduce methods for working with tables in Python, through a popular third-party package named pandas, introducing two table-related data structures: Series and DataFrame. As we will see, the latter data structures are closely related to numpy, which was covered in the previous chapter (see Arrays (numpy)), since the data in each table column is actually stored as an “extended” one-dimensional numpy array. We are going to cover standard table-related operations, such as:

In the next chapter, we are going to discuss more advanced operations with tables, such as table aggregation and table joins (see Table reshaping and joins). Later on, we are going to learn about the geopandas package for working with vector layers, which is an extension of pandas (see Vector layers (geopandas) and Geometric operations).

What is pandas?#

pandas is a popular Python package which provides a flexible set of tools for working with tabular data. Since tables are fundamental in data analysis, pandas is one of the most important packages for data analysis in Python. Moreover, as we will see later on in the book (see Vector layers (geopandas)), a vector layer is represented by an “extended” table—a table that has a “geometry” column. Accordingly, the package for working with vector layers, named geopandas, is an extension of pandas.

The pandas package defines two fundamental data structures:

  • Series—A one-dimensional structure, representing a table column

  • DataFrame—A two-dimensional structure, represent a table

As we will see shortly, pandas is essentially an extension of numpy. A pandas table (a data structure named DataFrame) is a collection of Series representing the table columns, whereas Series are actually nothing more than labelled numpy arrays. To repeat, these two data structures therefore form a hierarchy: each column in a DataFrame is a Series.

Note

For those familiar with the R programming language, the term DataFrame will sound familiar, resembling R’s own data structure for representing tables (the data.frame). Indeed, pandas borrowed many ideas from the R language. A pairwise comparison of pandas vs. R functionality can be found in the pandas documentation Comparison with R / R libraries.

Table from scratch#

Creating a Series#

To understand how a DataFrame is structured, we will experiment with creating one from scratch. First we need to load the pandas package, as well as numpy which we use in some of the examples:

import numpy as np
import pandas as pd

A Series can be created, from a list, or from an ndarray, using pd.Series. For example, the following expression creates a Series from a list containing three strings, the names of three railway stations in southern Israel:

name = pd.Series(['Beer-Sheva Center', 'Beer-Sheva University', 'Dimona'])
name
0        Beer-Sheva Center
1    Beer-Sheva University
2                   Dimona
dtype: object

Note

In pandas, strings are always stored as 'object' arrays (Series), unlike numpy which has specialized native data types for strings.

The following expression creates another Series, named lines, containing the number of railway lines going through the latter stations. This time, we create the Series from an ndarray:

lines = pd.Series(np.array([4, 5, 1]))
lines
0    4
1    5
2    1
dtype: int64

A Series object is essentially a numpy array, combined with indices. Here is how we can access each of these components, separately:

lines.to_numpy()  ## Series values, as 'ndarray'
array([4, 5, 1])
lines.index       ## Series index
RangeIndex(start=0, stop=3, step=1)

As you can see, Series values comprise a numpy array, which we are already familiar with. Series indices, however, comprise a special type of object used for indexing in pandas:

type(name.index)
pandas.core.indexes.range.RangeIndex

By default, the index is just a sequence of consecutive integers (such as in names and lines). When necessary, we can define any other sequence of values as a Series index (see Modifying the index), then use it to access specific values (see Subsetting Series).

Index objects are similar to arrays in many ways. For example, they have .shape and .dtype attributes, just like arrays do:

lines.index.shape
(3,)
lines.index.dtype
dtype('int64')

Creating a DataFrame#

While a Series can be thought of as an extended one-dimensional array, a DataFrame can be thought of as a collection of Series comprising table columns. Unlike a two-dimensional numpy array, a DataFrame can be composed of different types of values (among the columns). Consequently:

  • a DataFrame column (i.e., a Series) contains values of the same type, usually representing a particular variable

  • a DataFrame row may contain values of different types, usually representing a specific observation

Since a DataFrame is two-dimensional, it has two sets of idices, row and column indices, just like a two-dimensional array. It is important to note that all Series that comprise the columns of a DataFrame share the same index.

A DataFrame can be created from scratch, using the pd.DataFrame function. To do that, let us create four more Series which are going to comprise the DataFrame columns. Altogether, the DataFrame is going to represent various properties of three railway stations:

city = pd.Series(['Beer-Sheva', 'Beer-Sheva', 'Dimona'])
piano = pd.Series([False, True, False])
lon = pd.Series([34.798443, 34.812831, 35.011635])
lat = pd.Series([31.243288, 31.260284, 31.068616])

Here is a printout of the four Series we just created:

city
0    Beer-Sheva
1    Beer-Sheva
2        Dimona
dtype: object
piano
0    False
1     True
2    False
dtype: bool
lon
0    34.798443
1    34.812831
2    35.011635
dtype: float64
lat
0    31.243288
1    31.260284
2    31.068616
dtype: float64

Now, the six different Series can be combined into a DataFrame by passing them, as a dict (see Dictionaries (dict)), to the pd.DataFrame function. That way, the dict names will form the DataFrame column names, while the corresponding Series will form the column contents:

d = {
    'name': name, 
    'city': city, 
    'lines': lines, 
    'piano': piano, 
    'lon': lon, 
    'lat': lat
}
stations = pd.DataFrame(d)
stations
name city lines piano lon lat
0 Beer-Sheva Center Beer-Sheva 4 False 34.798443 31.243288
1 Beer-Sheva University Beer-Sheva 5 True 34.812831 31.260284
2 Dimona Dimona 1 False 35.011635 31.068616

Note

Note that there are other methods to create a DataFrame, passing different types of input to the pd.DataFrame function. For example, another useful method of creating a DataFrame by passing a two-dimensional array to pd.DataFrame.

DataFrame row indices are analogous to Series indices, and accessed exactly the same way:

stations.index
RangeIndex(start=0, stop=3, step=1)

Additionally, a DataFrame has column indices, which function as column names and can be accessed through the .columns property:

stations.columns
Index(['name', 'city', 'lines', 'piano', 'lon', 'lat'], dtype='object')

Reading from file#

In practice, we rarely need to create a DataFrame from scratch. More often, we read a table from a file with tabular data, such as a CSV file.

The pd.read_csv function is used to read a CSV file into a DataFrame in our Python environment. For example, the following expression reads the CSV file named 'ZonAnn.Ts+dSST.csv', which contains global temperature data from GISS. The values are anomalies, i.e. deviations from the 1951-1980 mean, in the years 1880-2023 ('Year'), both globally ('Glob') and for specific latitudes ('NHem'=Nothern Hemishpere, '24N-90N' latitudes 24-90, etc.):

dat = pd.read_csv('data/ZonAnn.Ts+dSST.csv')
dat
Year Glob NHem SHem 24N-90N ... EQU-24N 24S-EQU 44S-24S 64S-44S 90S-64S
0 1880 -0.17 -0.28 -0.05 -0.38 ... -0.14 -0.11 -0.04 0.05 0.67
1 1881 -0.09 -0.18 0.00 -0.36 ... 0.11 0.10 -0.06 -0.07 0.60
2 1882 -0.11 -0.21 -0.01 -0.31 ... -0.04 -0.05 0.01 0.04 0.63
3 1883 -0.17 -0.28 -0.07 -0.34 ... -0.17 -0.16 -0.04 0.07 0.50
4 1884 -0.28 -0.42 -0.15 -0.61 ... -0.12 -0.17 -0.19 -0.02 0.65
... ... ... ... ... ... ... ... ... ... ... ...
139 2019 0.98 1.20 0.75 1.42 ... 0.90 0.90 0.75 0.39 0.83
140 2020 1.01 1.35 0.68 1.67 ... 0.88 0.84 0.58 0.39 0.89
141 2021 0.85 1.14 0.56 1.42 ... 0.72 0.60 0.72 0.32 0.30
142 2022 0.89 1.16 0.62 1.52 ... 0.62 0.51 0.79 0.38 1.09
143 2023 1.17 1.49 0.85 1.78 ... 1.07 1.04 0.91 0.44 0.63

144 rows × 15 columns

Note

pandas has several functions named pd.read_* to read formats other than CSV, such as pd.read_excel, pd.read_sql, pd.read_json, pd.read_spss, and pd.read_stata. Similarly, DataFrame objects have multiple methods (see Writing DataFrame to file) to write to various output formats. See the IO tools section in the pandas user guide for an overview of the input/output capabilities.

Note that rows and columns are labelled using indices, as shown above. In this case:

  • Rows are labelled with consecutive integers

  • Columns are labelled with column names, which are imported from the first line in CSV file

DataFrame properties#

Overview#

One of the first things we may want to do with a DataFrame imported from a file is to examine its properties, as shown in the next few sections:

DataFrame dimensions#

DataFrame dimensions are accessible through the .shape property, just like in a numpy array (see Array dimensions). For example, dat (with the temperature data) has 144 rows and 15 columns:

dat.shape
(144, 15)

DataFrame column names#

The DataFrame column names can be obtained through the .columns property. Recall that this is a special data structure representing pandas indices (see Table from scratch):

dat.columns
Index(['Year', 'Glob', 'NHem', 'SHem', '24N-90N', '24S-24N', '90S-24S',
       '64N-90N', '44N-64N', '24N-44N', 'EQU-24N', '24S-EQU', '44S-24S',
       '64S-44S', '90S-64S'],
      dtype='object')

Exercise 05-a

  • From your experience with Python so far, can you guess how can the columns names be transformed to a plain list? (You can also try searching for the answer online.)

DataFrame column types#

The data types (see Data types) of the columns are contained in the .dtypes property:

dat.dtypes
Year         int64
Glob       float64
NHem       float64
SHem       float64
24N-90N    float64
            ...   
EQU-24N    float64
24S-EQU    float64
44S-24S    float64
64S-44S    float64
90S-64S    float64
Length: 15, dtype: object

Note that this is a Series (of dtype objects). This means we can get a specific value by index using the ordinary Series subsetting methods which we learn about later on (see Subsetting Series). For example, the 'Year' column was imported into a Series of type int64:

dat.dtypes.loc['Year']
dtype('int64')

while the 'Glob' column (global temperature anomaly) was imported into a float64 column:

dat.dtypes.loc['Glob']
dtype('float64')

The basic pandas data types are in agreement with numpy data types (see Data types). For example, the dat table contains int64 and float64 data types. However, pandas and its extensions (such as geopandas, see What is geopandas?) also extend the numpy functionality with new data types (such as GeometryDtype, see GeoSeries (geometry) column).

Renaming columns#

Sometimes it is necessary to rename DataFrame columns. For example, we may wish to use shorter names which are easier to type, or make the columns match another table we are working with. Columns can be renamed using the .rename method, which accepts a columns argument of the form {'old_name':'new_name',...}. For example, here is how we can replace the 'Year' column name with lowercase 'year':

dat.rename(columns={'Year': 'year'})
year Glob NHem SHem 24N-90N ... EQU-24N 24S-EQU 44S-24S 64S-44S 90S-64S
0 1880 -0.17 -0.28 -0.05 -0.38 ... -0.14 -0.11 -0.04 0.05 0.67
1 1881 -0.09 -0.18 0.00 -0.36 ... 0.11 0.10 -0.06 -0.07 0.60
2 1882 -0.11 -0.21 -0.01 -0.31 ... -0.04 -0.05 0.01 0.04 0.63
3 1883 -0.17 -0.28 -0.07 -0.34 ... -0.17 -0.16 -0.04 0.07 0.50
4 1884 -0.28 -0.42 -0.15 -0.61 ... -0.12 -0.17 -0.19 -0.02 0.65
... ... ... ... ... ... ... ... ... ... ... ...
139 2019 0.98 1.20 0.75 1.42 ... 0.90 0.90 0.75 0.39 0.83
140 2020 1.01 1.35 0.68 1.67 ... 0.88 0.84 0.58 0.39 0.89
141 2021 0.85 1.14 0.56 1.42 ... 0.72 0.60 0.72 0.32 0.30
142 2022 0.89 1.16 0.62 1.52 ... 0.62 0.51 0.79 0.38 1.09
143 2023 1.17 1.49 0.85 1.78 ... 1.07 1.04 0.91 0.44 0.63

144 rows × 15 columns

Note

Another column renaming technique is to assign a list of strings of the same length as the number of columns to the .columns attribute, as in dat.columns=['col1','col2',...].

Subsetting in pandas#

Overview#

There are numerous methods of subsetting Series and DataFrames in pandas. For example, Series values can be subsetted using the [ operator. However, this can be confusing since both using numeric indices and the index values are permitted, and it can be unclear which method is actually employed.

In agreement with the Python philosophy, where explicit is better than implicit, the recommended subsetting methods are using .loc and .iloc, which are discussed in the following sub-sections. The principal difference between .loc and .iloc is:

  • .loc uses pandas indices, while

  • .iloc uses (implicit) numpy-style numeric indices.

In this book, we are going to use .iloc and .loc methods when subsetting Series (this section) and DataFrame (see Subsetting DataFrames) objects. Nevertheless, for shorter syntax, we are going to use the [ shortcut in the following cases:

  • Selecting DataFrame columns using the [ operator and a string, or a list of strings, as in:

  • Filtering DataFrame rows using the [ operator and a boolean Series, as in:

Subsetting Series#

As mentioned above, the recommended way to subset Series is using the specialized methods named .loc and .iloc:

  • .loc—For subsetting using the Series index

  • .iloc—For subsetting using the (implicit) numpy index

What follows after the method are the indices, inside square brackets ([), whereas the indices can be one of:

  • An individual value, such as .loc['a'] or .iloc[0]

  • A slice, such as .loc['a':'b'] or .iloc[1:2]

  • A list, such as .loc[['a','c']] or .iloc[[0,2]]

For the next few examples, let us create a Series object named s, with:

  • values 11, 12, 13, 14, and

  • indices 'a', 'b', 'c', 'd',

using the index parameter of pd.Series:

s = pd.Series([11, 12, 13, 14], index=['a', 'b', 'c', 'd'])
s
a    11
b    12
c    13
d    14
dtype: int64

Here is how we can use .loc to select values using the three above-mentioned types of indices:

s.loc['a']         ## Individual index
11
s.loc['b':'c']     ## Slice
b    12
c    13
dtype: int64
s.loc[['a', 'c']]  ## 'list' of indices
a    11
c    13
dtype: int64

And here is how we can use .iloc to select values using numpy-style numeric indices:

s.iloc[0]       ## Individual index
11
s.iloc[1:2]     ## Slice
b    12
dtype: int64
s.iloc[[0, 2]]  ## 'list' of indices
a    11
c    13
dtype: int64

In the above examples, note how using an individual index returns a standalone value, while using a slice, or a list of indices, return a Series, even if the list or the slice contains just one element:

s.iloc[[0]]
a    11
dtype: int64

Also note that a slice using indices (such as s.loc['a':'b']) is inclusive, while a slice using implicit indices (such as .iloc[0:1]) excludes the last index similarly to list and numpy behavior.

Subsetting DataFrames#

Selecting DataFrame columns#

A DataFrame is a two-dimensional object, unlike a Series which is one-dimensional. Accordingly, the .loc and iloc methods of a DataFrame accept two indices, separated by a comma:

  • The first index refers to rows

  • The second index refers to columns

When we want to select all rows or columns, we place : in the respective index, similarly to numpy array subsetting (see Subsetting arrays). Using an individual index returns a Series, while using a slice or a list of indices—even if length 1—returns a DataFrame.

For example, this is how we can use .loc to extract one DataFrame column as a Series:

dat.loc[:, 'Glob']
0     -0.17
1     -0.09
2     -0.11
3     -0.17
4     -0.28
       ... 
139    0.98
140    1.01
141    0.85
142    0.89
143    1.17
Name: Glob, Length: 144, dtype: float64

and here is how we can extract a single column as a DataFrame:

dat.loc[:, ['Glob']]
Glob
0 -0.17
1 -0.09
2 -0.11
3 -0.17
4 -0.28
... ...
139 0.98
140 1.01
141 0.85
142 0.89
143 1.17

144 rows × 1 columns

As mentioned above (see Overview), we can also use the shortcut [ operator to subset columns(s):

dat['Glob']  ## Shortcut for `dat.loc[:,'Glob']`
0     -0.17
1     -0.09
2     -0.11
3     -0.17
4     -0.28
       ... 
139    0.98
140    1.01
141    0.85
142    0.89
143    1.17
Name: Glob, Length: 144, dtype: float64
dat[['Glob']]  ## Shortcut for `dat.loc[:,['Glob']]`
Glob
0 -0.17
1 -0.09
2 -0.11
3 -0.17
4 -0.28
... ...
139 0.98
140 1.01
141 0.85
142 0.89
143 1.17

144 rows × 1 columns

Note

Dot notation, as in dat.Glob can also be used to select individual columns of a DataFrame. However, this is not recommended, because it does not work (1) with non-string column names, and (2) with column names that are the same as DataFrame methods, such as .pop.

We can pass a list of column names to select more then one column. For example:

dat[['Year', 'Glob', 'NHem', 'SHem']]
Year Glob NHem SHem
0 1880 -0.17 -0.28 -0.05
1 1881 -0.09 -0.18 0.00
2 1882 -0.11 -0.21 -0.01
3 1883 -0.17 -0.28 -0.07
4 1884 -0.28 -0.42 -0.15
... ... ... ... ...
139 2019 0.98 1.20 0.75
140 2020 1.01 1.35 0.68
141 2021 0.85 1.14 0.56
142 2022 0.89 1.16 0.62
143 2023 1.17 1.49 0.85

144 rows × 4 columns

To select all columns except for the specified ones, we can use the .drop method combined with axis=1:

dat.drop(['Year', 'Glob', 'NHem', 'SHem', '24N-90N', '24S-24N', '90S-24S'], axis=1)
64N-90N 44N-64N 24N-44N EQU-24N 24S-EQU 44S-24S 64S-44S 90S-64S
0 -0.80 -0.52 -0.26 -0.14 -0.11 -0.04 0.05 0.67
1 -0.93 -0.50 -0.19 0.11 0.10 -0.06 -0.07 0.60
2 -1.41 -0.31 -0.13 -0.04 -0.05 0.01 0.04 0.63
3 -0.18 -0.58 -0.24 -0.17 -0.16 -0.04 0.07 0.50
4 -1.30 -0.66 -0.45 -0.12 -0.17 -0.19 -0.02 0.65
... ... ... ... ... ... ... ... ...
139 2.71 1.43 0.99 0.90 0.90 0.75 0.39 0.83
140 2.88 1.81 1.19 0.88 0.84 0.58 0.39 0.89
141 2.05 1.35 1.26 0.72 0.60 0.72 0.32 0.30
142 2.34 1.50 1.27 0.62 0.51 0.79 0.38 1.09
143 2.58 1.87 1.47 1.07 1.04 0.91 0.44 0.63

144 rows × 8 columns

Another useful technique is to use .loc combined with slices of column names. For example, we can select all columns between two specified ones:

dat.loc[:, 'Year':'SHem']
Year Glob NHem SHem
0 1880 -0.17 -0.28 -0.05
1 1881 -0.09 -0.18 0.00
2 1882 -0.11 -0.21 -0.01
3 1883 -0.17 -0.28 -0.07
4 1884 -0.28 -0.42 -0.15
... ... ... ... ...
139 2019 0.98 1.20 0.75
140 2020 1.01 1.35 0.68
141 2021 0.85 1.14 0.56
142 2022 0.89 1.16 0.62
143 2023 1.17 1.49 0.85

144 rows × 4 columns

or all columns before a specified one:

dat.loc[:, :'SHem']
Year Glob NHem SHem
0 1880 -0.17 -0.28 -0.05
1 1881 -0.09 -0.18 0.00
2 1882 -0.11 -0.21 -0.01
3 1883 -0.17 -0.28 -0.07
4 1884 -0.28 -0.42 -0.15
... ... ... ... ...
139 2019 0.98 1.20 0.75
140 2020 1.01 1.35 0.68
141 2021 0.85 1.14 0.56
142 2022 0.89 1.16 0.62
143 2023 1.17 1.49 0.85

144 rows × 4 columns

or all columns after a specified one:

dat.loc[:, 'SHem':]
SHem 24N-90N 24S-24N 90S-24S 64N-90N ... EQU-24N 24S-EQU 44S-24S 64S-44S 90S-64S
0 -0.05 -0.38 -0.13 -0.02 -0.80 ... -0.14 -0.11 -0.04 0.05 0.67
1 0.00 -0.36 0.10 -0.07 -0.93 ... 0.11 0.10 -0.06 -0.07 0.60
2 -0.01 -0.31 -0.04 0.01 -1.41 ... -0.04 -0.05 0.01 0.04 0.63
3 -0.07 -0.34 -0.17 -0.01 -0.18 ... -0.17 -0.16 -0.04 0.07 0.50
4 -0.15 -0.61 -0.14 -0.14 -1.30 ... -0.12 -0.17 -0.19 -0.02 0.65
... ... ... ... ... ... ... ... ... ... ... ...
139 0.75 1.42 0.90 0.64 2.71 ... 0.90 0.90 0.75 0.39 0.83
140 0.68 1.67 0.86 0.57 2.88 ... 0.88 0.84 0.58 0.39 0.89
141 0.56 1.42 0.66 0.53 2.05 ... 0.72 0.60 0.72 0.32 0.30
142 0.62 1.52 0.57 0.70 2.34 ... 0.62 0.51 0.79 0.38 1.09
143 0.85 1.78 1.05 0.72 2.58 ... 1.07 1.04 0.91 0.44 0.63

144 rows × 12 columns

Again, note that in the last examples, the subset is inclusive, i.e., includes both the start and end columns.

Exercise 05-b

  • How can we select all columns after the specified one excluding itself?

Selecting DataFrame rows#

Rows can be selected using .loc or .iloc, similarly to the way we use those methods to select columns. The difference is that we specify the first index, instead of the second. For example:

dat.iloc[[0], :]     ## 1st row
Year Glob NHem SHem 24N-90N ... EQU-24N 24S-EQU 44S-24S 64S-44S 90S-64S
0 1880 -0.17 -0.28 -0.05 -0.38 ... -0.14 -0.11 -0.04 0.05 0.67

1 rows × 15 columns

dat.iloc[0:3, :]     ## 1st row (inclusive) to 4th row (exclusive)
Year Glob NHem SHem 24N-90N ... EQU-24N 24S-EQU 44S-24S 64S-44S 90S-64S
0 1880 -0.17 -0.28 -0.05 -0.38 ... -0.14 -0.11 -0.04 0.05 0.67
1 1881 -0.09 -0.18 0.00 -0.36 ... 0.11 0.10 -0.06 -0.07 0.60
2 1882 -0.11 -0.21 -0.01 -0.31 ... -0.04 -0.05 0.01 0.04 0.63

3 rows × 15 columns

dat.iloc[[0, 2], :]  ## 1st and 3rd rows
Year Glob NHem SHem 24N-90N ... EQU-24N 24S-EQU 44S-24S 64S-44S 90S-64S
0 1880 -0.17 -0.28 -0.05 -0.38 ... -0.14 -0.11 -0.04 0.05 0.67
2 1882 -0.11 -0.21 -0.01 -0.31 ... -0.04 -0.05 0.01 0.04 0.63

2 rows × 15 columns

Exercise 05-c

  • Create a subset of dat with the 1st, 3rd, and 5th rows.

Selecting rows and columns#

Both DataFrame dimensions can be subsetted at once, by passing two indices to .loc or to .iloc. For example, here we select a subset of dat with the first four rows and the first four columns (i.e., the top-left “corner” of the table), using .iloc:

dat.iloc[0:4, 0:4]
Year Glob NHem SHem
0 1880 -0.17 -0.28 -0.05
1 1881 -0.09 -0.18 0.00
2 1882 -0.11 -0.21 -0.01
3 1883 -0.17 -0.28 -0.07

The same can be acheived using separate steps, for example using .loc to select columns, and then .iloc to select rows:

dat.loc[:, 'Year':'SHem'].iloc[0:4, :]
Year Glob NHem SHem
0 1880 -0.17 -0.28 -0.05
1 1881 -0.09 -0.18 0.00
2 1882 -0.11 -0.21 -0.01
3 1883 -0.17 -0.28 -0.07

Selecting DataFrame values#

Using the above-mentioned methods, we can access individual values from a DataFrame in several ways. The clearest syntax is probably splitting the operation into two parts: first selecting the column, then selecting the index within the column. For example, here is how we can get the first value in the 'Year' column of dat:

dat['Year'].iloc[0]
1880

Exercise 05-d

  • How can we get the last value in the 'Year' column of dat, without explicitly typing its index (e.g. 142)?

Series/DataFrame to ndarray#

A Series or a DataFrame is accessible as an ndarray, through the .to_numpy method. For example:

dat['Year'].iloc[0:3].to_numpy()  ## 'Series' to array
array([1880, 1881, 1882])
dat.iloc[0:3, 1:3].to_numpy()     ## 'DataFrame' to array
array([[-0.17, -0.28],
       [-0.09, -0.18],
       [-0.11, -0.21]])

Note that a Series is translated to a one-dimensional array, while a DataFrame is translated to a two-dimensional array.

Conversion to list#

Series to list#

In some cases we may prefer to use basic Python methods to work with data contained in a Series or in a DataFrame. For that purpose, the Series or the DataFrame can be converted to a list.

A Series can be converted to a list by directly applying the .to_list method, which is analogous to the .tolist method of ndarray (see ndarray to list). For example, let us take a small Series of length 5:

s = dat['Glob'].iloc[0:5]
s
0   -0.17
1   -0.09
2   -0.11
3   -0.17
4   -0.28
Name: Glob, dtype: float64

Here is how the Series can be converted to a list, using the .to_list method:

s.to_list()
[-0.17, -0.09, -0.11, -0.17, -0.28]

DataFrame to list#

A DataFrame can also be converted to a list, in which case the list will be nested, with each column represented by an internal list. Let us take a small DataFrame with five rows and three columns as an example:

dat1 = dat[['Year', 'NHem', 'SHem']].iloc[0:5, :]
dat1
Year NHem SHem
0 1880 -0.28 -0.05
1 1881 -0.18 0.00
2 1882 -0.21 -0.01
3 1883 -0.28 -0.07
4 1884 -0.42 -0.15

To convert the DataFrame to a list, we can use the .to_records method (possibly with index=False to omit the index), followed by .tolist. The result is a list to tuples, where each tuple represents a row:

dat1.to_records(index=False).tolist()
[(1880, -0.28, -0.05),
 (1881, -0.18, 0.0),
 (1882, -0.21, -0.01),
 (1883, -0.28, -0.07),
 (1884, -0.42, -0.15)]

Modifying the index#

Overivew#

In the above examples, the Series and DataFrame we created got the default, consecutive integer, index. As we will see, the index plays an important role in many operations in pandas. Therefore, often we would like to set a more meaningful, custom index. For example, it usually makes sense to set the index of a table representing a time series (such as dat) to the time points (such as 'Year' values). Many pandas operations then utilize the index. For example, when plotting the data, the x-axis will show the time labels (see Line plots (pandas)).

To set a new index, we:

To reset the index, i.e., get back to the default consecutive integer index, we use the .reset_index method (see Resetting the index).

We demonstrate these three techniques in the following sub-sections.

Setting Series index#

Let us go back to the name series from the beginning of the chapter (Creating a Series):

name
0        Beer-Sheva Center
1    Beer-Sheva University
2                   Dimona
dtype: object

The following expression sets the index of the name series as ['a','b','c'], by assigning a list into the .index property:

name.index = ['a', 'b', 'c']
name
a        Beer-Sheva Center
b    Beer-Sheva University
c                   Dimona
dtype: object

Setting DataFrame index#

The (row) index of a DataFrame can also be “manually” set by assignment to .index. However, a more useful scenario is where we want one of the columns in the DataFrame to serve as the index, while also removing it from the DataFrame columns to avoid duplication. This can be done using .set_index.

For example, the following expression changes the index of stations to the station names:

stations = stations.set_index('name')
stations
city lines piano lon lat
name
Beer-Sheva Center Beer-Sheva 4 False 34.798443 31.243288
Beer-Sheva University Beer-Sheva 5 True 34.812831 31.260284
Dimona Dimona 1 False 35.011635 31.068616

Now, name is no longer a column, but the index of stations.

Resetting the index#

To reset the index of a Series or a DataFrame, we use the .reset_index method. As a result, the index will be replaced with the default consecutive integer sequence. In case we want to remove the information in the index altoghether, we use the drop=True option. Otherwise, the index will be “transferred” into a new column.

For example, here is how we remove the ['a','b','c'] index, to get the original default index in the name Series. Note that the information in the index is lost and we get back to the original Series, due to the drop=True option:

name = name.reset_index(drop=True)
name
0        Beer-Sheva Center
1    Beer-Sheva University
2                   Dimona
dtype: object

And here is how we reset the row index in the DataFrame named stations. This time, we do not want to lose the information, therefore the index “goes back” to the 'name' column:

stations = stations.reset_index()
stations
name city lines piano lon lat
0 Beer-Sheva Center Beer-Sheva 4 False 34.798443 31.243288
1 Beer-Sheva University Beer-Sheva 5 True 34.812831 31.260284
2 Dimona Dimona 1 False 35.011635 31.068616

Sorting#

A DataFrame can be sorted using the .sort_values method. The first parameter (by) accepts a column name, or a list of column names, to sort by.

For example, the following expression sorts the rows of dat according to the global temperature anomaly, from lowest (in 1909) to highest (in 2016):

dat.sort_values('Glob')
Year Glob NHem SHem 24N-90N ... EQU-24N 24S-EQU 44S-24S 64S-44S 90S-64S
29 1909 -0.48 -0.49 -0.48 -0.53 ... -0.44 -0.53 -0.37 -0.52 -0.57
24 1904 -0.47 -0.48 -0.46 -0.51 ... -0.44 -0.50 -0.37 -0.49 -1.32
37 1917 -0.46 -0.57 -0.36 -0.50 ... -0.68 -0.59 -0.23 -0.09 0.06
31 1911 -0.44 -0.42 -0.47 -0.42 ... -0.42 -0.45 -0.43 -0.53 0.05
30 1910 -0.44 -0.44 -0.43 -0.42 ... -0.48 -0.51 -0.33 -0.45 0.16
... ... ... ... ... ... ... ... ... ... ... ...
137 2017 0.92 1.17 0.67 1.38 ... 0.86 0.78 0.76 0.35 0.54
139 2019 0.98 1.20 0.75 1.42 ... 0.90 0.90 0.75 0.39 0.83
140 2020 1.01 1.35 0.68 1.67 ... 0.88 0.84 0.58 0.39 0.89
136 2016 1.01 1.31 0.73 1.55 ... 0.95 1.07 0.67 0.25 0.38
143 2023 1.17 1.49 0.85 1.78 ... 1.07 1.04 0.91 0.44 0.63

144 rows × 15 columns

The default is to sort in ascending order. If we want the opposite, i.e., a descending order, we need to specify ascending=False:

dat.sort_values('Glob', ascending=False)
Year Glob NHem SHem 24N-90N ... EQU-24N 24S-EQU 44S-24S 64S-44S 90S-64S
143 2023 1.17 1.49 0.85 1.78 ... 1.07 1.04 0.91 0.44 0.63
140 2020 1.01 1.35 0.68 1.67 ... 0.88 0.84 0.58 0.39 0.89
136 2016 1.01 1.31 0.73 1.55 ... 0.95 1.07 0.67 0.25 0.38
139 2019 0.98 1.20 0.75 1.42 ... 0.90 0.90 0.75 0.39 0.83
137 2017 0.92 1.17 0.67 1.38 ... 0.86 0.78 0.76 0.35 0.54
... ... ... ... ... ... ... ... ... ... ... ...
30 1910 -0.44 -0.44 -0.43 -0.42 ... -0.48 -0.51 -0.33 -0.45 0.16
31 1911 -0.44 -0.42 -0.47 -0.42 ... -0.42 -0.45 -0.43 -0.53 0.05
37 1917 -0.46 -0.57 -0.36 -0.50 ... -0.68 -0.59 -0.23 -0.09 0.06
24 1904 -0.47 -0.48 -0.46 -0.51 ... -0.44 -0.50 -0.37 -0.49 -1.32
29 1909 -0.48 -0.49 -0.48 -0.53 ... -0.44 -0.53 -0.37 -0.52 -0.57

144 rows × 15 columns

Plotting (pandas)#

Histograms (pandas)#

pandas has several methods to visualize the information in a Series or a DataFrame, which are actually shortcuts to matplotlib functions (see The matplotlib package). These methods are useful for quick visual inspection of the data.

For example, we can draw a histogram of the column(s) in a DataFrame using the .hist method. For example:

dat[['Glob', 'SHem', 'NHem']].hist();
_images/0602dc80ff8ecb1b83e81827637be11222b3cefc7037f23a09e915e863eb312b.png

Line plots (pandas)#

The .plot method, given a Series (or a DataFrame with one column), creates a line plot with the indices on the x-axis and the values on the y-axis:

dat['Glob'].plot();
_images/a0abc2bdda713064358b99173c689fcc34297ebc45af8e910e467164d0e575e9.png

In case we need to change the values on the x-axis, the simplest way is to set the required values as the index (see Modifying the index). For example, here we set the 'Year' column as the DataFrame index, then plot the 'Glob' column. As a result, the measurement year appears on the x-axis instead of the default consecutive index:

dat.set_index('Year')['Glob'].plot();
_images/16af7ff367fe5d25aa213ff5b6f57c0f85b937435c08a00ada1c4725ec0310f5.png

Pay attention to the order of operations (from left to right) in the above expression:

  • dat.set_index('Year')—Setting the index

  • ['Glob']—Extracting a column (as Series)

  • .plot();—Plotting

When we plot a DataFrame with more than one column, the columns are plotted as separate series in the same line plot. This is useful to compare the values in different columns. For example, plotting the 'Glob', 'NHem', and 'SHem' columns demonstrates that in recent years Northern Hemisphere ('NHem') temperatures have been relatively higher, while Southern Hemisphere ('SHem') temperatures have been relatively lower, compared to the global average ('Glob'):

dat.set_index('Year')[['Glob', 'NHem', 'SHem']].plot();
_images/03abc595b9c2d9ffc75f42ccf5b02ca38925a9e399647a9975fb6cb02cbecc75.png

Scatterplots (pandas)#

One more useful type of plot is a scatterplot, where we display the association between two series in the form of scattered points. To produce a scatterplot, we use the .plot.scatter method of a DataFrame, specifying the names of the columns to be displayed in the x-axis (x) and y-axis (y).

For example, the following expression shows the relation between the values in the 'NHem' and 'SHem' columns of dat:

dat.plot.scatter(x='NHem', y='SHem');
_images/c4ebcc1f1926605a85cfcc900152b7d0d49a95216a74d508c347959a1cdde7b1.png

We can see a strong positive association between Northern and Southern Hemisphere temperatures across years. Namely, in years when the Nothern Hemisphere temperature is high, the Southern Hemisphere temperature also tends to be high, and vice versa.

Operators#

Operators on Series#

Operators and summary functions can be applied to Series objects, similarly to the way they can be applied to numpy arrays (see Vectorized operations and Summarizing array values, respectively). For example, summary methods such as .min, .max, and .mean can be applied on a Series to summarize its respective properties. Here is how we can find out the start and end 'Year' in dat:

dat['Year'].min()
1880
dat['Year'].max()
2023

and the average 'Glob' value:

dat['Glob'].mean()
0.06770833333333333

We can also combine Series with individual values, or combine two series, to apply pairwise arithmetic or boolean operators. For example, here is how we can subtract 1 from all 'Year' values:

dat['Year'] - 1
0      1879
1      1880
2      1881
3      1882
4      1883
       ... 
139    2018
140    2019
141    2020
142    2021
143    2022
Name: Year, Length: 144, dtype: int64

and here is how we can calculate the yearly differences between the Northern Hemisphere and Southern Hemisphere temperature anomalies:

dat['NHem'] - dat['SHem']
0     -0.23
1     -0.18
2     -0.20
3     -0.21
4     -0.27
       ... 
139    0.45
140    0.67
141    0.58
142    0.54
143    0.64
Length: 144, dtype: float64

Note that this is an arithmetic operation between two Series, which returns a new series.

Note

When performing operations between pairs of Series or DataFrames, pandas aligns the elements according to the index, rather than the position. When the data come from the same table (such as in the last example), the indices are guaranteed to match. However, when the data come from different tables, you must make sure their indices match.

Operators on DataFrames#

Operating on an entire DataFrame is more complex than operating on a Series (see Operators on Series). Accordingly, operators behave in different ways. For example, arithmetic and boolean operations combined with an individual value are applied per-element, resulting in a new DataFrame (assuming that all columns are numeric):

dat - 10
Year Glob NHem SHem 24N-90N ... EQU-24N 24S-EQU 44S-24S 64S-44S 90S-64S
0 1870 -10.17 -10.28 -10.05 -10.38 ... -10.14 -10.11 -10.04 -9.95 -9.33
1 1871 -10.09 -10.18 -10.00 -10.36 ... -9.89 -9.90 -10.06 -10.07 -9.40
2 1872 -10.11 -10.21 -10.01 -10.31 ... -10.04 -10.05 -9.99 -9.96 -9.37
3 1873 -10.17 -10.28 -10.07 -10.34 ... -10.17 -10.16 -10.04 -9.93 -9.50
4 1874 -10.28 -10.42 -10.15 -10.61 ... -10.12 -10.17 -10.19 -10.02 -9.35
... ... ... ... ... ... ... ... ... ... ... ...
139 2009 -9.02 -8.80 -9.25 -8.58 ... -9.10 -9.10 -9.25 -9.61 -9.17
140 2010 -8.99 -8.65 -9.32 -8.33 ... -9.12 -9.16 -9.42 -9.61 -9.11
141 2011 -9.15 -8.86 -9.44 -8.58 ... -9.28 -9.40 -9.28 -9.68 -9.70
142 2012 -9.11 -8.84 -9.38 -8.48 ... -9.38 -9.49 -9.21 -9.62 -8.91
143 2013 -8.83 -8.51 -9.15 -8.22 ... -8.93 -8.96 -9.09 -9.56 -9.37

144 rows × 15 columns

Methods such as .mean, however, are by default applied per column (i.e., axis=0):

dat.mean()
Year       1951.500000
Glob          0.067708
NHem          0.099444
SHem          0.036667
24N-90N       0.124722
              ...     
EQU-24N       0.066181
24S-EQU       0.083056
44S-24S       0.047917
64S-44S      -0.053681
90S-64S      -0.074028
Length: 15, dtype: float64

We elaborate on this type of row- and column-wise operations later on (see Row/col-wise operations).

Creating new columns#

New DataFrame columns can be created by assignment of a Series to a non-existing column index. For example, the following expression calculates a Series of yearly differences between the Northern Hemisphere and Southern Hemisphere temperatures, as shown above, and assigns it to a new column named diff:

dat['diff'] = dat['NHem'] - dat['SHem']
dat
Year Glob NHem SHem 24N-90N ... 24S-EQU 44S-24S 64S-44S 90S-64S diff
0 1880 -0.17 -0.28 -0.05 -0.38 ... -0.11 -0.04 0.05 0.67 -0.23
1 1881 -0.09 -0.18 0.00 -0.36 ... 0.10 -0.06 -0.07 0.60 -0.18
2 1882 -0.11 -0.21 -0.01 -0.31 ... -0.05 0.01 0.04 0.63 -0.20
3 1883 -0.17 -0.28 -0.07 -0.34 ... -0.16 -0.04 0.07 0.50 -0.21
4 1884 -0.28 -0.42 -0.15 -0.61 ... -0.17 -0.19 -0.02 0.65 -0.27
... ... ... ... ... ... ... ... ... ... ... ...
139 2019 0.98 1.20 0.75 1.42 ... 0.90 0.75 0.39 0.83 0.45
140 2020 1.01 1.35 0.68 1.67 ... 0.84 0.58 0.39 0.89 0.67
141 2021 0.85 1.14 0.56 1.42 ... 0.60 0.72 0.32 0.30 0.58
142 2022 0.89 1.16 0.62 1.52 ... 0.51 0.79 0.38 1.09 0.54
143 2023 1.17 1.49 0.85 1.78 ... 1.04 0.91 0.44 0.63 0.64

144 rows × 16 columns

Here is a plot (Line plots (pandas)) of the differences we just calulated, as function of time:

dat.set_index('Year')['diff'].plot();
_images/860be41eecab528ede501d7d98f923bb66e15505af329216694f82bcfed88569.png

We can also assign an expression that combines series with individual values:

dat['NHem2'] = dat['NHem'] * 2
dat
Year Glob NHem SHem 24N-90N ... 44S-24S 64S-44S 90S-64S diff NHem2
0 1880 -0.17 -0.28 -0.05 -0.38 ... -0.04 0.05 0.67 -0.23 -0.56
1 1881 -0.09 -0.18 0.00 -0.36 ... -0.06 -0.07 0.60 -0.18 -0.36
2 1882 -0.11 -0.21 -0.01 -0.31 ... 0.01 0.04 0.63 -0.20 -0.42
3 1883 -0.17 -0.28 -0.07 -0.34 ... -0.04 0.07 0.50 -0.21 -0.56
4 1884 -0.28 -0.42 -0.15 -0.61 ... -0.19 -0.02 0.65 -0.27 -0.84
... ... ... ... ... ... ... ... ... ... ... ...
139 2019 0.98 1.20 0.75 1.42 ... 0.75 0.39 0.83 0.45 2.40
140 2020 1.01 1.35 0.68 1.67 ... 0.58 0.39 0.89 0.67 2.70
141 2021 0.85 1.14 0.56 1.42 ... 0.72 0.32 0.30 0.58 2.28
142 2022 0.89 1.16 0.62 1.52 ... 0.79 0.38 1.09 0.54 2.32
143 2023 1.17 1.49 0.85 1.78 ... 0.91 0.44 0.63 0.64 2.98

144 rows × 17 columns

or an individual value on its own (in which case it is duplicated across all rows):

dat['variable'] = 'temperature'
dat
Year Glob NHem SHem 24N-90N ... 64S-44S 90S-64S diff NHem2 variable
0 1880 -0.17 -0.28 -0.05 -0.38 ... 0.05 0.67 -0.23 -0.56 temperature
1 1881 -0.09 -0.18 0.00 -0.36 ... -0.07 0.60 -0.18 -0.36 temperature
2 1882 -0.11 -0.21 -0.01 -0.31 ... 0.04 0.63 -0.20 -0.42 temperature
3 1883 -0.17 -0.28 -0.07 -0.34 ... 0.07 0.50 -0.21 -0.56 temperature
4 1884 -0.28 -0.42 -0.15 -0.61 ... -0.02 0.65 -0.27 -0.84 temperature
... ... ... ... ... ... ... ... ... ... ... ...
139 2019 0.98 1.20 0.75 1.42 ... 0.39 0.83 0.45 2.40 temperature
140 2020 1.01 1.35 0.68 1.67 ... 0.39 0.89 0.67 2.70 temperature
141 2021 0.85 1.14 0.56 1.42 ... 0.32 0.30 0.58 2.28 temperature
142 2022 0.89 1.16 0.62 1.52 ... 0.38 1.09 0.54 2.32 temperature
143 2023 1.17 1.49 0.85 1.78 ... 0.44 0.63 0.64 2.98 temperature

144 rows × 18 columns

Let us delete the 'diff', 'NHem2', 'variable' columns we just created to get the original version of dat:

dat = dat.drop(['diff', 'NHem2', 'variable'], axis=1)
dat
Year Glob NHem SHem 24N-90N ... EQU-24N 24S-EQU 44S-24S 64S-44S 90S-64S
0 1880 -0.17 -0.28 -0.05 -0.38 ... -0.14 -0.11 -0.04 0.05 0.67
1 1881 -0.09 -0.18 0.00 -0.36 ... 0.11 0.10 -0.06 -0.07 0.60
2 1882 -0.11 -0.21 -0.01 -0.31 ... -0.04 -0.05 0.01 0.04 0.63
3 1883 -0.17 -0.28 -0.07 -0.34 ... -0.17 -0.16 -0.04 0.07 0.50
4 1884 -0.28 -0.42 -0.15 -0.61 ... -0.12 -0.17 -0.19 -0.02 0.65
... ... ... ... ... ... ... ... ... ... ... ...
139 2019 0.98 1.20 0.75 1.42 ... 0.90 0.90 0.75 0.39 0.83
140 2020 1.01 1.35 0.68 1.67 ... 0.88 0.84 0.58 0.39 0.89
141 2021 0.85 1.14 0.56 1.42 ... 0.72 0.60 0.72 0.32 0.30
142 2022 0.89 1.16 0.62 1.52 ... 0.62 0.51 0.79 0.38 1.09
143 2023 1.17 1.49 0.85 1.78 ... 1.07 1.04 0.91 0.44 0.63

144 rows × 15 columns

DataFrame filtering#

Using conditional operators#

Most often, instead of selecting rows by index or by position (see Selecting DataFrame rows), we want to filter rows by a condition that we apply on table values. For example, suppose that we want to get a subset of dat with all rows after the year 2020. To do that, we can create a boolean Series, specifying which rows are to be retained:

sel = dat['Year'] > 2020
sel
0      False
1      False
2      False
3      False
4      False
       ...  
139    False
140    False
141     True
142     True
143     True
Name: Year, Length: 144, dtype: bool

Then, the boolean series can be passed as an index inside square brackets ([), to select those rows corresponding to True:

dat[sel]
Year Glob NHem SHem 24N-90N ... EQU-24N 24S-EQU 44S-24S 64S-44S 90S-64S
141 2021 0.85 1.14 0.56 1.42 ... 0.72 0.60 0.72 0.32 0.30
142 2022 0.89 1.16 0.62 1.52 ... 0.62 0.51 0.79 0.38 1.09
143 2023 1.17 1.49 0.85 1.78 ... 1.07 1.04 0.91 0.44 0.63

3 rows × 15 columns

The same can be acheived in a single step, without storing the Series in a variable such as sel, as follows:

dat[dat['Year'] > 2020]
Year Glob NHem SHem 24N-90N ... EQU-24N 24S-EQU 44S-24S 64S-44S 90S-64S
141 2021 0.85 1.14 0.56 1.42 ... 0.72 0.60 0.72 0.32 0.30
142 2022 0.89 1.16 0.62 1.52 ... 0.62 0.51 0.79 0.38 1.09
143 2023 1.17 1.49 0.85 1.78 ... 1.07 1.04 0.91 0.44 0.63

3 rows × 15 columns

Here is an example with a more complex condition (note the parantheses!):

dat[(dat['Year'] > 2020) & (dat['Year'] <= 2022)]
Year Glob NHem SHem 24N-90N ... EQU-24N 24S-EQU 44S-24S 64S-44S 90S-64S
141 2021 0.85 1.14 0.56 1.42 ... 0.72 0.60 0.72 0.32 0.30
142 2022 0.89 1.16 0.62 1.52 ... 0.62 0.51 0.79 0.38 1.09

2 rows × 15 columns

To get the values of partucular column(s), we can use .loc, as follows:

dat.loc[(dat['Year'] > 2020) & (dat['Year'] <= 2022), 'Glob']
141    0.85
142    0.89
Name: Glob, dtype: float64

Exercise 05-e

  • Create a subset of dat with all rows where the global temperature anomaly ('Glob') was above 1 degree.

Using .isin#

Sometimes, the condition we are interested in is not a comparison against an individual value (such as >2017, see Using conditional operators), but multiple values. For example, suppose we want to extract the rows corresponding to the years 2008, 2012, and 2016, in dat. One way is to compose a combined condition using |, using the numpy syntax we are familiar with (see numpy conditional operators). Again, recall that we must enclose each internal expression in breackets ( so that it is evaluated before |:

sel = (dat['Year'] == 2008) | (dat['Year'] == 2012) | (dat['Year'] == 2016)
dat[sel]
Year Glob NHem SHem 24N-90N ... EQU-24N 24S-EQU 44S-24S 64S-44S 90S-64S
128 2008 0.54 0.68 0.40 0.91 ... 0.34 0.41 0.52 0.08 0.60
132 2012 0.65 0.81 0.49 1.04 ... 0.48 0.57 0.56 0.19 0.45
136 2016 1.01 1.31 0.73 1.55 ... 0.95 1.07 0.67 0.25 0.38

3 rows × 15 columns

As you can imagine, this method quickly becomes unfeasible if the number of items becomes large. Instead, we can use the .isin method, combined with a list of items to compare with the Series values:

sel = dat['Year'].isin([2008, 2012, 2016])
dat[sel]
Year Glob NHem SHem 24N-90N ... EQU-24N 24S-EQU 44S-24S 64S-44S 90S-64S
128 2008 0.54 0.68 0.40 0.91 ... 0.34 0.41 0.52 0.08 0.60
132 2012 0.65 0.81 0.49 1.04 ... 0.48 0.57 0.56 0.19 0.45
136 2016 1.01 1.31 0.73 1.55 ... 0.95 1.07 0.67 0.25 0.38

3 rows × 15 columns

Recall the in operator which we learned about earlier (see The in operator). The in operator is analogous to the .isin method, but intended for lists.

Exercise 05-f

  • What do you think is the meaning of dat[~sel]? Try it to check your answer.

Assignment to subsets#

When assigning to subsets of a column, we need to use the .loc method. Typically we combine:

  • A boolean Series in the rows index, to select the observations we want to modify

  • An str in the columns index, to select the variable we want to modify

For example, the following expression sets the 'NHem' column value to 9999 in all years after 2020:

dat.loc[dat['Year'] > 2020, 'NHem'] = 9999
dat
Year Glob NHem SHem 24N-90N ... EQU-24N 24S-EQU 44S-24S 64S-44S 90S-64S
0 1880 -0.17 -0.28 -0.05 -0.38 ... -0.14 -0.11 -0.04 0.05 0.67
1 1881 -0.09 -0.18 0.00 -0.36 ... 0.11 0.10 -0.06 -0.07 0.60
2 1882 -0.11 -0.21 -0.01 -0.31 ... -0.04 -0.05 0.01 0.04 0.63
3 1883 -0.17 -0.28 -0.07 -0.34 ... -0.17 -0.16 -0.04 0.07 0.50
4 1884 -0.28 -0.42 -0.15 -0.61 ... -0.12 -0.17 -0.19 -0.02 0.65
... ... ... ... ... ... ... ... ... ... ... ...
139 2019 0.98 1.20 0.75 1.42 ... 0.90 0.90 0.75 0.39 0.83
140 2020 1.01 1.35 0.68 1.67 ... 0.88 0.84 0.58 0.39 0.89
141 2021 0.85 9999.00 0.56 1.42 ... 0.72 0.60 0.72 0.32 0.30
142 2022 0.89 9999.00 0.62 1.52 ... 0.62 0.51 0.79 0.38 1.09
143 2023 1.17 9999.00 0.85 1.78 ... 1.07 1.04 0.91 0.44 0.63

144 rows × 15 columns

To get back to the original values, let us read the CSV file once again:

dat = pd.read_csv('data/ZonAnn.Ts+dSST.csv')
dat
Year Glob NHem SHem 24N-90N ... EQU-24N 24S-EQU 44S-24S 64S-44S 90S-64S
0 1880 -0.17 -0.28 -0.05 -0.38 ... -0.14 -0.11 -0.04 0.05 0.67
1 1881 -0.09 -0.18 0.00 -0.36 ... 0.11 0.10 -0.06 -0.07 0.60
2 1882 -0.11 -0.21 -0.01 -0.31 ... -0.04 -0.05 0.01 0.04 0.63
3 1883 -0.17 -0.28 -0.07 -0.34 ... -0.17 -0.16 -0.04 0.07 0.50
4 1884 -0.28 -0.42 -0.15 -0.61 ... -0.12 -0.17 -0.19 -0.02 0.65
... ... ... ... ... ... ... ... ... ... ... ...
139 2019 0.98 1.20 0.75 1.42 ... 0.90 0.90 0.75 0.39 0.83
140 2020 1.01 1.35 0.68 1.67 ... 0.88 0.84 0.58 0.39 0.89
141 2021 0.85 1.14 0.56 1.42 ... 0.72 0.60 0.72 0.32 0.30
142 2022 0.89 1.16 0.62 1.52 ... 0.62 0.51 0.79 0.38 1.09
143 2023 1.17 1.49 0.85 1.78 ... 1.07 1.04 0.91 0.44 0.63

144 rows × 15 columns

Working with missing data (pandas)#

Missing data in pandas#

Missing values, denoting that the true value is unknown, are an inevitable feature of real-world data. In plain-text formats, such as CSV, missing values are commonly just “missing”, i.e., left blank, or marked with a text label such as 'NA'.

Treatment of “No Data” values in Series and DataFrames is derived from the underlying behavior in numpy (Working with missing data (numpy)), as pandas is based on numpy. Namely, “No Data” values are typically represented using np.nan.

pandas may automatically transform the Series data type, and “No Data” representation, for optimized data storage. For example, a Series of type int which contains one or more np.nan value, is automatically transformed to float, similarly to the behaviour we have seen with numpy arrays (Working with missing data (numpy)):

pd.Series([7, 4, 3])
0    7
1    4
2    3
dtype: int64
pd.Series([7, np.nan, 3])
0    7.0
1    NaN
2    3.0
dtype: float64

Moreover, None values are automatically transformed to np.nan (unlike numpy, where None values are preserved, resulting in an object array):

pd.Series([7, None, 3])
0    7.0
1    NaN
2    3.0
dtype: float64

“No Data” values are automatically assigned where necessary when creating a Series or a DataFrame through other methods too. For example, empty cells in a CSV file are assigned with np.nan when importing the CSV into a DataFrame (see Reading from file).

The trees.csv file contains the locations, and other details, about trees in Beer-Sheva:

  • 'plate_num'—The tree ID

  • 'tree_speci'—The tree species

  • 'street_nam', 'neighborho'—Street and neighborhood name, respectively

  • 'itm_x', 'itm_y'—x- and y-coordinates in ITM, respectively

  • 'lon', 'lat'—longitude and latitude, respectively

If you open the CSV file in a spreadsheet program (such as Excel), you will see that some of the cells are empty. Those cells are assigned with np.nan when importing the table:

trees = pd.read_csv('data/trees.csv')
trees
plate_num tree_speci street_nam neighborho itm_x itm_y lat lon
0 0 NaN NaN NaN 178851.0986 573942.6775 31.256378 34.778136
1 0 NaN קדושי יאסי NaN 177190.9032 575006.8485 31.265916 34.760662
2 0 NaN טוביהו דוד NaN 177262.2382 573712.8533 31.254248 34.761465
3 0 NaN NaN NaN 180647.9995 576674.5454 31.281077 34.796895
4 0 NaN NaN NaN 178965.5725 571592.4473 31.235185 34.779433
... ... ... ... ... ... ... ... ...
59566 0 NaN אצ"ל NaN 177777.1526 574971.2819 31.265617 34.766818
59567 0 NaN NaN NaN 178736.2796 573397.7334 31.251459 34.776953
59568 0 NaN NaN NaN 178960.3601 570520.2996 31.225515 34.779421
59569 794 סיגלון עלי מימוסה אלישר חיים משה רמות 182565.0560 575781.1640 31.273082 34.817059
59570 0 NaN NaN NaN 178706.7497 570401.6750 31.224437 34.776764

59571 rows × 8 columns

First, let’s make the column names more convenient:

trees = trees.rename(columns={
    'plate_num': 'id', 
    'tree_speci': 'species', 
    'street_nam': 'street', 
    'neighborho': 'neighborhood'
})
trees
id species street neighborhood itm_x itm_y lat lon
0 0 NaN NaN NaN 178851.0986 573942.6775 31.256378 34.778136
1 0 NaN קדושי יאסי NaN 177190.9032 575006.8485 31.265916 34.760662
2 0 NaN טוביהו דוד NaN 177262.2382 573712.8533 31.254248 34.761465
3 0 NaN NaN NaN 180647.9995 576674.5454 31.281077 34.796895
4 0 NaN NaN NaN 178965.5725 571592.4473 31.235185 34.779433
... ... ... ... ... ... ... ... ...
59566 0 NaN אצ"ל NaN 177777.1526 574971.2819 31.265617 34.766818
59567 0 NaN NaN NaN 178736.2796 573397.7334 31.251459 34.776953
59568 0 NaN NaN NaN 178960.3601 570520.2996 31.225515 34.779421
59569 794 סיגלון עלי מימוסה אלישר חיים משה רמות 182565.0560 575781.1640 31.273082 34.817059
59570 0 NaN NaN NaN 178706.7497 570401.6750 31.224437 34.776764

59571 rows × 8 columns

For example, the species of the first tree is unknown:

trees['species'].iloc[0]
nan

Detecting with pd.isna#

To detect “No Data” values, the .isna method can be applied on a Series or a DataFrame. The result is a boolean mask that marks the missing values. This is analogous to the np.isnan function for arrays, which we learned about earlier (see Detecting “No Data” (np.isnan)).

For example, trees.isna() returns a boolean DataFrame, of the same shape as trees, with True marking “No Data” values:

trees.isna()
id species street neighborhood itm_x itm_y lat lon
0 False True True True False False False False
1 False True False True False False False False
2 False True False True False False False False
3 False True True True False False False False
4 False True True True False False False False
... ... ... ... ... ... ... ... ...
59566 False True False True False False False False
59567 False True True True False False False False
59568 False True True True False False False False
59569 False False False False False False False False
59570 False True True True False False False False

59571 rows × 8 columns

Similarly, the .isna() method applied on a Series returns a new boolean Series where “No Data” values are marked with True:

trees['species'].isna()
0         True
1         True
2         True
3         True
4         True
         ...  
59566     True
59567     True
59568     True
59569    False
59570     True
Name: species, Length: 59571, dtype: bool

The ~ operator (see numpy conditional operators) reverses a boolean Series. This is very useful in the context of “No Data” masks, since it can reverse the mask so that it points at non-missing values.

For example, here is how we can get the reverse of trees.isna():

~trees.isna()
id species street neighborhood itm_x itm_y lat lon
0 True False False False True True True True
1 True False True False True True True True
2 True False True False True True True True
3 True False False False True True True True
4 True False False False True True True True
... ... ... ... ... ... ... ... ...
59566 True False True False True True True True
59567 True False False False True True True True
59568 True False False False True True True True
59569 True True True True True True True True
59570 True False False False True True True True

59571 rows × 8 columns

and here is how we can get the reverse of trees['spedies'].isna():

~trees['species'].isna()
0        False
1        False
2        False
3        False
4        False
         ...  
59566    False
59567    False
59568    False
59569     True
59570    False
Name: species, Length: 59571, dtype: bool

Another possible use of ~ and .isna is to extract the non-missing values from a series. Here is an example:

trees['species'][~trees['species'].isna()]
16                אשל הפרקים
24         סיגלון עלה מימוזה
26                זית אירופי
31                מילה סורית
38            מילה פנסילבנית
                ...         
59552         מילה פנסילבנית
59554    מילה ירוקת עד אובלי
59557         בוהיניה מגוונת
59558          מילה ירוקת עד
59569      סיגלון עלי מימוסה
Name: species, Length: 14866, dtype: object

However, there is a dedicated method called .dropna to do that with shorter code (see Removing with .dropna):

trees['species'].dropna()
16                אשל הפרקים
24         סיגלון עלה מימוזה
26                זית אירופי
31                מילה סורית
38            מילה פנסילבנית
                ...         
59552         מילה פנסילבנית
59554    מילה ירוקת עד אובלי
59557         בוהיניה מגוונת
59558          מילה ירוקת עד
59569      סיגלון עלי מימוסה
Name: species, Length: 14866, dtype: object

Similarly, we can extract DataFrame rows where a particular column is non-missing. For example, here we have those trees rows that have a non-missing 'species' value:

trees[~trees['species'].isna()]
id species street neighborhood itm_x itm_y lat lon
16 3781 אשל הפרקים אבן עזרא אברהם שכונה ד 179643.7211 575520.3612 31.270634 34.786395
24 11181 סיגלון עלה מימוזה מאיר גרוסמן נווה זאב 177779.1452 571815.6028 31.237156 34.766971
26 9863 זית אירופי יהודה אלחריזי נאות לון 177533.4705 573428.4568 31.251693 34.764324
31 1722 מילה סורית שריג נחום רמות הרכס 181497.2680 576827.3730 31.282484 34.805808
38 16510 מילה פנסילבנית NaN שכונה א 179617.3045 572772.0315 31.245846 34.786227
... ... ... ... ... ... ... ... ...
59552 16542 מילה פנסילבנית NaN שכונה א 179430.3376 572819.7487 31.246270 34.784262
59554 7501 מילה ירוקת עד אובלי אברהם אבינו שכונה ד 180421.5505 575430.0755 31.269846 34.794565
59557 11225 בוהיניה מגוונת מישאל דהאן נווה זאב 177619.4177 571312.0208 31.232608 34.765315
59558 9324 מילה ירוקת עד אילן רמון נאות לון 176737.6179 573644.9874 31.253617 34.755960
59569 794 סיגלון עלי מימוסה אלישר חיים משה רמות 182565.0560 575781.1640 31.273082 34.817059

14866 rows × 8 columns

Note

Instead of reversing the output of .isna with the ~operator, another option is to use the inverse method .notna, which returns the non-missing mask.

Finally, it is often useful to combine the .isna method with .any to find out if a Series contains at least one missing value:

trees['species'].isna().any()
True
trees['id'].isna().any()
False

or to find out which columns contain at least one missing value:

trees.isna().any()
id              False
species          True
street           True
neighborhood     True
itm_x           False
itm_y           False
lat             False
lon             False
dtype: bool

We can also use .sum to find out how many missing value are there in each column:

trees.isna().sum()
id                  0
species         44705
street          31678
neighborhood    42121
itm_x               0
itm_y               0
lat                 0
lon                 0
dtype: int64

or .mean to find the proportion of missing values in each column:

trees.isna().mean()
id              0.000000
species         0.750449
street          0.531769
neighborhood    0.707072
itm_x           0.000000
itm_y           0.000000
lat             0.000000
lon             0.000000
dtype: float64

Operators with missing values#

When applying a method such as .sum, .mean, .min, or .max (or the respective np.* function) on a Series, the “No Data” values are ignored. That is, the result is based only on the non-missing values. For example:

trees.loc[trees['species'] == 'סיגלון עלי מימוסה', 'sigalon'] = 1
trees
id species street neighborhood itm_x itm_y lat lon sigalon
0 0 NaN NaN NaN 178851.0986 573942.6775 31.256378 34.778136 NaN
1 0 NaN קדושי יאסי NaN 177190.9032 575006.8485 31.265916 34.760662 NaN
2 0 NaN טוביהו דוד NaN 177262.2382 573712.8533 31.254248 34.761465 NaN
3 0 NaN NaN NaN 180647.9995 576674.5454 31.281077 34.796895 NaN
4 0 NaN NaN NaN 178965.5725 571592.4473 31.235185 34.779433 NaN
... ... ... ... ... ... ... ... ... ...
59566 0 NaN אצ"ל NaN 177777.1526 574971.2819 31.265617 34.766818 NaN
59567 0 NaN NaN NaN 178736.2796 573397.7334 31.251459 34.776953 NaN
59568 0 NaN NaN NaN 178960.3601 570520.2996 31.225515 34.779421 NaN
59569 794 סיגלון עלי מימוסה אלישר חיים משה רמות 182565.0560 575781.1640 31.273082 34.817059 1.0
59570 0 NaN NaN NaN 178706.7497 570401.6750 31.224437 34.776764 NaN

59571 rows × 9 columns

trees['sigalon'].sum()
2055.0

Note that this behavior is specific to pandas, and unlike what we have seen with numpy arrays, where these opertions return “No Data” if at least one of the values in the array is “No Data” (see Operations with “No Data”):

trees['sigalon'].to_numpy().sum()
nan

Element-by-element operations, however, expectedly result in “No Data” when applied to “No Data” elements, since the result cannot be computed at all. For example:

trees['sigalon'] * 2
0        NaN
1        NaN
2        NaN
3        NaN
4        NaN
        ... 
59566    NaN
59567    NaN
59568    NaN
59569    2.0
59570    NaN
Name: sigalon, Length: 59571, dtype: float64

Removing with .dropna#

The .dropna method can be used to remove missing values from a DataFrame (or a Series). The default is to drop all rows that have at least one missing value, i.e., keeping only complete observations. For example:

trees.dropna()
id species street neighborhood itm_x itm_y lat lon sigalon
46 147 סיגלון עלי מימוסה אורי צבי רמות 182514.8560 576183.1820 31.276706 34.816517 1.0
94 6879 סיגלון עלי מימוסה פינקל שמעון נווה מנחם 176749.0495 574848.2124 31.264469 34.756029 1.0
101 8590 סיגלון עלי מימוסה דרך אילת מרכז אזרחי 180686.9014 572493.2768 31.243368 34.797465 1.0
109 750 סיגלון עלי מימוסה אלישר חיים משה רמות 182639.7940 575771.6170 31.272998 34.817845 1.0
118 1697 סיגלון עלי מימוסה האוזנר גדעון רמות הרכס 181421.0310 577302.3120 31.286765 34.804989 1.0
... ... ... ... ... ... ... ... ... ...
59497 6881 סיגלון עלי מימוסה פינקל שמעון נווה מנחם 176769.5141 574839.2052 31.264389 34.756244 1.0
59512 1702 סיגלון עלי מימוסה האוזנר גדעון רמות הרכס 181449.1440 577213.9830 31.285969 34.805287 1.0
59529 217 סיגלון עלי מימוסה הר בוקר רמות 181888.6660 575877.2040 31.273927 34.809953 1.0
59538 417 סיגלון עלי מימוסה אמיר אברהם רמות 183246.8270 576113.8890 31.276104 34.824206 1.0
59569 794 סיגלון עלי מימוסה אלישר חיים משה רמות 182565.0560 575781.1640 31.273082 34.817059 1.0

1930 rows × 9 columns

Note

There are other parameters of .dropna for dropping columns instead of rows, or dropping rows or column using a threshold of “acceptable” minimum number of non-missing values. See the documentation for details.

Replacing with .fillna#

Missing values can be replaced using the .fillna method. The parameter specifies which new value should be placed instead of the “No Data” values. For example, here is how we can replace the “No Data” values in the 'sigalon' column of trees with 0:

trees['sigalon'].fillna(0)
0        0.0
1        0.0
2        0.0
3        0.0
4        0.0
        ... 
59566    0.0
59567    0.0
59568    0.0
59569    1.0
59570    0.0
Name: sigalon, Length: 59571, dtype: float64

or assign the result back to the trees table:

trees['sigalon'] = trees['sigalon'].fillna(0)
trees
id species street neighborhood itm_x itm_y lat lon sigalon
0 0 NaN NaN NaN 178851.0986 573942.6775 31.256378 34.778136 0.0
1 0 NaN קדושי יאסי NaN 177190.9032 575006.8485 31.265916 34.760662 0.0
2 0 NaN טוביהו דוד NaN 177262.2382 573712.8533 31.254248 34.761465 0.0
3 0 NaN NaN NaN 180647.9995 576674.5454 31.281077 34.796895 0.0
4 0 NaN NaN NaN 178965.5725 571592.4473 31.235185 34.779433 0.0
... ... ... ... ... ... ... ... ... ...
59566 0 NaN אצ"ל NaN 177777.1526 574971.2819 31.265617 34.766818 0.0
59567 0 NaN NaN NaN 178736.2796 573397.7334 31.251459 34.776953 0.0
59568 0 NaN NaN NaN 178960.3601 570520.2996 31.225515 34.779421 0.0
59569 794 סיגלון עלי מימוסה אלישר חיים משה רמות 182565.0560 575781.1640 31.273082 34.817059 1.0
59570 0 NaN NaN NaN 178706.7497 570401.6750 31.224437 34.776764 0.0

59571 rows × 9 columns

Let us return to the original trees table:

trees = pd.read_csv('data/trees.csv')
trees
plate_num tree_speci street_nam neighborho itm_x itm_y lat lon
0 0 NaN NaN NaN 178851.0986 573942.6775 31.256378 34.778136
1 0 NaN קדושי יאסי NaN 177190.9032 575006.8485 31.265916 34.760662
2 0 NaN טוביהו דוד NaN 177262.2382 573712.8533 31.254248 34.761465
3 0 NaN NaN NaN 180647.9995 576674.5454 31.281077 34.796895
4 0 NaN NaN NaN 178965.5725 571592.4473 31.235185 34.779433
... ... ... ... ... ... ... ... ...
59566 0 NaN אצ"ל NaN 177777.1526 574971.2819 31.265617 34.766818
59567 0 NaN NaN NaN 178736.2796 573397.7334 31.251459 34.776953
59568 0 NaN NaN NaN 178960.3601 570520.2996 31.225515 34.779421
59569 794 סיגלון עלי מימוסה אלישר חיים משה רמות 182565.0560 575781.1640 31.273082 34.817059
59570 0 NaN NaN NaN 178706.7497 570401.6750 31.224437 34.776764

59571 rows × 8 columns

Views and copies (pandas)#

One more thing we need to be aware when working with pandas is the distinction between views and copies. The pandas package (and geopandas which extends it, see Vector layers (geopandas)) behaves similarly to numpy with respect to copies (see Views and copies (numpy)). We must be aware of whether we are creating a view or a copy of a DataFrame, to avoid unexpected results.

Here is a small demonstration, very similar to the one we did with numpy arrays (see Views and copies (numpy)). Suppose that we create a copy of dat, named dat2:

dat2 = dat

Next, we modify dat2, assigning a new value such as 9999 into a particular cell (e.g., 2nd row, 2nd column):

dat2.iloc[1, 1] = 9999

The printout of the table top-left “corner” demonstrates the value was actually changed:

dat2.iloc[:4, :4]
Year Glob NHem SHem
0 1880 -0.17 -0.28 -0.05
1 1881 9999.00 -0.18 0.00
2 1882 -0.11 -0.21 -0.01
3 1883 -0.17 -0.28 -0.07

What may be surprising, again, is that the original table dat has also changed. We thus demonstrated that dat2 is a reference to dat, rather than an independent copy:

dat.iloc[:4, :4]
Year Glob NHem SHem
0 1880 -0.17 -0.28 -0.05
1 1881 9999.00 -0.18 0.00
2 1882 -0.11 -0.21 -0.01
3 1883 -0.17 -0.28 -0.07

To create an independent copy, so that its modifications are not reflected in the original, we need to explicitly use the .copy method, as in:

dat2 = dat.copy()

Note

The behavior described in this section can be disabled with pd.options.mode.copy_on_write=True, and will be disabled by default in pandas version 3.0. When disabled, every pandas operation will create an independent copy, and, for example, it would be impossible to update two objects at once as done above.

Let us read the CSV file again to get back to the original values:

dat = pd.read_csv('data/ZonAnn.Ts+dSST.csv')

Writing DataFrame to file#

A DataFrame can be exported to several formats, such as CSV or Excel, using the appropriate method. For example, exporting to CSV is done using the .to_csv method.

Here is how we can export the DataFrame named stations, which we created in the beginning of the chapter (see Creating a DataFrame), to a CSV file named stations.csv in the output directory. The additional index=False argument specifies that we do not want to export the index values, which is usually the case:

stations.to_csv('output/stations.csv', index=False)

Note that the file is exported to a sub-directory named output (which must already exist!).

The way the resulting file stations.csv is displayed in a spreadsheet program, such as Microsoft Excel or LibreOffice Calc, is shown in Fig. 28.

_images/exported_csv.png

Fig. 28 The DataFrame named stations when exported to CSV file and opened in LibreOffice Calc#

More exercises#

Exercise 05-g

  • Import the file trees.csv into a DataFrame.

  • How many observations with non-missing 'species' are there? (answer: 14866)

  • How many observations with non-missing 'street' and non-missing 'neighborhood' are there? (answer: 12450)

  • How many unique species names are there (excluding np.nan)? (answer: 130)

Exercise solutions#

Exercise 05-d#

import pandas as pd
dat = pd.read_csv('data/ZonAnn.Ts+dSST.csv')
dat['Year'].iloc[dat.shape[0]-1]
2023

Exercise 05-g#

import pandas as pd
trees = pd.read_csv('data/trees.csv')
trees = trees.rename(columns={
    'plate_num': 'id', 
    'tree_speci': 'species', 
    'street_nam': 'street', 
    'neighborho': 'neighborhood'
})
trees['species'].notna().sum()
14866
(trees['street'].notna() & trees['neighborhood'].notna()).sum()
12450
trees['species'].nunique()
130