Table Of Contents

Search

Enter search terms or a module, class or function name.

Essential Basic Functionality

Here we discuss a lot of the essential functionality common to the pandas data structures. Here’s how to create some of the objects used in the examples from the previous section:

In [131]: index = date_range('1/1/2000', periods=8)

In [132]: s = Series(randn(5), index=['a', 'b', 'c', 'd', 'e'])

In [133]: df = DataFrame(randn(8, 3), index=index,
   .....:                columns=['A', 'B', 'C'])
   .....:

In [134]: wp = Panel(randn(2, 5, 4), items=['Item1', 'Item2'],
   .....:            major_axis=date_range('1/1/2000', periods=5),
   .....:            minor_axis=['A', 'B', 'C', 'D'])
   .....:

Head and Tail

To view a small sample of a Series or DataFrame object, use the head and tail methods. The default number of elements to display is five, but you may pass a custom number.

In [135]: long_series = Series(randn(1000))

In [136]: long_series.head()
Out[136]: 
0   -0.199038
1    1.095864
2   -0.200875
3    0.162291
4   -0.430489
dtype: float64

In [137]: long_series.tail(3)
Out[137]: 
997   -1.198693
998    1.238029
999   -1.344716
dtype: float64

Attributes and the raw ndarray(s)

pandas objects have a number of attributes enabling you to access the metadata

  • shape: gives the axis dimensions of the object, consistent with ndarray
  • Axis labels
    • Series: index (only axis)
    • DataFrame: index (rows) and columns
    • Panel: items, major_axis, and minor_axis

Note, these attributes can be safely assigned to!

In [138]: df[:2]
Out[138]: 
                   A         B         C
2000-01-01  0.232465 -0.789552 -0.364308
2000-01-02 -0.534541  0.822239 -0.443109

In [139]: df.columns = [x.lower() for x in df.columns]

In [140]: df
Out[140]: 
                   a         b         c
2000-01-01  0.232465 -0.789552 -0.364308
2000-01-02 -0.534541  0.822239 -0.443109
2000-01-03 -2.119990 -0.460149  1.813962
2000-01-04 -1.053571  0.009412 -0.165966
2000-01-05 -0.848662 -0.495553 -0.176421
2000-01-06 -0.423595 -1.035433 -1.035374
2000-01-07 -2.369079  0.524408 -0.871120
2000-01-08  1.585433  0.039501  2.274101

To get the actual data inside a data structure, one need only access the values property:

In [141]: s.values
Out[141]: array([ 1.1292,  0.2313, -0.1847, -0.1386, -0.9243])

In [142]: df.values
Out[142]: 
array([[ 0.2325, -0.7896, -0.3643],
       [-0.5345,  0.8222, -0.4431],
       [-2.12  , -0.4601,  1.814 ],
       [-1.0536,  0.0094, -0.166 ],
       [-0.8487, -0.4956, -0.1764],
       [-0.4236, -1.0354, -1.0354],
       [-2.3691,  0.5244, -0.8711],
       [ 1.5854,  0.0395,  2.2741]])

In [143]: wp.values
Out[143]: 
array([[[-1.1181,  0.4313,  0.5547, -1.3336],
        [-0.3322, -0.4859,  1.7259,  1.7993],
        [-0.9689, -0.7795, -2.0007, -1.8666],
        [-1.1013,  1.9575,  0.0589,  0.7581],
        [ 0.0766, -0.5485, -0.1605, -0.3778]],
       [[ 0.2499, -0.3413, -0.2726, -0.2774],
        [-1.1029,  0.1003, -1.6028,  0.9201],
        [-0.6439,  0.0603, -0.4349, -0.4943],
        [ 0.738 ,  0.4516,  0.3341, -0.7871],
        [ 0.6514, -0.7419,  1.1939, -2.3958]]])

If a DataFrame or Panel contains homogeneously-typed data, the ndarray can actually be modified in-place, and the changes will be reflected in the data structure. For heterogeneous data (e.g. some of the DataFrame’s columns are not all the same dtype), this will not be the case. The values attribute itself, unlike the axis labels, cannot be assigned to.

Note

When working with heterogeneous data, the dtype of the resulting ndarray will be chosen to accommodate all of the data involved. For example, if strings are involved, the result will be of object dtype. If there are only floats and integers, the resulting array will be of float dtype.

Accelerated operations

Pandas has support for accelerating certain types of binary numerical and boolean operations using the numexpr library (starting in 0.11.0) and the bottleneck libraries.

These libraries are especially useful when dealing with large data sets, and provide large speedups. numexpr uses smart chunking, caching, and multiple cores. bottleneck is a set of specialized cython routines that are especially fast when dealing with arrays that have nans.

Here is a sample (using 100 column x 100,000 row DataFrames):

Operation 0.11.0 (ms) Prior Vern (ms) Ratio to Prior
df1 > df2 13.32 125.35 0.1063
df1 * df2 21.71 36.63 0.5928
df1 + df2 22.04 36.50 0.6039

You are highly encouraged to install both libraries. See the section Recommended Dependencies for more installation info.

Flexible binary operations

With binary operations between pandas data structures, there are two key points of interest:

  • Broadcasting behavior between higher- (e.g. DataFrame) and lower-dimensional (e.g. Series) objects.
  • Missing data in computations

We will demonstrate how to manage these issues independently, though they can be handled simultaneously.

Matching / broadcasting behavior

DataFrame has the methods add, sub, mul, div and related functions radd, rsub, ... for carrying out binary operations. For broadcasting behavior, Series input is of primary interest. Using these functions, you can use to either match on the index or columns via the axis keyword:

In [144]: d = {'one' : Series(randn(3), index=['a', 'b', 'c']),
   .....:      'two' : Series(randn(4), index=['a', 'b', 'c', 'd']),
   .....:      'three' : Series(randn(3), index=['b', 'c', 'd'])}
   .....:

In [145]: df = df_orig = DataFrame(d)

In [146]: df
Out[146]: 
        one     three       two
a -0.701368       NaN -0.087103
b  0.109333 -0.354359  0.637674
c -0.231617 -0.148387 -0.002666
d       NaN -0.167407  0.104044

In [147]: row = df.ix[1]

In [148]: column = df['two']

In [149]: df.sub(row, axis='columns')
Out[149]: 
        one     three       two
a -0.810701       NaN -0.724777
b  0.000000  0.000000  0.000000
c -0.340950  0.205973 -0.640340
d       NaN  0.186952 -0.533630

In [150]: df.sub(row, axis=1)
Out[150]: 
        one     three       two
a -0.810701       NaN -0.724777
b  0.000000  0.000000  0.000000
c -0.340950  0.205973 -0.640340
d       NaN  0.186952 -0.533630

In [151]: df.sub(column, axis='index')
Out[151]: 
        one     three  two
a -0.614265       NaN    0
b -0.528341 -0.992033    0
c -0.228950 -0.145720    0
d       NaN -0.271451    0

In [152]: df.sub(column, axis=0)
Out[152]: 
        one     three  two
a -0.614265       NaN    0
b -0.528341 -0.992033    0
c -0.228950 -0.145720    0
d       NaN -0.271451    0

With Panel, describing the matching behavior is a bit more difficult, so the arithmetic methods instead (and perhaps confusingly?) give you the option to specify the broadcast axis. For example, suppose we wished to demean the data over a particular axis. This can be accomplished by taking the mean over an axis and broadcasting over the same axis:

In [153]: major_mean = wp.mean(axis='major')

In [154]: major_mean
Out[154]: 
      Item1     Item2
A -0.688773 -0.021497
B  0.114982 -0.094183
C  0.035674 -0.156470
D -0.204142 -0.606887

In [155]: wp.sub(major_mean, axis='major')
Out[155]: 
<class 'pandas.core.panel.Panel'>
Dimensions: 2 (items) x 5 (major_axis) x 4 (minor_axis)
Items axis: Item1 to Item2
Major_axis axis: 2000-01-01 00:00:00 to 2000-01-05 00:00:00
Minor_axis axis: A to D

And similarly for axis="items" and axis="minor".

Note

I could be convinced to make the axis argument in the DataFrame methods match the broadcasting behavior of Panel. Though it would require a transition period so users can change their code...

Missing data / operations with fill values

In Series and DataFrame (though not yet in Panel), the arithmetic functions have the option of inputting a fill_value, namely a value to substitute when at most one of the values at a location are missing. For example, when adding two DataFrame objects, you may wish to treat NaN as 0 unless both DataFrames are missing that value, in which case the result will be NaN (you can later replace NaN with some other value using fillna if you wish).

