Skip to content

Latest commit

 

History

History
401 lines (291 loc) · 12.8 KB

gotchas.rst

File metadata and controls

401 lines (291 loc) · 12.8 KB

{{ header }}

Frequently Asked Questions (FAQ)

DataFrame memory usage

The memory usage of a :class:`DataFrame` (including the index) is shown when calling the :meth:`~DataFrame.info`. A configuration option, display.memory_usage (see :ref:`the list of options <options.available>`), specifies if the :class:`DataFrame` memory usage will be displayed when invoking the df.info() method.

For example, the memory usage of the :class:`DataFrame` below is shown when calling :meth:`~DataFrame.info`:

.. ipython:: python

    dtypes = [
        "int64",
        "float64",
        "datetime64[ns]",
        "timedelta64[ns]",
        "complex128",
        "object",
        "bool",
    ]
    n = 5000
    data = {t: np.random.randint(100, size=n).astype(t) for t in dtypes}
    df = pd.DataFrame(data)
    df["categorical"] = df["object"].astype("category")

    df.info()

The + symbol indicates that the true memory usage could be higher, because pandas does not count the memory used by values in columns with dtype=object.

Passing memory_usage='deep' will enable a more accurate memory usage report, accounting for the full usage of the contained objects. This is optional as it can be expensive to do this deeper introspection.

.. ipython:: python

   df.info(memory_usage="deep")

By default the display option is set to True but can be explicitly overridden by passing the memory_usage argument when invoking df.info().

The memory usage of each column can be found by calling the :meth:`~DataFrame.memory_usage` method. This returns a :class:`Series` with an index represented by column names and memory usage of each column shown in bytes. For the :class:`DataFrame` above, the memory usage of each column and the total memory usage can be found with the memory_usage method:

.. ipython:: python

    df.memory_usage()

    # total memory usage of dataframe
    df.memory_usage().sum()

By default the memory usage of the :class:`DataFrame` index is shown in the returned :class:`Series`, the memory usage of the index can be suppressed by passing the index=False argument:

.. ipython:: python

    df.memory_usage(index=False)

The memory usage displayed by the :meth:`~DataFrame.info` method utilizes the :meth:`~DataFrame.memory_usage` method to determine the memory usage of a :class:`DataFrame` while also formatting the output in human-readable units (base-2 representation; i.e. 1KB = 1024 bytes).

See also :ref:`Categorical Memory Usage <categorical.memory>`.

Using if/truth statements with pandas

pandas follows the NumPy convention of raising an error when you try to convert something to a bool. This happens in an if-statement or when using the boolean operations: and, or, and not. It is not clear what the result of the following code should be:

>>> if pd.Series([False, True, False]):
...     pass

Should it be True because it's not zero-length, or False because there are False values? It is unclear, so instead, pandas raises a ValueError:

.. ipython:: python
    :okexcept:

    if pd.Series([False, True, False]):
        print("I was true")

You need to explicitly choose what you want to do with the :class:`DataFrame`, e.g. use :meth:`~DataFrame.any`, :meth:`~DataFrame.all` or :meth:`~DataFrame.empty`. Alternatively, you might want to compare if the pandas object is None:

.. ipython:: python

    if pd.Series([False, True, False]) is not None:
        print("I was not None")


Below is how to check if any of the values are True:

.. ipython:: python

    if pd.Series([False, True, False]).any():
        print("I am any")

Bitwise boolean

Bitwise boolean operators like == and != return a boolean :class:`Series` which performs an element-wise comparison when compared to a scalar.

.. ipython:: python

   s = pd.Series(range(5))
   s == 4

See :ref:`boolean comparisons<basics.compare>` for more examples.

Using the in operator

Using the Python in operator on a :class:`Series` tests for membership in the index, not membership among the values.

.. ipython:: python

    s = pd.Series(range(5), index=list("abcde"))
    2 in s
    'b' in s

If this behavior is surprising, keep in mind that using in on a Python dictionary tests keys, not values, and :class:`Series` are dict-like. To test for membership in the values, use the method :meth:`~pandas.Series.isin`:

.. ipython:: python

    s.isin([2])
    s.isin([2]).any()

For :class:`DataFrame`, likewise, in applies to the column axis, testing for membership in the list of column names.

Mutating with User Defined Function (UDF) methods

