1.2 NaN
, Integer NA
values and NA
type promotions
1.2.1 Choice of NA
representation
For lack of NA
(missing) support from the ground up in NumPy and Python in
general, we were given the difficult choice between either
- A masked array solution: an array of data and an array of boolean values indicating whether a value
- Using a special sentinel value, bit pattern, or set of sentinel values to
denote
NA
across the dtypes
For many reasons we chose the latter. After years of production use it has
proven, at least in my opinion, to be the best decision given the state of
affairs in NumPy and Python in general. The special value NaN
(Not-A-Number) is used everywhere as the NA
value, and there are API
functions isnull
and notnull
which can be used across the dtypes to
detect NA values.
However, it comes with it a couple of trade-offs which I most certainly have not ignored.
1.2.2 Support for integer NA
In the absence of high performance NA
support being built into NumPy from
the ground up, the primary casualty is the ability to represent NAs in integer
arrays. For example:
In [1]: s = pd.Series([1, 2, 3, 4, 5], index=list('abcde'))
In [2]: s
Out[2]:
a 1
b 2
c 3
d 4
e 5
dtype: int64
In [3]: s.dtype
Out[3]: dtype('int64')
In [4]: s2 = s.reindex(['a', 'b', 'c', 'f', 'u'])
In [5]: s2
Out[5]:
a 1.0
b 2.0
c 3.0
f NaN
u NaN
dtype: float64
In [6]: s2.dtype
Out[6]: dtype('float64')
This trade-off is made largely for memory and performance reasons, and also so
that the resulting Series continues to be “numeric”. One possibility is to use
dtype=object
arrays instead.
1.2.3 NA
type promotions
When introducing NAs into an existing Series or DataFrame via reindex
or
some other means, boolean and integer types will be promoted to a different
dtype in order to store the NAs. These are summarized by this table:
Typeclass | Promotion dtype for storing NAs |
---|---|
floating |
no change |
object |
no change |
integer |
cast to float64 |
boolean |
cast to object |
While this may seem like a heavy trade-off, I have found very few cases where this is an issue in practice. Some explanation for the motivation here in the next section.
1.2.4 Why not make NumPy like R?
Many people have suggested that NumPy should simply emulate the NA
support
present in the more domain-specific statistical programming language R. Part of the reason is the NumPy type hierarchy:
Typeclass | Dtypes |
---|---|
numpy.floating |
float16, float32, float64, float128 |
numpy.integer |
int8, int16, int32, int64 |
numpy.unsignedinteger |
uint8, uint16, uint32, uint64 |
numpy.object_ |
object_ |
numpy.bool_ |
bool_ |
numpy.character |
string_, unicode_ |
The R language, by contrast, only has a handful of built-in data types:
integer
, numeric
(floating-point), character
, and
boolean
. NA
types are implemented by reserving special bit patterns for
each type to be used as the missing value. While doing this with the full NumPy
type hierarchy would be possible, it would be a more substantial trade-off
(especially for the 8- and 16-bit data types) and implementation undertaking.
An alternate approach is that of using masked arrays. A masked array is an
array of data with an associated boolean mask denoting whether each value
should be considered NA
or not. I am personally not in love with this
approach as I feel that overall it places a fairly heavy burden on the user and
the library implementer. Additionally, it exacts a fairly high performance cost
when working with numerical data compared with the simple approach of using
NaN
. Thus, I have chosen the Pythonic “practicality beats purity” approach
and traded integer NA
capability for a much simpler approach of using a
special value in float and object arrays to denote NA
, and promoting
integer arrays to floating when NAs must be introduced.