In [156]: df
Out[156]: 
        one     three       two
a -0.701368       NaN -0.087103
b  0.109333 -0.354359  0.637674
c -0.231617 -0.148387 -0.002666
d       NaN -0.167407  0.104044

In [157]: df2
Out[157]: 
        one     three       two
a -0.701368  1.000000 -0.087103
b  0.109333 -0.354359  0.637674
c -0.231617 -0.148387 -0.002666
d       NaN -0.167407  0.104044

In [158]: df + df2
Out[158]: 
        one     three       two
a -1.402736       NaN -0.174206
b  0.218666 -0.708719  1.275347
c -0.463233 -0.296773 -0.005333
d       NaN -0.334814  0.208088

In [159]: df.add(df2, fill_value=0)
Out[159]: 
        one     three       two
a -1.402736  1.000000 -0.174206
b  0.218666 -0.708719  1.275347
c -0.463233 -0.296773 -0.005333
d       NaN -0.334814  0.208088

Flexible Comparisons

Starting in v0.8, pandas introduced binary comparison methods eq, ne, lt, gt, le, and ge to Series and DataFrame whose behavior is analogous to the binary arithmetic operations described above:

In [160]: df.gt(df2)
Out[160]: 
     one  three    two
a  False  False  False
b  False  False  False
c  False  False  False
d  False  False  False

In [161]: df2.ne(df)
Out[161]: 
     one  three    two
a  False   True  False
b  False  False  False
c  False  False  False
d   True  False  False

Combining overlapping data sets

A problem occasionally arising is the combination of two similar data sets where values in one are preferred over the other. An example would be two data series representing a particular economic indicator where one is considered to be of “higher quality”. However, the lower quality series might extend further back in history or have more complete data coverage. As such, we would like to combine two DataFrame objects where missing values in one DataFrame are conditionally filled with like-labeled values from the other DataFrame. The function implementing this operation is combine_first, which we illustrate:

In [162]: df1 = DataFrame({'A' : [1., np.nan, 3., 5., np.nan],
   .....:                  'B' : [np.nan, 2., 3., np.nan, 6.]})
   .....:

In [163]: df2 = DataFrame({'A' : [5., 2., 4., np.nan, 3., 7.],
   .....:                  'B' : [np.nan, np.nan, 3., 4., 6., 8.]})
   .....:

In [164]: df1
Out[164]: 
    A   B
0   1 NaN
1 NaN   2
2   3   3
3   5 NaN
4 NaN   6

In [165]: df2
Out[165]: 
    A   B
0   5 NaN
1   2 NaN
2   4   3
3 NaN   4
4   3   6
5   7   8

In [166]: df1.combine_first(df2)
Out[166]: 
   A   B
0  1 NaN
1  2   2
2  3   3
3  5   4
4  3   6
5  7   8

General DataFrame Combine

The combine_first method above calls the more general DataFrame method combine. This method takes another DataFrame and a combiner function, aligns the input DataFrame and then passes the combiner function pairs of Series (ie, columns whose names are the same).

So, for instance, to reproduce combine_first as above:

In [167]: combiner = lambda x, y: np.where(isnull(x), y, x)

In [168]: df1.combine(df2, combiner)
Out[168]: 
   A   B
0  1 NaN
1  2   2
2  3   3
3  5   4
4  3   6
5  7   8

Descriptive statistics

A large number of methods for computing descriptive statistics and other related operations on Series, DataFrame, and Panel. Most of these are aggregations (hence producing a lower-dimensional result) like sum, mean, and quantile, but some of them, like cumsum and cumprod, produce an object of the same size. Generally speaking, these methods take an axis argument, just like ndarray.{sum, std, ...}, but the axis can be specified by name or integer:

  • Series: no axis argument needed
  • DataFrame: “index” (axis=0, default), “columns” (axis=1)
  • Panel: “items” (axis=0), “major” (axis=1, default), “minor” (axis=2)

For example:

In [169]: df
Out[169]: 
        one     three       two
a -0.701368       NaN -0.087103
b  0.109333 -0.354359  0.637674
c -0.231617 -0.148387 -0.002666
d       NaN -0.167407  0.104044

In [170]: df.mean(0)
Out[170]: 
one     -0.274551
three   -0.223384
two      0.162987
dtype: float64

In [171]: df.mean(1)
Out[171]: 
a   -0.394235
b    0.130882
c   -0.127557
d   -0.031682
dtype: float64

All such methods have a skipna option signaling whether to exclude missing data (True by default):

In [172]: df.sum(0, skipna=False)
Out[172]: 
one           NaN
three         NaN
two      0.651948
dtype: float64

In [173]: df.sum(axis=1, skipna=True)
Out[173]: 
a   -0.788471
b    0.392647
c   -0.382670
d   -0.063363
dtype: float64

Combined with the broadcasting / arithmetic behavior, one can describe various statistical procedures, like standardization (rendering data zero mean and standard deviation 1), very concisely:

In [174]: ts_stand = (df - df.mean()) / df.std()

In [175]: ts_stand.std()
Out[175]: 
one      1
three    1
two      1
dtype: float64

In [176]: xs_stand = df.sub(df.mean(1), axis=0).div(df.std(1), axis=0)

In [177]: xs_stand.std(1)
Out[177]: 
a    1
b    1
c    1
d    1
dtype: float64

Note that methods like cumsum and cumprod preserve the location of NA values:

In [178]: df.cumsum()
Out[178]: 
        one     three       two
a -0.701368       NaN -0.087103
b -0.592035 -0.354359  0.550570
c -0.823652 -0.502746  0.547904
d       NaN -0.670153  0.651948

Here is a quick reference summary table of common functions. Each also takes an optional level parameter which applies only if the object has a hierarchical index.

Function Description
count Number of non-null observations
sum Sum of values
mean Mean of values
mad Mean absolute deviation
median Arithmetic median of values
min Minimum
max Maximum
abs Absolute Value
prod Product of values
std Unbiased standard deviation
var Unbiased variance
skew Unbiased skewness (3rd moment)
kurt Unbiased kurtosis (4th moment)
quantile Sample quantile (value at %)
cumsum Cumulative sum
cumprod Cumulative product
cummax Cumulative maximum
cummin Cumulative minimum

Note that by chance some NumPy methods, like mean, std, and sum, will exclude NAs on Series input by default:

In [179]: np.mean(df['one'])
Out[179]: -0.27455055654271204

In [180]: np.mean(df['one'].values)
Out[180]: nan

Series also has a method nunique which will return the number of unique non-null values:

In [181]: series = Series(randn(500))

In [182]: series[20:500] = np.nan

In [183]: series[10:20]  = 5

In [184]: series.nunique()
Out[184]: 11

Summarizing data: describe

There is a convenient describe function which computes a variety of summary statistics about a Series or the columns of a DataFrame (excluding NAs of course):

In [185]: series = Series(randn(1000))

In [186]: series[::2] = np.nan

In [187]: series.describe()
Out[187]: 
count    500.000000
mean      -0.019898
std        1.019180
min       -2.628792
25%       -0.649795
50%       -0.059405
75%        0.651932
max        3.240991
dtype: float64

In [188]: frame = DataFrame(randn(1000, 5), columns=['a', 'b', 'c', 'd', 'e'])

In [189]: frame.ix[::2] = np.nan

In [190]: frame.describe()
Out[190]: 
                a           b           c           d           e
count  500.000000  500.000000  500.000000  500.000000  500.000000
mean     0.051388    0.053476   -0.035612    0.015388    0.057804
std      0.989217    0.995961    0.977047    0.968385    1.022528
min     -3.224136   -2.606460   -2.762875   -2.961757   -2.829100
25%     -0.657420   -0.597123   -0.688961   -0.695019   -0.738097
50%      0.042928    0.018837   -0.071830   -0.011326    0.073287
75%      0.702445    0.693542    0.600454    0.680924    0.807670
max      3.034008    3.104512    2.812028    2.623914    3.542846

For a non-numerical Series object, describe will give a simple summary of the number of unique values and most frequently occurring values:

In [191]: s = Series(['a', 'a', 'b', 'b', 'a', 'a', np.nan, 'c', 'd', 'a'])

In [192]: s.describe()
Out[192]: 
count     9
unique    4
top       a
freq      5
dtype: object