This section applies to pandas methods that take a UDF. In particular, the methods .apply, .aggregate, .transform, and .filter.

It is a general rule in programming that one should not mutate a container while it is being iterated over. Mutation will invalidate the iterator, causing unexpected behavior. Consider the example:

.. ipython:: python

   values = [0, 1, 2, 3, 4, 5]
   n_removed = 0
   for k, value in enumerate(values):
       idx = k - n_removed
       if value % 2 == 1:
           del values[idx]
           n_removed += 1
       else:
           values[idx] = value + 1
   values

One probably would have expected that the result would be [1, 3, 5]. When using a pandas method that takes a UDF, internally pandas is often iterating over the :class:`DataFrame` or other pandas object. Therefore, if the UDF mutates (changes) the :class:`DataFrame`, unexpected behavior can arise.

Here is a similar example with :meth:`DataFrame.apply`:

.. ipython:: python

   def f(s):
       s.pop("a")
       return s

   df = pd.DataFrame({"a": [1, 2, 3], "b": [4, 5, 6]})
   try:
       df.apply(f, axis="columns")
   except Exception as err:
       print(repr(err))

To resolve this issue, one can make a copy so that the mutation does not apply to the container being iterated over.

.. ipython:: python

   values = [0, 1, 2, 3, 4, 5]
   n_removed = 0
   for k, value in enumerate(values.copy()):
       idx = k - n_removed
       if value % 2 == 1:
           del values[idx]
           n_removed += 1
       else:
           values[idx] = value + 1
   values

.. ipython:: python

   def f(s):
       s = s.copy()
       s.pop("a")
       return s

   df = pd.DataFrame({"a": [1, 2, 3], 'b': [4, 5, 6]})
   df.apply(f, axis="columns")

NaN, Integer NA values and NA type promotions

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 is there or is missing.
  • 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 :meth:`DataFrame.isna` and :meth:`DataFrame.notna` 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.

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:

.. ipython:: python

   s = pd.Series([1, 2, 3, 4, 5], index=list("abcde"))
   s
   s.dtype

   s2 = s.reindex(["a", "b", "c", "f", "u"])
   s2
   s2.dtype

This trade-off is made largely for memory and performance reasons, and also so that the resulting :class:`Series` continues to be "numeric".

If you need to represent integers with possibly missing values, use one of the nullable-integer extension dtypes provided by pandas

.. ipython:: python

   s_int = pd.Series([1, 2, 3, 4, 5], index=list("abcde"), dtype=pd.Int64Dtype())
   s_int
   s_int.dtype

   s2_int = s_int.reindex(["a", "b", "c", "f", "u"])
   s2_int
   s2_int.dtype

See :ref:`integer_na` for more.

NA type promotions

When introducing NAs into an existing :class:`Series` or :class:`DataFrame` via :meth:`~Series.reindex` or some other means, boolean and integer types will be promoted to a different dtype in order to store the NAs. The promotions are summarized in 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 i.e. storing values greater than 2**53. Some explanation for the motivation is in the next section.

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.

Differences with NumPy

For :class:`Series` and :class:`DataFrame` objects, :meth:`~DataFrame.var` normalizes by N-1 to produce unbiased estimates of the population variance, while NumPy's :meth:`numpy.var` normalizes by N, which measures the variance of the sample. Note that :meth:`~DataFrame.cov` normalizes by N-1 in both pandas and NumPy.

Thread-safety

pandas is not 100% thread safe. The known issues relate to the :meth:`~DataFrame.copy` method. If you are doing a lot of copying of :class:`DataFrame` objects shared among threads, we recommend holding locks inside the threads where the data copying occurs.

See this link for more information.

Byte-ordering issues

Occasionally you may have to deal with data that were created on a machine with a different byte order than the one on which you are running Python. A common symptom of this issue is an error like:

Traceback
    ...
ValueError: Big-endian buffer not supported on little-endian compiler

To deal with this issue you should convert the underlying NumPy array to the native system byte order before passing it to :class:`Series` or :class:`DataFrame` constructors using something similar to the following:

.. ipython:: python

   x = np.array(list(range(10)), ">i4")  # big endian
   newx = x.byteswap().newbyteorder()  # force native byteorder
   s = pd.Series(newx)

See the NumPy documentation on byte order for more details.