There also is a utility function, value_range which takes a DataFrame and returns a series with the minimum/maximum values in the DataFrame.

Index of Min/Max Values

The idxmin and idxmax functions on Series and DataFrame compute the index labels with the minimum and maximum corresponding values:

In [193]: s1 = Series(randn(5))

In [194]: s1
Out[194]: 
0   -0.574018
1    0.668292
2    0.303418
3   -1.190271
4    0.138399
dtype: float64

In [195]: s1.idxmin(), s1.idxmax()
Out[195]: (3, 1)

In [196]: df1 = DataFrame(randn(5,3), columns=['A','B','C'])

In [197]: df1
Out[197]: 
          A         B         C
0 -0.184355 -1.054354 -1.613138
1 -0.050807 -2.130168 -1.852271
2  0.455674  2.571061 -1.152538
3 -1.638940 -0.364831 -0.348520
4  0.202856  0.777088 -0.358316

In [198]: df1.idxmin(axis=0)
Out[198]: 
A    3
B    1
C    1
dtype: int64

In [199]: df1.idxmax(axis=1)
Out[199]: 
0    A
1    A
2    B
3    C
4    B
dtype: object

When there are multiple rows (or columns) matching the minimum or maximum value, idxmin and idxmax return the first matching index:

In [200]: df3 = DataFrame([2, 1, 1, 3, np.nan], columns=['A'], index=list('edcba'))

In [201]: df3
Out[201]: 
    A
e   2
d   1
c   1
b   3
a NaN

In [202]: df3['A'].idxmin()
Out[202]: 'd'

Value counts (histogramming)

The value_counts Series method and top-level function computes a histogram of a 1D array of values. It can also be used as a function on regular arrays:

In [203]: data = np.random.randint(0, 7, size=50)

In [204]: data
Out[204]: 
array([4, 6, 6, 1, 2, 1, 0, 5, 3, 2, 4, 3, 1, 3, 5, 3, 0, 0, 4, 4, 6, 1, 0,
       4, 3, 2, 1, 3, 1, 5, 6, 3, 1, 2, 4, 4, 3, 3, 2, 2, 2, 3, 2, 3, 0, 1,
       2, 4, 5, 5])

In [205]: s = Series(data)

In [206]: s.value_counts()
Out[206]: 
3    11
2     9
4     8
1     8
5     5
0     5
6     4
dtype: int64

In [207]: value_counts(data)
Out[207]: 
3    11
2     9
4     8
1     8
5     5
0     5
6     4
dtype: int64

Discretization and quantiling

Continuous values can be discretized using the cut (bins based on values) and qcut (bins based on sample quantiles) functions:

In [208]: arr = np.random.randn(20)

In [209]: factor = cut(arr, 4)

In [210]: factor
Out[210]: 
Categorical: 
array(['(-0.837, -0.0162]', '(-1.658, -0.837]', '(-2.483, -1.658]',
       '(-1.658, -0.837]', '(-0.837, -0.0162]', '(-0.0162, 0.805]',
       '(-2.483, -1.658]', '(-0.0162, 0.805]', '(-0.0162, 0.805]',
       '(-0.0162, 0.805]', '(-1.658, -0.837]', '(-0.837, -0.0162]',
       '(-1.658, -0.837]', '(-0.837, -0.0162]', '(-0.0162, 0.805]',
       '(-0.837, -0.0162]', '(-0.837, -0.0162]', '(-0.837, -0.0162]',
       '(-0.0162, 0.805]', '(-0.837, -0.0162]'], dtype=object)
Levels (4): Index(['(-2.483, -1.658]', '(-1.658, -0.837]',
                   '(-0.837, -0.0162]', '(-0.0162, 0.805]'], dtype=object)

In [211]: factor = cut(arr, [-5, -1, 0, 1, 5])

In [212]: factor
Out[212]: 
Categorical: 
array(['(-1, 0]', '(-5, -1]', '(-5, -1]', '(-5, -1]', '(-1, 0]', '(0, 1]',
       '(-5, -1]', '(0, 1]', '(0, 1]', '(0, 1]', '(-1, 0]', '(-1, 0]',
       '(-5, -1]', '(-1, 0]', '(0, 1]', '(-1, 0]', '(-1, 0]', '(-1, 0]',
       '(0, 1]', '(-1, 0]'], dtype=object)
Levels (4): Index(['(-5, -1]', '(-1, 0]', '(0, 1]', '(1, 5]'], dtype=object)

qcut computes sample quantiles. For example, we could slice up some normally distributed data into equal-size quartiles like so:

In [213]: arr = np.random.randn(30)

In [214]: factor = qcut(arr, [0, .25, .5, .75, 1])

In [215]: factor
Out[215]: 
Categorical: 
array(['[-2.891, -0.868]', '(0.525, 3.19]', '(-0.868, -0.0118]',
       '(-0.0118, 0.525]', '(-0.0118, 0.525]', '(0.525, 3.19]',
       '(-0.868, -0.0118]', '[-2.891, -0.868]', '(-0.868, -0.0118]',
       '(0.525, 3.19]', '[-2.891, -0.868]', '(-0.0118, 0.525]',
       '(-0.0118, 0.525]', '(-0.868, -0.0118]', '(0.525, 3.19]',
       '(0.525, 3.19]', '(-0.868, -0.0118]', '[-2.891, -0.868]',
       '(-0.0118, 0.525]', '[-2.891, -0.868]', '[-2.891, -0.868]',
       '[-2.891, -0.868]', '(-0.0118, 0.525]', '(0.525, 3.19]',
       '(-0.868, -0.0118]', '(-0.0118, 0.525]', '[-2.891, -0.868]',
       '(-0.868, -0.0118]', '(0.525, 3.19]', '(0.525, 3.19]'], dtype=object)
Levels (4): Index(['[-2.891, -0.868]', '(-0.868, -0.0118]',
                   '(-0.0118, 0.525]', '(0.525, 3.19]'], dtype=object)

In [216]: value_counts(factor)
Out[216]: 
[-2.891, -0.868]     8
(0.525, 3.19]        8
(-0.868, -0.0118]    7
(-0.0118, 0.525]     7
dtype: int64

Function application

Arbitrary functions can be applied along the axes of a DataFrame or Panel using the apply method, which, like the descriptive statistics methods, take an optional axis argument:

In [217]: df.apply(np.mean)
Out[217]: 
one     -0.274551
three   -0.223384
two      0.162987
dtype: float64

In [218]: df.apply(np.mean, axis=1)
Out[218]: 
a   -0.394235
b    0.130882
c   -0.127557
d   -0.031682
dtype: float64

In [219]: df.apply(lambda x: x.max() - x.min())
Out[219]: 
one      0.810701
three    0.205973
two      0.724777
dtype: float64

In [220]: df.apply(np.cumsum)
Out[220]: 
        one     three       two
a -0.701368       NaN -0.087103
b -0.592035 -0.354359  0.550570
c -0.823652 -0.502746  0.547904
d       NaN -0.670153  0.651948

In [221]: df.apply(np.exp)
Out[221]: 
        one     three       two
a  0.495907       NaN  0.916583
b  1.115534  0.701623  1.892074
c  0.793250  0.862098  0.997337
d       NaN  0.845855  1.109649

Depending on the return type of the function passed to apply, the result will either be of lower dimension or the same dimension.

apply combined with some cleverness can be used to answer many questions about a data set. For example, suppose we wanted to extract the date where the maximum value for each column occurred:

In [222]: tsdf = DataFrame(randn(1000, 3), columns=['A', 'B', 'C'],
   .....:                  index=date_range('1/1/2000', periods=1000))
   .....:

In [223]: tsdf.apply(lambda x: x.index[x.dropna().argmax()])
Out[223]: 
A   2000-10-05 00:00:00
B   2002-05-26 00:00:00
C   2000-07-10 00:00:00
dtype: datetime64[ns]

You may also pass additional arguments and keyword arguments to the apply method. For instance, consider the following function you would like to apply:

def subtract_and_divide(x, sub, divide=1):
    return (x - sub) / divide

You may then apply this function as follows:

df.apply(subtract_and_divide, args=(5,), divide=3)

Another useful feature is the ability to pass Series methods to carry out some Series operation on each column or row:

In [224]: tsdf
Out[224]: 
                   A         B         C
2000-01-01 -0.748358  0.938378 -0.421370
2000-01-02  0.310699  0.247939  0.480243
2000-01-03 -0.135533 -0.754617  0.669998
2000-01-04       NaN       NaN       NaN
2000-01-05       NaN       NaN       NaN
2000-01-06       NaN       NaN       NaN
2000-01-07       NaN       NaN       NaN
2000-01-08 -1.421098 -1.527750 -0.391382
2000-01-09  0.881063  0.173443 -0.290646
2000-01-10  2.189553  2.017892 -1.140611

In [225]: tsdf.apply(Series.interpolate)
Out[225]: 
                   A         B         C
2000-01-01 -0.748358  0.938378 -0.421370
2000-01-02  0.310699  0.247939  0.480243
2000-01-03 -0.135533 -0.754617  0.669998
2000-01-04 -0.392646 -0.909243  0.457722
2000-01-05 -0.649759 -1.063870  0.245446
2000-01-06 -0.906872 -1.218497  0.033170
2000-01-07 -1.163985 -1.373123 -0.179106
2000-01-08 -1.421098 -1.527750 -0.391382
2000-01-09  0.881063  0.173443 -0.290646
2000-01-10  2.189553  2.017892 -1.140611

Finally, apply takes an argument raw which is False by default, which converts each row or column into a Series before applying the function. When set to True, the passed function will instead receive an ndarray object, which has positive performance implications if you do not need the indexing functionality.

See also

The section on GroupBy demonstrates related, flexible functionality for grouping by some criterion, applying, and combining the results into a Series, DataFrame, etc.

Applying elementwise Python functions

Since not all functions can be vectorized (accept NumPy arrays and return another array or value), the methods applymap on DataFrame and analogously map on Series accept any Python function taking a single value and returning a single value. For example:

In [226]: f = lambda x: len(str(x))

In [227]: df['one'].map(f)
Out[227]: 
a    15
b    14
c    15
d     3
Name: one, dtype: int64

In [228]: df.applymap(f)
Out[228]: 
   one  three  two
a   15      3   16
b   14     15   14
c   15     15   17
d    3     15   14

Series.map has an additional feature which is that it can be used to easily “link” or “map” values defined by a secondary series. This is closely related to merging/joining functionality:

In [229]: s = Series(['six', 'seven', 'six', 'seven', 'six'],
   .....:            index=['a', 'b', 'c', 'd', 'e'])
   .....:

In [230]: t = Series({'six' : 6., 'seven' : 7.})

In [231]: s
Out[231]: 
a      six
b    seven
c      six
d    seven
e      six
dtype: object

In [232]: s.map(t)
Out[232]: 
a    6
b    7
c    6
d    7
e    6
dtype: float64

Reindexing and altering labels

reindex is the fundamental data alignment method in pandas. It is used to implement nearly all other features relying on label-alignment functionality. To reindex means to conform the data to match a given set of labels along a particular axis. This accomplishes several things:

  • Reorders the existing data to match a new set of labels
  • Inserts missing value (NA) markers in label locations where no data for that label existed
  • If specified, fill data for missing labels using logic (highly relevant to working with time series data)

Here is a simple example:

In [233]: s = Series(randn(5), index=['a', 'b', 'c', 'd', 'e'])

In [234]: s
Out[234]: 
a    1.721293
b    0.355636
c    0.498722
d   -0.277859
e    0.713249
dtype: float64

In [235]: s.reindex(['e', 'b', 'f', 'd'])
Out[235]: 
e    0.713249
b    0.355636
f         NaN
d   -0.277859
dtype: float64

Here, the f label was not contained in the Series and hence appears as NaN in the result.

With a DataFrame, you can simultaneously reindex the index and columns:

In [236]: df
Out[236]: 
        one     three       two
a -0.701368       NaN -0.087103
b  0.109333 -0.354359  0.637674
c -0.231617 -0.148387 -0.002666
d       NaN -0.167407  0.104044

In [237]: df.reindex(index=['c', 'f', 'b'], columns=['three', 'two', 'one'])
Out[237]: 
      three       two       one
c -0.148387 -0.002666 -0.231617
f       NaN       NaN       NaN
b -0.354359  0.637674  0.109333

For convenience, you may utilize the reindex_axis method, which takes the labels and a keyword axis parameter.

Note that the Index objects containing the actual axis labels can be shared between objects. So if we have a Series and a DataFrame, the following can be done:

In [238]: rs = s.reindex(df.index)

In [239]: rs
Out[239]: 
a    1.721293
b    0.355636
c    0.498722
d   -0.277859
dtype: float64

In [240]: rs.index is df.index
Out[240]: True

This means that the reindexed Series’s index is the same Python object as the DataFrame’s index.

See also

Advanced indexing is an even more concise way of doing reindexing.

Note

When writing performance-sensitive code, there is a good reason to spend some time becoming a reindexing ninja: many operations are faster on pre-aligned data. Adding two unaligned DataFrames internally triggers a reindexing step. For exploratory analysis you will hardly notice the difference (because reindex has been heavily optimized), but when CPU cycles matter sprinking a few explicit reindex calls here and there can have an impact.

Reindexing to align with another object

You may wish to take an object and reindex its axes to be labeled the same as another object. While the syntax for this is straightforward albeit verbose, it is a common enough operation that the reindex_like method is available to make this simpler:

In [241]: df
Out[241]: 
        one     three       two
a -0.701368       NaN -0.087103
b  0.109333 -0.354359  0.637674
c -0.231617 -0.148387 -0.002666
d       NaN -0.167407  0.104044

In [242]: df2
Out[242]: 
        one       two
a -0.426817 -0.269738
b  0.383883  0.455039
c  0.042934 -0.185301

In [243]: df.reindex_like(df2)
Out[243]: 
        one       two
a -0.701368 -0.087103
b  0.109333  0.637674
c -0.231617 -0.002666

Reindexing with reindex_axis

Aligning objects with each other with align

The align method is the fastest way to simultaneously align two objects. It supports a join argument (related to joining and merging):

  • join='outer': take the union of the indexes
  • join='left': use the calling object’s index
  • join='right': use the passed object’s index
  • join='inner': intersect the indexes

It returns a tuple with both of the reindexed Series:

In [244]: s = Series(randn(5), index=['a', 'b', 'c', 'd', 'e'])

In [245]: s1 = s[:4]

In [246]: s2 = s[1:]

In [247]: s1.align(s2)
Out[247]: 
(a   -0.013026
b    2.249919
c    0.449017
d   -0.486899
e         NaN
dtype: float64,
 a         NaN
b    2.249919
c    0.449017
d   -0.486899
e   -1.666155
dtype: float64)

In [248]: s1.align(s2, join='inner')
Out[248]: 
(b    2.249919
c    0.449017
d   -0.486899
dtype: float64,
 b    2.249919
c    0.449017
d   -0.486899
dtype: float64)

In [249]: s1.align(s2, join='left')
Out[249]: 
(a   -0.013026
b    2.249919
c    0.449017
d   -0.486899
dtype: float64,
 a         NaN
b    2.249919
c    0.449017
d   -0.486899
dtype: float64)

For DataFrames, the join method will be applied to both the index and the columns by default:

In [250]: df.align(df2, join='inner')
Out[250]: 
(        one       two
a -0.701368 -0.087103
b  0.109333  0.637674
c -0.231617 -0.002666,
         one       two
a -0.426817 -0.269738
b  0.383883  0.455039
c  0.042934 -0.185301)

You can also pass an axis option to only align on the specified axis:

In [251]: df.align(df2, join='inner', axis=0)
Out[251]: 
(        one     three       two
a -0.701368       NaN -0.087103
b  0.109333 -0.354359  0.637674
c -0.231617 -0.148387 -0.002666,
         one       two
a -0.426817 -0.269738
b  0.383883  0.455039
c  0.042934 -0.185301)

If you pass a Series to DataFrame.align, you can choose to align both objects either on the DataFrame’s index or columns using the axis argument:

In [252]: df.align(df2.ix[0], axis=1)
Out[252]: 
(        one     three       two
a -0.701368       NaN -0.087103
b  0.109333 -0.354359  0.637674
c -0.231617 -0.148387 -0.002666
d       NaN -0.167407  0.104044,
 one     -0.426817
three         NaN
two     -0.269738
Name: a, dtype: float64)

Filling while reindexing

reindex takes an optional parameter method which is a filling method chosen from the following table:

Method Action
pad / ffill Fill values forward
bfill / backfill Fill values backward

Other fill methods could be added, of course, but these are the two most commonly used for time series data. In a way they only make sense for time series or otherwise ordered data, but you may have an application on non-time series data where this sort of “interpolation” logic is the correct thing to do. More sophisticated interpolation of missing values would be an obvious extension.

We illustrate these fill methods on a simple TimeSeries:

In [253]: rng = date_range('1/3/2000', periods=8)

In [254]: ts = Series(randn(8), index=rng)

In [255]: ts2 = ts[[0, 3, 6]]

In [256]: ts
Out[256]: 
2000-01-03    1.093167
2000-01-04    0.214964
2000-01-05   -0.355204
2000-01-06    1.228301
2000-01-07   -0.449976
2000-01-08   -0.923040
2000-01-09    0.701979
2000-01-10   -0.629836
Freq: D, dtype: float64

In [257]: ts2
Out[257]: 
2000-01-03    1.093167
2000-01-06    1.228301
2000-01-09    0.701979
dtype: float64

In [258]: ts2.reindex(ts.index)
Out[258]: 
2000-01-03    1.093167
2000-01-04         NaN
2000-01-05         NaN
2000-01-06    1.228301
2000-01-07         NaN
2000-01-08         NaN
2000-01-09    0.701979
2000-01-10         NaN
Freq: D, dtype: float64

In [259]: ts2.reindex(ts.index, method='ffill')
Out[259]: 
2000-01-03    1.093167
2000-01-04    1.093167
2000-01-05    1.093167
2000-01-06    1.228301
2000-01-07    1.228301
2000-01-08    1.228301
2000-01-09    0.701979
2000-01-10    0.701979
Freq: D, dtype: float64

In [260]: ts2.reindex(ts.index, method='bfill')
Out[260]: 
2000-01-03    1.093167
2000-01-04    1.228301
2000-01-05    1.228301
2000-01-06    1.228301
2000-01-07    0.701979
2000-01-08    0.701979
2000-01-09    0.701979
2000-01-10         NaN
Freq: D, dtype: float64

Note the same result could have been achieved using fillna:

In [261]: ts2.reindex(ts.index).fillna(method='ffill')
Out[261]: 
2000-01-03    1.093167
2000-01-04    1.093167
2000-01-05    1.093167
2000-01-06    1.228301
2000-01-07    1.228301
2000-01-08    1.228301
2000-01-09    0.701979
2000-01-10    0.701979
Freq: D, dtype: float64

Note these methods generally assume that the indexes are sorted. They may be modified in the future to be a bit more flexible but as time series data is ordered most of the time anyway, this has not been a major priority.

Dropping labels from an axis

A method closely related to reindex is the drop function. It removes a set of labels from an axis:

In [262]: df
Out[262]: 
        one     three       two
a -0.701368       NaN -0.087103
b  0.109333 -0.354359  0.637674
c -0.231617 -0.148387 -0.002666
d       NaN -0.167407  0.104044

In [263]: df.drop(['a', 'd'], axis=0)
Out[263]: 
        one     three       two
b  0.109333 -0.354359  0.637674
c -0.231617 -0.148387 -0.002666

In [264]: df.drop(['one'], axis=1)
Out[264]: 
      three       two
a       NaN -0.087103
b -0.354359  0.637674
c -0.148387 -0.002666
d -0.167407  0.104044

Note that the following also works, but is a bit less obvious / clean:

In [265]: df.reindex(df.index - ['a', 'd'])
Out[265]: 
        one     three       two
b  0.109333 -0.354359  0.637674
c -0.231617 -0.148387 -0.002666

Renaming / mapping labels

The rename method allows you to relabel an axis based on some mapping (a dict or Series) or an arbitrary function.

In [266]: s
Out[266]: 
a   -0.013026
b    2.249919
c    0.449017
d   -0.486899
e   -1.666155
dtype: float64

In [267]: s.rename(str.upper)
Out[267]: 
A   -0.013026
B    2.249919
C    0.449017
D   -0.486899
E   -1.666155
dtype: float64

If you pass a function, it must return a value when called with any of the labels (and must produce a set of unique values). But if you pass a dict or Series, it need only contain a subset of the labels as keys:

In [268]: df.rename(columns={'one' : 'foo', 'two' : 'bar'},
   .....:           index={'a' : 'apple', 'b' : 'banana', 'd' : 'durian'})
   .....:
Out[268]: 
             foo     three       bar
apple  -0.701368       NaN -0.087103
banana  0.109333 -0.354359  0.637674
c      -0.231617 -0.148387 -0.002666
durian       NaN -0.167407  0.104044

The rename method also provides an inplace named parameter that is by default False and copies the underlying data. Pass inplace=True to rename the data in place.

The Panel class has a related rename_axis class which can rename any of its three axes.

Iteration

Because Series is array-like, basic iteration produces the values. Other data structures follow the dict-like convention of iterating over the “keys” of the objects. In short:

  • Series: values
  • DataFrame: column labels
  • Panel: item labels

Thus, for example:

In [269]: for col in df:
   .....:     print col
   .....:
one
three
two

iteritems

Consistent with the dict-like interface, iteritems iterates through key-value pairs:

  • Series: (index, scalar value) pairs
  • DataFrame: (column, Series) pairs
  • Panel: (item, DataFrame) pairs

For example:

In [270]: for item, frame in wp.iteritems():
   .....:     print item
   .....:     print frame
   .....:
Item1
                   A         B         C         D
2000-01-01 -1.118121  0.431279  0.554724 -1.333649
2000-01-02 -0.332174 -0.485882  1.725945  1.799276
2000-01-03 -0.968916 -0.779465 -2.000701 -1.866630
2000-01-04 -1.101268  1.957478  0.058889  0.758071
2000-01-05  0.076612 -0.548502 -0.160485 -0.377780
Item2
                   A         B         C         D
2000-01-01  0.249911 -0.341270 -0.272599 -0.277446
2000-01-02 -1.102896  0.100307 -1.602814  0.920139
2000-01-03 -0.643870  0.060336 -0.434942 -0.494305
2000-01-04  0.737973  0.451632  0.334124 -0.787062
2000-01-05  0.651396 -0.741919  1.193881 -2.395763

iterrows

New in v0.7 is the ability to iterate efficiently through rows of a DataFrame. It returns an iterator yielding each index value along with a Series containing the data in each row:

In [271]: for row_index, row in df2.iterrows():
   .....:     print '%s\n%s' % (row_index, row)
   .....:
a
one   -0.426817
two   -0.269738
Name: a, dtype: float64
b
one    0.383883
two    0.455039
Name: b, dtype: float64
c
one    0.042934
two   -0.185301
Name: c, dtype: float64

For instance, a contrived way to transpose the dataframe would be:

In [272]: df2 = DataFrame({'x': [1, 2, 3], 'y': [4, 5, 6]})

In [273]: print df2
   x  y
0  1  4
1  2  5
2  3  6

In [274]: print df2.T
   0  1  2
x  1  2  3
y  4  5  6

In [275]: df2_t = DataFrame(dict((idx,values) for idx, values in df2.iterrows()))

In [276]: print df2_t
   0  1  2
x  1  2  3
y  4  5  6

itertuples

This method will return an iterator yielding a tuple for each row in the DataFrame. The first element of the tuple will be the row’s corresponding index value, while the remaining values are the row values proper.

For instance,

In [277]: for r in df2.itertuples(): print r
(0, 1, 4)
(1, 2, 5)
(2, 3, 6)

Vectorized string methods

Series is equipped (as of pandas 0.8.1) with a set of string processing methods that make it easy to operate on each element of the array. Perhaps most importantly, these methods exclude missing/NA values automatically. These are accessed via the Series’s str attribute and generally have names matching the equivalent (scalar) build-in string methods:

In [278]: s = Series(['A', 'B', 'C', 'Aaba', 'Baca', np.nan, 'CABA', 'dog', 'cat'])

In [279]: s.str.lower()
Out[279]: 
0       a
1       b
2       c
3    aaba
4    baca
5     NaN
6    caba
7     dog
8     cat
dtype: object

In [280]: s.str.upper()
Out[280]: 
0       A
1       B
2       C
3    AABA
4    BACA
5     NaN
6    CABA
7     DOG
8     CAT
dtype: object

In [281]: s.str.len()
Out[281]: 
0     1
1     1
2     1
3     4
4     4
5   NaN
6     4
7     3
8     3
dtype: float64

Methods like split return a Series of lists:

In [282]: s2 = Series(['a_b_c', 'c_d_e', np.nan, 'f_g_h'])

In [283]: s2.str.split('_')
Out[283]: 
0    [a, b, c]
1    [c, d, e]
2          NaN
3    [f, g, h]
dtype: object

Elements in the split lists can be accessed using get or [] notation:

In [284]: s2.str.split('_').str.get(1)
Out[284]: 
0      b
1      d
2    NaN
3      g
dtype: object

In [285]: s2.str.split('_').str[1]
Out[285]: 
0      b
1      d
2    NaN
3      g
dtype: object

Methods like replace and findall take regular expressions, too:

In [286]: s3 = Series(['A', 'B', 'C', 'Aaba', 'Baca',
   .....:             '', np.nan, 'CABA', 'dog', 'cat'])
   .....:

In [287]: s3
Out[287]: 
0       A
1       B
2       C
3    Aaba
4    Baca
5        
6     NaN
7    CABA
8     dog
9     cat
dtype: object

In [288]: s3.str.replace('^.a|dog', 'XX-XX ', case=False)
Out[288]: 
0           A
1           B
2           C
3    XX-XX ba
4    XX-XX ca
5            
6         NaN
7    XX-XX BA
8      XX-XX 
9     XX-XX t
dtype: object

Methods like contains, startswith, and endswith takes an extra na arguement so missing values can be considered True or False:

In [289]: s4 = Series(['A', 'B', 'C', 'Aaba', 'Baca', np.nan, 'CABA', 'dog', 'cat'])

In [290]: s4.str.contains('A', na=False)
Out[290]: 
0     True
1    False
2    False
3     True
4    False
5    False
6     True
7    False
8    False
dtype: bool
Method Description
cat Concatenate strings
split Split strings on delimiter
get Index into each element (retrieve i-th element)
join Join strings in each element of the Series with passed separator
contains Return boolean array if each string contains pattern/regex
replace Replace occurrences of pattern/regex with some other string
repeat Duplicate values (s.str.repeat(3) equivalent to x * 3)
pad Add whitespace to left, right, or both sides of strings
center Equivalent to pad(side='both')
slice Slice each string in the Series
slice_replace Replace slice in each string with passed value
count Count occurrences of pattern
startswith Equivalent to str.startswith(pat) for each element
endswidth Equivalent to str.endswith(pat) for each element
findall Compute list of all occurrences of pattern/regex for each string
match Call re.match on each element, returning matched groups as list
len Compute string lengths
strip Equivalent to str.strip
rstrip Equivalent to str.rstrip
lstrip Equivalent to str.lstrip
lower Equivalent to str.lower
upper Equivalent to str.upper

Sorting by index and value

There are two obvious kinds of sorting that you may be interested in: sorting by label and sorting by actual values. The primary method for sorting axis labels (indexes) across data structures is the sort_index method.

In [291]: unsorted_df = df.reindex(index=['a', 'd', 'c', 'b'],
   .....:                          columns=['three', 'two', 'one'])
   .....:

In [292]: unsorted_df.sort_index()
Out[292]: 
      three       two       one
a       NaN -0.087103 -0.701368
b -0.354359  0.637674  0.109333
c -0.148387 -0.002666 -0.231617
d -0.167407  0.104044       NaN

In [293]: unsorted_df.sort_index(ascending=False)
Out[293]: 
      three       two       one
d -0.167407  0.104044       NaN
c -0.148387 -0.002666 -0.231617
b -0.354359  0.637674  0.109333
a       NaN -0.087103 -0.701368

In [294]: unsorted_df.sort_index(axis=1)
Out[294]: 
        one     three       two
a -0.701368       NaN -0.087103
d       NaN -0.167407  0.104044
c -0.231617 -0.148387 -0.002666
b  0.109333 -0.354359  0.637674

DataFrame.sort_index can accept an optional by argument for axis=0 which will use an arbitrary vector or a column name of the DataFrame to determine the sort order:

In [295]: df.sort_index(by='two')
Out[295]: 
        one     three       two
a -0.701368       NaN -0.087103
c -0.231617 -0.148387 -0.002666
d       NaN -0.167407  0.104044
b  0.109333 -0.354359  0.637674

The by argument can take a list of column names, e.g.:

In [296]: df1 = DataFrame({'one':[2,1,1,1],'two':[1,3,2,4],'three':[5,4,3,2]})

In [297]: df1[['one', 'two', 'three']].sort_index(by=['one','two'])
Out[297]: 
   one  two  three
2    1    2      3
1    1    3      4
3    1    4      2
0    2    1      5

Series has the method order (analogous to R’s order function) which sorts by value, with special treatment of NA values via the na_last argument:

In [298]: s[2] = np.nan

In [299]: s.order()
Out[299]: 
0       A
3    Aaba
1       B
4    Baca
6    CABA
8     cat
7     dog
2     NaN
5     NaN
dtype: object

In [300]: s.order(na_last=False)
Out[300]: 
2     NaN
5     NaN
0       A
3    Aaba
1       B
4    Baca
6    CABA
8     cat
7     dog
dtype: object

Some other sorting notes / nuances:

  • Series.sort sorts a Series by value in-place. This is to provide compatibility with NumPy methods which expect the ndarray.sort behavior.
  • DataFrame.sort takes a column argument instead of by. This method will likely be deprecated in a future release in favor of just using sort_index.

Copying

The copy method on pandas objects copies the underlying data (though not the axis indexes, since they are immutable) and returns a new object. Note that it is seldom necessary to copy objects. For example, there are only a handful of ways to alter a DataFrame in-place:

  • Inserting, deleting, or modifying a column
  • Assigning to the index or columns attributes
  • For homogeneous data, directly modifying the values via the values attribute or advanced indexing

To be clear, no pandas methods have the side effect of modifying your data; almost all methods return new objects, leaving the original object untouched. If data is modified, it is because you did so explicitly.

dtypes

The main types stored in pandas objects are float, int, bool, datetime64[ns], timedelta[ns], and object. In addition these dtypes have item sizes, e.g. int64 and int32. A convenient dtypes attribute for DataFrames returns a Series with the data type of each column.

In [301]: dft = DataFrame(dict( A = np.random.rand(3),
   .....:                       B = 1,
   .....:                       C = 'foo',
   .....:                       D = Timestamp('20010102'),
   .....:                       E = Series([1.0]*3).astype('float32'),
   .....:                       F = False,
   .....:                       G = Series([1]*3,dtype='int8')))
   .....:

In [302]: dft
Out[302]: 
          A  B    C                   D  E      F  G
0  0.736120  1  foo 2001-01-02 00:00:00  1  False  1
1  0.364264  1  foo 2001-01-02 00:00:00  1  False  1
2  0.091972  1  foo 2001-01-02 00:00:00  1  False  1

In [303]: dft.dtypes
Out[303]: 
A           float64
B             int64
C            object
D    datetime64[ns]
E           float32
F              bool
G              int8
dtype: object

On a Series use the dtype method.

In [304]: dft['A'].dtype
Out[304]: dtype('float64')

If a pandas object contains data multiple dtypes IN A SINGLE COLUMN, the dtype of the column will be chosen to accommodate all of the data types (object is the most general).

# these ints are coerced to floats
In [305]: Series([1, 2, 3, 4, 5, 6.])
Out[305]: 
0    1
1    2
2    3
3    4
4    5
5    6
dtype: float64

# string data forces an ``object`` dtype
In [306]: Series([1, 2, 3, 6., 'foo'])
Out[306]: 
0      1
1      2
2      3
3      6
4    foo
dtype: object

The method get_dtype_counts will return the number of columns of each type in a DataFrame:

In [307]: dft.get_dtype_counts()
Out[307]: 
bool              1
datetime64[ns]    1
float32           1
float64           1
int64             1
int8              1
object            1
dtype: int64

Numeric dtypes will propagate and can coexist in DataFrames (starting in v0.11.0). If a dtype is passed (either directly via the dtype keyword, a passed ndarray, or a passed Series, then it will be preserved in DataFrame operations. Furthermore, different numeric dtypes will NOT be combined. The following example will give you a taste.

In [308]: df1 = DataFrame(randn(8, 1), columns = ['A'], dtype = 'float32')

In [309]: df1
Out[309]: 
          A
0 -0.693708
1  0.084626
2 -0.003949
3  0.268088
4  0.357356
5  0.052999
6 -0.632983
7  1.332674

In [310]: df1.dtypes
Out[310]: 
A    float32
dtype: object

In [311]: df2 = DataFrame(dict( A = Series(randn(8),dtype='float16'),
   .....:                       B = Series(randn(8)),
   .....:                       C = Series(np.array(randn(8),dtype='uint8')) ))
   .....:

In [312]: df2
Out[312]: 
          A         B    C
0  1.921875 -0.311588    0
1 -0.101746  0.550255    1
2  1.352539  0.718337    2
3  1.264648  1.252982  255
4 -1.261719 -0.453845    0
5 -1.037109  1.151367    1
6  1.552734  1.406869    0
7 -0.503418 -2.264574    0

In [313]: df2.dtypes
Out[313]: 
A    float16
B    float64
C      uint8
dtype: object

defaults

By default integer types are int64 and float types are float64, REGARDLESS of platform (32-bit or 64-bit). The following will all result in int64 dtypes.

In [314]: DataFrame([1,2],columns=['a']).dtypes
Out[314]: 
a    int64
dtype: object

In [315]: DataFrame({'a' : [1,2] }).dtypes
Out[315]: 
a    int64
dtype: object

In [316]: DataFrame({'a' : 1 }, index=range(2)).dtypes
Out[316]: 
a    int64
dtype: object

Numpy, however will choose platform-dependent types when creating arrays. The following WILL result in int32 on 32-bit platform.

In [317]: frame = DataFrame(np.array([1,2]))

upcasting

Types can potentially be upcasted when combined with other types, meaning they are promoted from the current type (say int to float)

In [318]: df3 = df1.reindex_like(df2).fillna(value=0.0) + df2

In [319]: df3
Out[319]: 
          A         B    C
0  1.228167 -0.311588    0
1 -0.017120  0.550255    1
2  1.348590  0.718337    2
3  1.532737  1.252982  255
4 -0.904363 -0.453845    0
5 -0.984110  1.151367    1
6  0.919751  1.406869    0
7  0.829256 -2.264574    0

In [320]: df3.dtypes
Out[320]: 
A    float32
B    float64
C    float64
dtype: object

The values attribute on a DataFrame return the lower-common-denominator of the dtypes, meaning the dtype that can accomodate ALL of the types in the resulting homogenous dtyped numpy array. This can force some upcasting.

In [321]: df3.values.dtype
Out[321]: dtype('float64')

astype

You can use the astype method to explicity convert dtypes from one to another. These will by default return a copy, even if the dtype was unchanged (pass copy=False to change this behavior). In addition, they will raise an exception if the astype operation is invalid.

Upcasting is always according to the numpy rules. If two different dtypes are involved in an operation, then the more general one will be used as the result of the operation.

In [322]: df3
Out[322]: 
          A         B    C
0  1.228167 -0.311588    0
1 -0.017120  0.550255    1
2  1.348590  0.718337    2
3  1.532737  1.252982  255
4 -0.904363 -0.453845    0
5 -0.984110  1.151367    1
6  0.919751  1.406869    0
7  0.829256 -2.264574    0

In [323]: df3.dtypes
Out[323]: 
A    float32
B    float64
C    float64
dtype: object

# conversion of dtypes
In [324]: df3.astype('float32').dtypes
Out[324]: 
A    float32
B    float32
C    float32
dtype: object

object conversion

convert_objects is a method to try to force conversion of types from the object dtype to other types. To force conversion of specific types that are number like, e.g. could be a string that represents a number, pass convert_numeric=True. This will force strings and numbers alike to be numbers if possible, otherwise they will be set to np.nan.

In [325]: df3['D'] = '1.'

In [326]: df3['E'] = '1'

In [327]: df3.convert_objects(convert_numeric=True).dtypes
Out[327]: 
A    float32
B    float64
C    float64
D    float64
E      int64
dtype: object

# same, but specific dtype conversion
In [328]: df3['D'] = df3['D'].astype('float16')

In [329]: df3['E'] = df3['E'].astype('int32')

In [330]: df3.dtypes
Out[330]: 
A    float32
B    float64
C    float64
D    float16
E      int32
dtype: object

To force conversion to datetime64[ns], pass convert_dates='coerce'. This will convert any datetimelike object to dates, forcing other values to NaT. This might be useful if you are reading in data which is mostly dates, but occasionally has non-dates intermixed and you want to represent as missing.

In [331]: s = Series([datetime(2001,1,1,0,0),
   .....:            'foo', 1.0, 1, Timestamp('20010104'),
   .....:            '20010105'],dtype='O')
   .....:

In [332]: s
Out[332]: 
0    2001-01-01 00:00:00
1                    foo
2                      1
3                      1
4    2001-01-04 00:00:00
5               20010105
dtype: object

In [333]: s.convert_objects(convert_dates='coerce')
Out[333]: 
0   2001-01-01 00:00:00
1                   NaT
2                   NaT
3                   NaT
4   2001-01-04 00:00:00
5   2001-01-05 00:00:00
dtype: datetime64[ns]

In addition, convert_objects will attempt the soft conversion of any object dtypes, meaning that if all the objects in a Series are of the same type, the Series will have that dtype.

gotchas

Performing selection operations on integer type data can easily upcast the data to floating. The dtype of the input data will be preserved in cases where nans are not introduced (starting in 0.11.0) See also integer na gotchas

In [334]: dfi = df3.astype('int32')

In [335]: dfi['E'] = 1

In [336]: dfi
Out[336]: 
   A  B    C  D  E
0  1  0    0  1  1
1  0  0    1  1  1
2  1  0    2  1  1
3  1  1  255  1  1
4  0  0    0  1  1
5  0  1    1  1  1
6  0  1    0  1  1
7  0 -2    0  1  1

In [337]: dfi.dtypes
Out[337]: 
A    int32
B    int32
C    int32
D    int32
E    int64
dtype: object

In [338]: casted = dfi[dfi>0]

In [339]: casted
Out[339]: 
    A   B    C  D  E
0   1 NaN  NaN  1  1
1 NaN NaN    1  1  1
2   1 NaN    2  1  1
3   1   1  255  1  1
4 NaN NaN  NaN  1  1
5 NaN   1    1  1  1
6 NaN   1  NaN  1  1
7 NaN NaN  NaN  1  1

In [340]: casted.dtypes
Out[340]: 
A    float64
B    float64
C    float64
D      int32
E      int64
dtype: object

While float dtypes are unchanged.

In [341]: dfa = df3.copy()

In [342]: dfa['A'] = dfa['A'].astype('float32')

In [343]: dfa.dtypes
Out[343]: 
A    float32
B    float64
C    float64
D    float16
E      int32
dtype: object

In [344]: casted = dfa[df2>0]

In [345]: casted
Out[345]: 
          A         B    C   D   E
0  1.228167       NaN  NaN NaN NaN
1       NaN  0.550255    1 NaN NaN
2  1.348590  0.718337    2 NaN NaN
3  1.532737  1.252982  255 NaN NaN
4       NaN       NaN  NaN NaN NaN
5       NaN  1.151367    1 NaN NaN
6  0.919751  1.406869  NaN NaN NaN
7       NaN       NaN  NaN NaN NaN

In [346]: casted.dtypes
Out[346]: 
A    float32
B    float64
C    float64
D    float16
E    float64
dtype: object

Pickling and serialization

All pandas objects are equipped with save methods which use Python’s cPickle module to save data structures to disk using the pickle format.

In [347]: df
Out[347]: 
        one     three       two
a -0.701368       NaN -0.087103
b  0.109333 -0.354359  0.637674
c -0.231617 -0.148387 -0.002666
d       NaN -0.167407  0.104044

In [348]: df.save('foo.pickle')

The load function in the pandas namespace can be used to load any pickled pandas object (or any other pickled object) from file:

In [349]: load('foo.pickle')
Out[349]: 
        one     three       two
a -0.701368       NaN -0.087103
b  0.109333 -0.354359  0.637674
c -0.231617 -0.148387 -0.002666
d       NaN -0.167407  0.104044

There is also a save function which takes any object as its first argument:

In [350]: save(df, 'foo.pickle')

In [351]: load('foo.pickle')
Out[351]: 
        one     three       two
a -0.701368       NaN -0.087103
b  0.109333 -0.354359  0.637674
c -0.231617 -0.148387 -0.002666
d       NaN -0.167407  0.104044

Working with package options

New in version 0.10.1.

Pandas has an options system that let’s you customize some aspects of it’s behaviour, display-related options being those the user is must likely to adjust.

Options have a full “dotted-style”, case-insensitive name (e.g. display.max_rows), You can get/set options directly as attributes of the top-level options attribute:

In [352]: import pandas as pd

In [353]: pd.options.display.max_rows
Out[353]: 60

In [354]: pd.options.display.max_rows = 999

In [355]: pd.options.display.max_rows
Out[355]: 999

There is also an API composed of 4 relavent functions, available directly from the pandas namespace, and they are:

  • get_option / set_option - get/set the value of a single option.
  • reset_option - reset one or more options to their default value.
  • describe_option - print the descriptions of one or more options.

Note: developers can check out pandas/core/config.py for more info.

All of the functions above accept a regexp pattern (re.search style) as an argument, and so passing in a substring will work - as long as it is unambiguous :

In [356]: get_option("display.max_rows")
Out[356]: 999

In [357]: set_option("display.max_rows",101)

In [358]: get_option("display.max_rows")
Out[358]: 101

In [359]: set_option("max_r",102)

In [360]: get_option("display.max_rows")
Out[360]: 102

The following will not work because it matches multiple option names, e.g.``display.max_colwidth``, display.max_rows, display.max_columns:

In [361]: try:
   .....:     get_option("display.max_")
   .....: except KeyError as e:
   .....:     print(e)
   .....:
  File "<ipython-input-361-633ae52b0297>", line 3
    except KeyError as e:
                         ^
IndentationError: unindent does not match any outer indentation level

Note: Using this form of convenient shorthand may make your code break if new options with similar names are added in future versions.

You can get a list of available options and their descriptions with describe_option. When called with no argument describe_option will print out the descriptions for all available options.

In [362]: describe_option()
display.chop_threshold: [default: None] [currently: None]
: float or None
        if set to a float value, all float values smaller then the given threshold
        will be displayed as exactly 0 by repr and friends.
display.colheader_justify: [default: right] [currently: right]
: 'left'/'right'
        Controls the justification of column headers. used by DataFrameFormatter.
display.column_space: [default: 12] [currently: 12]No description available.
display.date_dayfirst: [default: False] [currently: False]
: boolean
        When True, prints and parses dates with the day first, eg 20/01/2005
display.date_yearfirst: [default: False] [currently: False]
: boolean
        When True, prints and parses dates with the year first, eg 2005/01/20
display.encoding: [default: UTF-8] [currently: UTF-8]
: str/unicode
        Defaults to the detected encoding of the console.
        Specifies the encoding to be used for strings returned by to_string,
        these are generally strings meant to be displayed on the console.
display.expand_frame_repr: [default: True] [currently: True]
: boolean
        Whether to print out the full DataFrame repr for wide DataFrames
        across multiple lines.
        If False, the summary representation is shown.
display.float_format: [default: None] [currently: None]
: callable
        The callable should accept a floating point number and return
        a string with the desired format of the number. This is used
        in some places like SeriesFormatter.
        See core.format.EngFormatter for an example.
display.height: [default: 60] [currently: 60]
: int
        Height of the display in lines. In case python/IPython is running in a
        terminal this can be set to None and pandas will auto-detect the width.
        Note that the IPython notebook, IPython qtconsole, or IDLE do not run
        in a terminal, and hence it is not possible to correctly detect the height.
display.line_width: [default: 80] [currently: 80]
: int
        When printing wide DataFrames, this is the width of each line.
	(Deprecated, use `display.width` instead.)
display.max_columns: [default: 20] [currently: 20]
: int
        max_rows and max_columns are used in __repr__() methods to decide if
        to_string() or info() is used to render an object to a string.  In case
        python/IPython is running in a terminal this can be set to 0 and pandas
        will correctly auto-detect the width the terminal and swap to a smaller
        format in case all columns would not fit vertically. The IPython notebook,
        IPython qtconsole, or IDLE do not run in a terminal and hence it is not
        possible to do correct auto-detection.
        'None' value means unlimited.
display.max_colwidth: [default: 50] [currently: 50]
: int
        The maximum width in characters of a column in the repr of
        a pandas data structure. When the column overflows, a "..."
        placeholder is embedded in the output.
display.max_info_columns: [default: 100] [currently: 100]
: int
        max_info_columns is used in DataFrame.info method to decide if
        per column information will be printed.
display.max_info_rows: [default: 1690785] [currently: 1690785]
: int or None
        max_info_rows is the maximum number of rows for which a frame will
        perform a null check on its columns when repr'ing To a console.
        The default is 1,000,000 rows. So, if a DataFrame has more
        1,000,000 rows there will be no null check performed on the
        columns and thus the representation will take much less time to
        display in an interactive session. A value of None means always
        perform a null check when repr'ing.
display.max_rows: [default: 60] [currently: 102]
: int
        This sets the maximum number of rows pandas should output when printing
        out various output. For example, this value determines whether the repr()
        for a dataframe prints out fully or just a summary repr.
        'None' value means unlimited.
display.max_seq_items: [default: None] [currently: None]
: int or None
    
        when pretty-printing a long sequence, no more then `max_seq_items`
        will be printed. If items are ommitted, they will be denoted by the addition
        of "..." to the resulting string.
    
        If set to None, the number of items to be printed is unlimited.
display.mpl_style: [default: None] [currently: default]
: bool
    
        Setting this to 'default' will modify the rcParams used by matplotlib
        to give plots a more pleasing visual style by default.
        Setting this to None/False restores the values to their initial value.
display.multi_sparse: [default: True] [currently: True]
: boolean
        "sparsify" MultiIndex display (don't display repeated
        elements in outer levels within groups)
display.notebook_repr_html: [default: True] [currently: True]
: boolean
        When True, IPython notebook will use html representation for
        pandas objects (if it is available).
display.pprint_nest_depth: [default: 3] [currently: 3]
: int
        Controls the number of nested levels to process when pretty-printing
display.precision: [default: 7] [currently: 7]
: int
        Floating point output precision (number of significant digits). This is
        only a suggestion
display.width: [default: 80] [currently: 80]
: int
        Width of the display in characters. In case python/IPython is running in
        a terminal this can be set to None and pandas will correctly auto-detect the
        width.
        Note that the IPython notebook, IPython qtconsole, or IDLE do not run in a
        terminal and hence it is not possible to correctly detect the width.
mode.sim_interactive: [default: False] [currently: False]
: boolean
        Whether to simulate interactive mode for purposes of testing
mode.use_inf_as_null: [default: False] [currently: False]
: boolean
        True means treat None, NaN, INF, -INF as null (old way),
        False means None and NaN are null, but INF, -INF are not null
        (new way).

or you can get the description for just the options that match the regexp you pass in:

In [363]: describe_option("date")
display.date_dayfirst: [default: False] [currently: False]
: boolean
        When True, prints and parses dates with the day first, eg 20/01/2005
display.date_yearfirst: [default: False] [currently: False]
: boolean
        When True, prints and parses dates with the year first, eg 2005/01/20

All options also have a default value, and you can use the reset_option to do just that:

In [364]: get_option("display.max_rows")
Out[364]: 60

In [365]: set_option("display.max_rows",999)

In [366]: get_option("display.max_rows")
Out[366]: 999

In [367]: reset_option("display.max_rows")

In [368]: get_option("display.max_rows")
Out[368]: 60

It’s also possible to reset multiple options at once:

In [369]: reset_option("^display\.")

Console Output Formatting

Note: set_printoptions/ reset_printoptions are now deprecated (but functioning), and both, as well as set_eng_float_format, use the options API behind the scenes. The corresponding options now live under “print.XYZ”, and you can set them directly with get/set_option.

Use the set_eng_float_format function in the pandas.core.common module to alter the floating-point formatting of pandas objects to produce a particular format.

For instance:

In [370]: set_eng_float_format(accuracy=3, use_eng_prefix=True)

In [371]: s = Series(randn(5), index=['a', 'b', 'c', 'd', 'e'])

In [372]: s/1.e3
Out[372]: 
a      1.067m
b    -64.337u
c      1.484m
d   -524.332u
e   -688.585u
dtype: float64

In [373]: s/1.e6
Out[373]: 
a      1.067u
b    -64.337n
c      1.484u
d   -524.332n
e   -688.585n
dtype: float64

The set_printoptions function has a number of options for controlling how floating point numbers are formatted (using hte precision argument) in the console and . The max_rows and max_columns control how many rows and columns of DataFrame objects are shown by default. If max_columns is set to 0 (the default, in fact), the library will attempt to fit the DataFrame’s string representation into the current terminal width, and defaulting to the summary view otherwise.