Panda Python
Panda Python
Release 0.9.2.dev-61766ec
CONTENTS
Whats New 1.1 v0.9.1 (November 14, 2012) . . . . . . . . . . . . . . . 1.2 v0.9.0 (October 7, 2012) . . . . . . . . . . . . . . . . . 1.3 v0.8.1 (July 22, 2012) . . . . . . . . . . . . . . . . . . 1.4 v0.8.0 (June 29, 2012) . . . . . . . . . . . . . . . . . . 1.5 v.0.7.3 (April 12, 2012) . . . . . . . . . . . . . . . . . 1.6 v.0.7.2 (March 16, 2012) . . . . . . . . . . . . . . . . . 1.7 v.0.7.1 (February 29, 2012) . . . . . . . . . . . . . . . 1.8 v.0.7.0 (February 9, 2012) . . . . . . . . . . . . . . . . 1.9 v.0.6.1 (December 13, 2011) . . . . . . . . . . . . . . . 1.10 v.0.6.0 (November 25, 2011) . . . . . . . . . . . . . . . 1.11 v.0.5.0 (October 24, 2011) . . . . . . . . . . . . . . . . 1.12 v.0.4.3 through v0.4.1 (September 25 - October 9, 2011) Installation 2.1 Python version support 2.2 Binary installers . . . 2.3 Dependencies . . . . . 2.4 Optional dependencies 2.5 Installing from source 2.6 Running the test suite
. . . . . . . . . . . .
. . . . . . . . . . . .
. . . . . . . . . . . .
. . . . . . . . . . . .
. . . . . . . . . . . .
. . . . . . . . . . . .
. . . . . . . . . . . .
. . . . . . . . . . . .
. . . . . . . . . . . .
. . . . . . . . . . . .
. . . . . . . . . . . .
. . . . . . . . . . . .
. . . . . . . . . . . .
. . . . . . . . . . . .
. . . . . . . . . . . .
. . . . . . . . . . . .
. . . . . . . . . . . .
. . . . . . . . . . . .
. . . . . . . . . . . .
. . . . . . . . . . . .
. . . . . . . . . . . .
. . . . . . . . . . . .
. . . . . . . . . . . .
. . . . . . . . . . . .
3 3 7 8 9 15 19 19 20 25 25 27 28 31 31 31 32 32 33 33 35 35 41 41 42 42 42 42 42 47 47 51 64 69 69 i
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
Frequently Asked Questions (FAQ) 3.1 Migrating from scikits.timeseries to pandas >= 0.8.0 . . . . . . . . . . . . . . . . . . . . . . . . . . Package overview 4.1 Data structures at a glance . . . 4.2 Mutability and copying of data . 4.3 Getting Support . . . . . . . . 4.4 Credits . . . . . . . . . . . . . 4.5 Development Team . . . . . . . 4.6 License . . . . . . . . . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
Intro to Data Structures 5.1 Series . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 5.2 DataFrame . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 5.3 Panel . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . Essential basic functionality 6.1 Head and Tail . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . .
6.2 6.3 6.4 6.5 6.6 6.7 6.8 6.9 6.10 6.11 6.12 7
Attributes and the raw ndarray(s) Flexible binary operations . . . . Descriptive statistics . . . . . . . Function application . . . . . . . Reindexing and altering labels . . Iteration . . . . . . . . . . . . . . Vectorized string methods . . . . Sorting by index and value . . . . Copying, type casting . . . . . . Pickling and serialization . . . . Console Output Formatting . . .
. . . . . . . . . . .
. . . . . . . . . . .
. . . . . . . . . . .
. . . . . . . . . . .
. . . . . . . . . . .
. . . . . . . . . . .
. . . . . . . . . . .
. . . . . . . . . . .
. . . . . . . . . . .
. . . . . . . . . . .
. . . . . . . . . . .
. . . . . . . . . . .
. . . . . . . . . . .
. . . . . . . . . . .
. . . . . . . . . . .
. . . . . . . . . . .
. . . . . . . . . . .
. . . . . . . . . . .
. . . . . . . . . . .
. . . . . . . . . . .
. . . . . . . . . . .
. . . . . . . . . . .
. . . . . . . . . . .
. . . . . . . . . . .
. . . . . . . . . . .
. . . . . . . . . . .
. . . . . . . . . . .
. . . . . . . . . . .
. . . . . . . . . . .
. . . . . . . . . . .
. . . . . . . . . . .
. . . . . . . . . . .
. . . . . . . . . . .
. . . . . . . . . . .
. . . . . . . . . . .
. . . . . . . . . . .
70 71 75 81 83 90 91 94 96 97 98 99 99 108 112 113 123 126 127 127 130 134 136 137 145 145 147 148 154 157 157 161 162 165 168 169 170
Indexing and selecting data 7.1 Basics . . . . . . . . . . . . . . . . . . . . 7.2 Advanced indexing with labels . . . . . . . 7.3 Index objects . . . . . . . . . . . . . . . . 7.4 Hierarchical indexing (MultiIndex) . . . . 7.5 Adding an index to an existing DataFrame 7.6 Indexing internal details . . . . . . . . . . Computational tools 8.1 Statistical functions . . . . . . . . . . . . 8.2 Moving (rolling) statistics / moments . . . 8.3 Expanding window moment functions . . . 8.4 Exponentially weighted moment functions 8.5 Linear and panel regression . . . . . . . . Working with missing data 9.1 Missing data basics . . . . . . . . . . . 9.2 Calculations with missing data . . . . . 9.3 Cleaning / lling missing data . . . . . 9.4 Missing data casting rules and indexing
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
10 Group By: split-apply-combine 10.1 Splitting an object into groups . 10.2 Iterating through groups . . . . 10.3 Aggregation . . . . . . . . . . 10.4 Transformation . . . . . . . . . 10.5 Dispatching to instance methods 10.6 Flexible apply . . . . . . . . 10.7 Other useful features . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
11 Merge, join, and concatenate 173 11.1 Concatenating objects . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 173 11.2 Database-style DataFrame joining/merging . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 182 12 Reshaping and Pivot Tables 12.1 Reshaping by pivoting DataFrame objects 12.2 Reshaping by stacking and unstacking . . 12.3 Reshaping by Melt . . . . . . . . . . . . 12.4 Combining with stats and GroupBy . . . 12.5 Pivot tables and cross-tabulations . . . . 12.6 Tiling . . . . . . . . . . . . . . . . . . . 13 Time Series / Date functionality 191 191 192 196 196 197 201 203
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
. . . . . .
ii
Time Stamps vs. Time Spans . . . . Generating Ranges of Timestamps . . DateOffset objects . . . . . . . . . . Time series-related instance methods Up- and downsampling . . . . . . . . Time Span Representation . . . . . . Converting between Representations Time Zone Handling . . . . . . . . .
. . . . . . . .
. . . . . . . .
. . . . . . . .
. . . . . . . .
. . . . . . . .
. . . . . . . .
. . . . . . . .
. . . . . . . .
. . . . . . . .
. . . . . . . .
. . . . . . . .
. . . . . . . .
. . . . . . . .
. . . . . . . .
. . . . . . . .
. . . . . . . .
. . . . . . . .
. . . . . . . .
. . . . . . . .
. . . . . . . .
. . . . . . . .
. . . . . . . .
. . . . . . . .
. . . . . . . .
. . . . . . . .
. . . . . . . .
. . . . . . . .
. . . . . . . .
. . . . . . . .
. . . . . . . .
. . . . . . . .
. . . . . . . .
. . . . . . . .
. . . . . . . .
14 Plotting with matplotlib 225 14.1 Basic plotting: plot . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 225 14.2 Other plotting features . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 234 15 IO Tools (Text, CSV, HDF5, ...) 15.1 Clipboard . . . . . . . . . 15.2 CSV & Text les . . . . . 15.3 Excel les . . . . . . . . 15.4 HDF5 (PyTables) . . . . . 249 249 249 262 262
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
16 Sparse data structures 265 16.1 SparseArray . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 266 16.2 SparseList . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 267 16.3 SparseIndex objects . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 268 17 Caveats and Gotchas 17.1 NaN, Integer NA values and NA type promotions 17.2 Integer indexing . . . . . . . . . . . . . . . . . 17.3 Label-based slicing conventions . . . . . . . . . 17.4 Miscellaneous indexing gotchas . . . . . . . . . 17.5 Timestamp limitations . . . . . . . . . . . . . . 17.6 Parsing Dates from Text Files . . . . . . . . . . 17.7 Differences with NumPy . . . . . . . . . . . . . 18 rpy2 / R interface 18.1 Transferring R data sets into Python . . 18.2 Converting DataFrames into R objects . 18.3 Calling R functions with pandas objects 18.4 High-level interface to R estimators . . 269 269 271 271 272 273 274 274 275 275 276 276 276
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . . . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
. . . .
19 Related Python libraries 277 19.1 la (larry) . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 277 19.2 statsmodels . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 277 19.3 scikits.timeseries . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 277 20 Comparison with R / R libraries 20.1 data.frame . . . . . . . . . 20.2 zoo . . . . . . . . . . . . . 20.3 xts . . . . . . . . . . . . . 20.4 plyr . . . . . . . . . . . . . 20.5 reshape / reshape2 . . . . . 279 279 279 279 279 279
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
. . . . .
21 API Reference 281 21.1 General functions . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 281 21.2 Series . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 302 21.3 DataFrame . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 335
iii
21.4 Panel . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . 384 Python Module Index Python Module Index Index 385 387 389
iv
PDF Version Date: November 21, 2012 Version: 0.9.2.dev-61766ec Binary Installers: http://pypi.python.org/pypi/pandas Source Repository: http://github.com/pydata/pandas Issues & Ideas: https://github.com/pydata/pandas/issues Q&A Support: http://stackoverow.com/questions/tagged/pandas Developer Mailing List: http://groups.google.com/group/pystatsmodels pandas is a Python package providing fast, exible, and expressive data structures designed to make working with relational or labeled data both easy and intuitive. It aims to be the fundamental high-level building block for doing practical, real world data analysis in Python. Additionally, it has the broader goal of becoming the most powerful and exible open source data analysis / manipulation tool available in any language. It is already well on its way toward this goal. pandas is well suited for many different kinds of data: Tabular data with heterogeneously-typed columns, as in an SQL table or Excel spreadsheet Ordered and unordered (not necessarily xed-frequency) time series data. Arbitrary matrix data (homogeneously typed or heterogeneous) with row and column labels Any other form of observational / statistical data sets. The data actually need not be labeled at all to be placed into a pandas data structure The two primary data structures of pandas, Series (1-dimensional) and DataFrame (2-dimensional), handle the vast majority of typical use cases in nance, statistics, social science, and many areas of engineering. For R users, DataFrame provides everything that Rs data.frame provides and much more. pandas is built on top of NumPy and is intended to integrate well within a scientic computing environment with many other 3rd party libraries. Here are just a few of the things that pandas does well: Easy handling of missing data (represented as NaN) in oating point as well as non-oating point data Size mutability: columns can be inserted and deleted from DataFrame and higher dimensional objects Automatic and explicit data alignment: objects can be explicitly aligned to a set of labels, or the user can simply ignore the labels and let Series, DataFrame, etc. automatically align the data for you in computations Powerful, exible group by functionality to perform split-apply-combine operations on data sets, for both aggregating and transforming data Make it easy to convert ragged, differently-indexed data in other Python and NumPy data structures into DataFrame objects Intelligent label-based slicing, fancy indexing, and subsetting of large data sets Intuitive merging and joining data sets Flexible reshaping and pivoting of data sets Hierarchical labeling of axes (possible to have multiple labels per tick) Robust IO tools for loading data from at les (CSV and delimited), Excel les, databases, and saving / loading data from the ultrafast HDF5 format Time series-specic functionality: date range generation and frequency conversion, moving window statistics, moving window linear regressions, date shifting and lagging, etc. Many of these principles are here to address the shortcomings frequently experienced using other languages / scientic research environments. For data scientists, working with data is typically divided into multiple stages: munging and
CONTENTS
cleaning data, analyzing / modeling it, then organizing the results of the analysis into a form suitable for plotting or tabular display. pandas is the ideal tool for all of these tasks. Some other notes pandas is fast. Many of the low-level algorithmic bits have been extensively tweaked in Cython code. However, as with anything else generalization usually sacrices performance. So if you focus on one feature for your application you may be able to create a faster specialized tool. pandas is a dependency of statsmodels, making it an important part of the statistical computing ecosystem in Python. pandas has been used extensively in production in nancial applications. Note: This documentation assumes general familiarity with NumPy. If you havent used NumPy much or at all, do invest some time in learning about NumPy rst. See the package overview for more detail about whats in the library.
CONTENTS
CHAPTER
ONE
WHATS NEW
These are new features and improvements of note in each release.
DataFrame.rank now supports additional argument values for the na_option parameter so missing values can be assigned either the largest or the smallest rank (GH1508, GH2159)
In [1308]: df = DataFrame(np.random.randn(6, 3), columns=[A, B, C]) In [1309]: df.ix[2:4] = np.nan In [1310]: df.rank() Out[1310]: A B C 0 2 2 3 1 3 3 2 2 NaN NaN NaN 3 NaN NaN NaN 4 NaN NaN NaN
DataFrame has new where and mask methods to select values according to a given boolean mask (GH2109, GH2151) DataFrame currently supports slicing via a boolean vector the same length as the DataFrame (inside the []). The returned DataFrame has the same number of columns as the original, but is sliced on its index.
In [1313]: df = DataFrame(np.random.randn(5, 3), columns = [A,B,C]) In [1314]: df Out[1314]: A B C 0 0.203980 0.324917 -0.972813 1 0.154512 1.834703 1.712213 2 -0.024581 -0.364637 0.244539 3 -0.888715 -0.391710 1.088687 4 -0.030603 -1.518655 -0.591170 In [1315]: df[df[A] > 0] Out[1315]: A B C 0 0.203980 0.324917 -0.972813 1 0.154512 1.834703 1.712213
If a DataFrame is sliced with a DataFrame based boolean condition (with the same size as the original DataFrame), then a DataFrame the same size (index and columns) as the original is returned, with elements that do not meet the boolean condition as NaN. This is accomplished via the new method DataFrame.where. In addition, where takes an optional other argument for replacement.
In [1316]: df[df>0] Out[1316]: A B 0 0.203980 0.324917 1 0.154512 1.834703 2 NaN NaN 3 NaN NaN 4 NaN NaN
In [1317]: df.where(df>0) Out[1317]: A B C 0 0.203980 0.324917 NaN 1 0.154512 1.834703 1.712213 2 NaN NaN 0.244539 3 NaN NaN 1.088687 4 NaN NaN NaN In [1318]: df.where(df>0,-df) Out[1318]: A B C 0 0.203980 0.324917 0.972813 1 0.154512 1.834703 1.712213 2 0.024581 0.364637 0.244539 3 0.888715 0.391710 1.088687 4 0.030603 1.518655 0.591170
Furthermore, where now aligns the input boolean condition (ndarray or DataFrame), such that partial selection with setting is possible. This is analagous to partial setting via .ix (but on the contents rather than the axis labels)
In [1319]: df2 = df.copy() In [1320]: df2[ df2[1:4] > 0 ] = 3 In [1321]: df2 Out[1321]: A B C 0 0.203980 0.324917 -0.972813 1 3.000000 3.000000 3.000000 2 -0.024581 -0.364637 3.000000 3 -0.888715 -0.391710 3.000000 4 -0.030603 -1.518655 -0.591170
Added option to disable pandas-style tick locators and formatters using series.plot(x_compat=True) or pandas.plot_params[x_compat] = True (GH2205) Existing TimeSeries methods at_time and between_time were added to DataFrame (GH2149) DataFrame.dot can now accept ndarrays (GH2042) DataFrame.drop now supports non-unique indexes (GH2101) Panel.shift now supports negative periods (GH2164) DataFrame now support unary ~ operator (GH2110)
Period.end_time now returns the last nanosecond in the time interval (GH2124, GH2125, GH1764)
In [1328]: p = Period(2012) In [1329]: p.end_time Out[1329]: <Timestamp: 2012-12-31 23:59:59.999999999>
File parsers no longer coerce to oat or bool for columns that have custom converters specied (GH2184)
In [1330]: data = A,B,C\n00001,001,5\n00002,002,6 In [1331]: from cStringIO import StringIO In [1332]: read_csv(StringIO(data), converters={A : lambda x: x.strip()}) Out[1332]: A B C 0 00001 1 5 1 00002 2 6
See the full release notes or issue tracker on GitHub for a complete list.
Creating a Series from another Series, passing an index, will cause reindexing to happen inside rather than treating the Series like an ndarray. Technically improper usages like Series(df[col1], index=df[col2]) that worked before by accident (this was never intended) will lead to all NA Series in some cases. To be perfectly clear:
In [1337]: s1 = Series([1, 2, 3]) In [1338]: s1 Out[1338]: 0 1 1 2 2 3
In [1339]: s2 = Series(s1, index=[foo, bar, baz]) In [1340]: s2 Out[1340]: foo NaN bar NaN baz NaN
Deprecated day_of_year API removed from PeriodIndex, use dayofyear (GH1723) Dont modify NumPy suppress printoption to True at import time The internal HDF5 data arrangement for DataFrames has been transposed. Legacy les will still be readable by HDFStore (GH1834, GH1824) Legacy cruft removed: pandas.stats.misc.quantileTS Use ISO8601 format for Period repr: monthly, daily, and on down (GH1776) Empty DataFrame columns are now created as object dtype. This will prevent a class of TypeErrors that was occurring in code where the dtype of a column would depend on the presence of data or not (e.g. a SQL query having results) (GH1783) Setting parts of DataFrame/Panel using ix now aligns input Series/DataFrame (GH1630) first and last methods in GroupBy no longer drop non-numeric columns (GH1809) Resolved inconsistencies in specifying custom NA values in text parser. na_values of type dict no longer override default NAs unless keep_default_na is set to false explicitly (GH1657) DataFrame.dot will not do data alignment, and also work with Series (GH1915) See the full release notes or issue tracker on GitHub for a complete list.
Note: See documentation for overview of pandas timeseries API. New datetime64 representation speeds up join operations and data alignment, reduces memory usage, and improve serialization / deserialization performance signicantly over datetime.datetime
High performance and exible resample method for converting from high-to-low and low-to-high frequency. Supports interpolation, user-dened aggregation functions, and control over how the intervals and result labeling are dened. A suite of high performance Cython/C-based resampling functions (including Open-High-LowClose) have also been implemented. Revamp of frequency aliases and support for frequency shortcuts like 15min, or 1h30min New DatetimeIndex class supports both xed frequency and irregular time series. Replaces now deprecated DateRange class New PeriodIndex and Period classes for representing time spans and performing calendar logic, including the 12 scal quarterly frequencies <timeseries.quarterly>. This is a partial port of, and a substantial enhancement to, elements of the scikits.timeseries codebase. Support for conversion between PeriodIndex and DatetimeIndex New Timestamp data type subclasses datetime.datetime, providing the same interface while enabling working with nanosecond-resolution data. Also provides easy time zone conversions. Enhanced support for time zones. Add tz_convert and tz_lcoalize methods to TimeSeries and DataFrame. All timestamps are stored as UTC; Timestamps from DatetimeIndex objects with time zone set will be localized to localtime. Time zone conversions are therefore essentially free. User needs to know very little about pytz library now; only time zone names as as strings are required. Time zone-aware timestamps are equal if and only if their UTC timestamps match. Operations between time zone-aware time series with different time zones will result in a UTC-indexed time series. Time series string indexing conveniences / shortcuts: slice years, year and month, and index values with strings Enhanced time series plotting; adaptation of scikits.timeseries matplotlib-based plotting code New date_range, bdate_range, and period_range factory functions Robust frequency inference function infer_freq and inferred_freq property of DatetimeIndex, with option to infer frequency on construction of DatetimeIndex to_datetime function efciently parses array of strings to DatetimeIndex. DatetimeIndex will parse array or list of strings to datetime64 Optimized support for datetime64-dtype data in Series and DataFrame columns New NaT (Not-a-Time) type to represent NA in timestamp arrays Optimize Series.asof for looking up as of values for arrays of timestamps Milli, Micro, Nano date offset objects Can index time series with datetime.time objects to select all data at particular time of day (TimeSeries.at_time) or between two times (TimeSeries.between_time) Add tshift method for leading/lagging using the frequency (if any) of the index, as opposed to a naive lead/lag using shift
10
Add exible replace method for efciently substituting values Enhanced read_csv/read_table for reading time series data and converting multiple columns to dates Add comments option to parser functions: read_csv, etc. Add :refdayrst <io.dayrst> option to parser functions for parsing international DD/MM/YYYY dates Allow the user to specify the CSV reader dialect to control quoting etc. Handling thousands separators in read_csv to improve integer parsing. Enable unstacking of multiple levels in one shot. Alleviate pivot_table bugs (empty columns being introduced) Move to klib-based hash tables for indexing; better performance and less memory usage than Pythons dict Add rst, last, min, max, and prod optimized GroupBy functions New ordered_merge function Add exible comparison instance methods eq, ne, lt, gt, etc. to DataFrame, Series Improve scatter_matrix plotting function and add histogram or kernel density estimates to diagonal Add kde plot option for density plots Support for converting DataFrame to R data.frame through rpy2 Improved support for complex numbers in Series and DataFrame Add pct_change method to all data structures Add max_colwidth conguration option for DataFrame console output Interpolate Series values using index values Can select multiple columns from GroupBy Add update methods to Series/DataFrame for updating values in place Add any and all method to DataFrame
11
Vytautas Jancauskas, the 2012 GSOC participant, has added many new plot types. For example, kde is a new option:
In [1344]: s = Series(np.concatenate((np.random.randn(1000), ......: np.random.randn(1000) * 0.5 + 3))) ......: In [1345]: plt.figure() Out[1345]: <matplotlib.figure.Figure at 0x1b017350> In [1346]: s.hist(normed=True, alpha=0.2) Out[1346]: <matplotlib.axes.AxesSubplot at 0x1b2f5910> In [1347]: s.plot(kind=kde) Out[1347]: <matplotlib.axes.AxesSubplot at 0x1b2f5910>
12
pandass Timestamp object is a subclass of datetime.datetime that has nanosecond support (the nanosecond eld store the nanosecond value between 0 and 999). It should substitute directly into any code that 1.4. v0.8.0 (June 29, 2012) 13
used datetime.datetime values before. Thus, I recommend not casting DatetimeIndex to regular NumPy arrays. If you have code that requires an array of datetime.datetime objects, you have a couple of options. First, the asobject property of DatetimeIndex produces an array of Timestamp objects:
In [1355]: stamp_array = rng.asobject
In [1356]: stamp_array Out[1356]: Index([2000-01-01 00:00:00, 2000-01-02 00:00:00, 2000-01-03 00:00:00, 2000-01-04 00:00:00, In [1357]: stamp_array[5] Out[1357]: <Timestamp: 2000-01-06 00:00:00>
matplotlib knows how to handle datetime.datetime but not Timestamp objects. While I recommend that you plot time series using TimeSeries.plot, you can either use to_pydatetime or register a converter for the Timestamp type. See matplotlib documentation for more on this. Warning: There are bugs in the user-facing API with the nanosecond datetime64 unit in NumPy 1.6. In particular, the string version of the array shows garbage values, and conversion to dtype=object is similarly broken.
In [1361]: rng = date_range(1/1/2000, periods=10) In [1362]: rng Out[1362]: <class pandas.tseries.index.DatetimeIndex> [2000-01-01 00:00:00, ..., 2000-01-10 00:00:00] Length: 10, Freq: D, Timezone: None In [1363]: np.asarray(rng) Out[1363]: array([1970-01-11 184:00:00, 1970-01-11 208:00:00, 1970-01-11 232:00:00, 1970-01-11 00:00:00, 1970-01-11 24:00:00, 1970-01-11 48:00:00, 1970-01-11 72:00:00, 1970-01-11 96:00:00, 1970-01-11 120:00:00, 1970-01-11 144:00:00], dtype=datetime64[ns]) In [1364]: converted = np.asarray(rng, dtype=object) In [1365]: converted[5] Out[1365]: datetime.datetime(1970, 1, 11, 48, 0)
Trust me: dont panic. If you are using NumPy 1.6 and restrict your interaction with datetime64 values to pandass API you will be just ne. There is nothing wrong with the data-type (a 64-bit integer internally); all of the important data processing happens in pandas and is heavily tested. I strongly recommend that you do not work directly with datetime64 arrays in NumPy 1.6 and only use the pandas API.
14
Support for non-unique indexes: In the latter case, you may have code inside a try:... catch: block that failed due to the index not being unique. In many cases it will no longer fail (some method like append still check for uniqueness unless disabled). However, all is not lost: you can inspect index.is_unique and raise an exception explicitly if it is False or go to a different code branch.
15
Add stacked argument to Series and DataFrames plot method for stacked bar plots.
df.plot(kind=bar, stacked=True)
16
df.plot(kind=barh, stacked=True)
Add log x and y scaling options to DataFrame.plot and Series.plot Add kurt methods to Series and DataFrame for computing kurtosis
In comparisons, NA / NaN will always come through as False except with != which is True. Be very careful with boolean arithmetic, especially negation, in the presence of NA data. You may wish to add an explicit NA lter into boolean array operations if you are worried about this:
In [1369]: mask = series == Steve In [1370]: series[mask & series.notnull()] Out[1370]: 0 Steve
While propagating NA in comparisons may seem like the right behavior to some users (and you could argue on purely technical grounds that this is the right thing to do), the evaluation was made that propagating NA everywhere, including in numerical arrays, would cause a large amount of problems for users. Thus, a practicality beats purity approach was taken. This issue may be revisited at some point in the future. 1.5. v.0.7.3 (April 12, 2012) 17
18
Add reorder_levels method to Series and DataFrame (PR534) Add dict-like get function to DataFrame and Panel (PR521) Add DataFrame.iterrows method for efciently iterating through the rows of a DataFrame Add DataFrame.to_panel with code adapted from LongPanel.to_long Add reindex_axis method added to DataFrame Add level option to binary arithmetic functions on DataFrame and Series Add level option to the reindex and align methods on Series and DataFrame for broadcasting values across a level (GH542, PR552, others) Add attribute-based item access to Panel and add IPython completion (PR563) Add logy option to Series.plot for log-scaling on the Y axis Add index and header options to DataFrame.to_string Can pass multiple DataFrames to DataFrame.join to join on index (GH115) Can pass multiple Panels to Panel.join (GH115) Added justify argument to DataFrame.to_string to allow different alignment of column headers Add sort option to GroupBy to allow disabling sorting of the group keys for potential speedups (GH595)
20
Can pass MaskedArray to Series constructor (PR563) Add Panel item access via attributes and IPython completion (GH554) Implement DataFrame.lookup, fancy-indexing analogue for retrieving values given a sequence of row and column labels (GH338) Can pass a list of functions to aggregate with groupby on a DataFrame, yielding an aggregated result with hierarchical columns (GH166) Can call cummin and cummax on Series and DataFrame to get cumulative minimum and maximum, respectively (GH647) value_range added as utility function to get min and max of a dataframe (GH288) Added encoding argument to read_csv, read_table, to_csv and from_csv for non-ascii text (GH717) Added abs method to pandas objects Added crosstab function for easily computing frequency tables Added isin method to index objects Added level argument to xs method of DataFrame.
This is all exactly identical to the behavior before. However, if you ask for a key not contained in the Series, in versions 0.6.1 and prior, Series would fall back on a location-based lookup. This now raises a KeyError:
In [2]: s[1] KeyError: 1
21
2 3 -0.1787 0.03162 0.2504 0.58374 -1.4996 0.27163 -0.2658 0.11503 -0.8566 0.61941 -1.0108 0.47990 -0.6049 -0.43544 0.0424 -0.16037
In order to support purely integer-based indexing, the following methods have been added: Method Series.iget_value(i) Series.iget(i) DataFrame.irow(i) DataFrame.icol(j) DataFrame.iget_value(i, j) Description Retrieve value stored at location i Alias for iget_value Retrieve the i-th row Retrieve the j-th column Retrieve the value at row i and column j
If the index had been sorted, the range selection would have been possible:
22
In [1386]: s2 = s.sort_index() In [1387]: s2 Out[1387]: a -1.210803 c -1.301238 e 0.150472 g -1.918951 k 0.577345 m 0.065440 In [1388]: s2.ix[b:h] Out[1388]: c -1.301238 e 0.150472 g -1.918951
In the case of integer indexes, the behavior will be exactly as before (shadowing ndarray):
23
In [1394]: s = Series(randn(6), index=range(0, 12, 2)) In [1395]: s[[4, 0, 2]] Out[1395]: 4 0.842681 0 -0.011901 2 -1.382787 In [1396]: s[1:5] Out[1396]: 2 -1.382787 4 0.842681 6 -0.064157 8 1.228715
If you wish to do indexing with sequences and slicing on an integer index with label semantics, use ix.
Use Cythonized groupers when possible in Series/DataFrame stat ops with level parameter passed (GH545) Ported skiplist data structure to C to speed up rolling_median by about 5-10x in most typical use cases (GH374)
Added head and tail methods to Series, analogous to to DataFrame (PR296) Added Series.isin function which checks if each value is contained in a passed sequence (GH289) Added float_format option to Series.to_string Added skip_footer (GH291) and converters (GH343) options to read_csv and read_table Added drop_duplicates and duplicated functions for removing duplicate DataFrame rows and checking for duplicate rows, respectively (GH319) Implemented operators &, |, ^, - on DataFrame (GH347) Added Series.mad, mean absolute deviation Added QuarterEnd DateOffset (PR321) Added dot to DataFrame (GH65) Added orient option to Panel.from_dict (GH359, GH301) Added orient option to DataFrame.from_dict Added passing list of tuples or list of lists to DataFrame.from_records (GH357) Added multiple levels to groupby (GH103) Allow multiple columns in by argument of DataFrame.sort_index (GH92, PR362) Added fast get_value and put_value methods to DataFrame (GH360) Added cov instance methods to Series and DataFrame (GH194, PR362) Added kind=bar option to DataFrame.plot (PR348) Added idxmin and idxmax to Series and DataFrame (PR286) Added read_clipboard function to parse DataFrame from clipboard (GH300) Added nunique function to Series for counting unique elements (GH297) Made DataFrame constructor use Series name if no columns passed (GH373) Support regular expressions in read_table/read_csv (GH364) Added DataFrame.to_html for writing DataFrame to HTML (PR387) Added support for MaskedArray data in DataFrame, masked values converted to NaN (PR396) Added DataFrame.boxplot function (GH368) Can pass extra args, kwds to DataFrame.apply (GH376) Implement DataFrame.join with vector on argument (GH312) Added legend boolean ag to DataFrame.plot (GH324) Can pass multiple levels to stack and unstack (GH370) Can pass multiple values columns to pivot_table (GH381) Use Series name in GroupBy for result index (GH363) Added raw option to DataFrame.apply for performance if only need ndarray (GH309) Added proper, tested weighted least squares to standard and panel OLS (GH303)
26
Added convenience set_index function for creating a DataFrame index from its existing columns Implemented groupby hierarchical index level name (GH223) Added support for different delimiters in DataFrame.to_csv (PR244) TODO: DOCS ABOUT TAKE METHODS
28
29
30
CHAPTER
TWO
INSTALLATION
You have the option to install an ofcial release or to build the development version. If you choose to install from source and are running Windows, you will have to ensure that you have a compatible C compiler (MinGW or Visual Studio) installed. How-to install MinGW on Windows
31
2.2.2 Overview
Plat- Distriform bution Win- all dows Mac all Linux Debian Status stable stable stable Download / Repository Link All platforms All platforms ofcial Debian repository Install method pip install pandas pip install pandas sudo apt-get install python-pandas sudo apt-get install python-pandas sudo apt-get install python-pandas sudo apt-get install python-pandas zypper in python-pandas
NeuroDebian
Linux Ubuntu
PythonXY PPA; activate by: sudo add-apt-repository ppa:pythonxy/pythonxy-devel && sudo apt-get update OpenSuse Repository
2.3 Dependencies
NumPy: 1.6.1 or higher python-dateutil 1.5
32
Chapter 2. Installation
On Windows, I suggest installing the MinGW compiler suite following the directions linked to above. Once congured property, run the following on the command line:
python setup.py build --compiler=mingw32 python setup.py install
Note that you will not be able to import pandas if you open an interpreter in the source directory unless you build the C extensions in place:
python setup.py build_ext --inplace
The most recent version of MinGW (any installer dated after 2011-08-03) has removed the -mno-cygwin option but Distutils has not yet been updated to reect that. Thus, you may run into an error like unrecognized command line option -mno-cygwin. Until the bug is xed in Distutils, you may need to install a slightly older version of MinGW (2011-08-02 installer).
33
34
Chapter 2. Installation
CHAPTER
THREE
Date and DateArray from scikits.timeseries have been reincarnated in pandas Period and PeriodIndex:
In [394]: pnow(D) # scikits.timeseries.now() Out[394]: Period(2012-11-21, D) In [395]: Period(year=2007, month=3, day=15, freq=D) Out[395]: Period(2007-03-15, D) In [396]: p = Period(1984Q3) In [397]: p Out[397]: Period(1984Q3, Q-DEC) In [398]: p.asfreq(D, start) Out[398]: Period(1984-07-01, D)
35
In [399]: p.asfreq(D, end) Out[399]: Period(1984-09-30, D) In [400]: (p + 3).asfreq(T) + 6 * 60 + 30 Out[400]: Period(1985-07-01 06:29, T) In [401]: rng = period_range(1990, 2010, freq=A) In [402]: rng Out[402]: <class pandas.tseries.period.PeriodIndex> freq: A-DEC [1990, ..., 2010] length: 21 In [403]: rng.asfreq(B, end) - 3 Out[403]: <class pandas.tseries.period.PeriodIndex> freq: B [1990-12-26, ..., 2010-12-28] length: 21
Notes A span of time, from yearly through to secondly An array of timespans Frequency conversion in scikits.timeseries currently supports up to daily frequency, see issue 736
36
-0.3706 -1.1579 -1.3443 0.8449 1.0758 -0.109 1.6436 -1.4694 -0.6746 -1.7769 -0.9689 -1.2945 0.4137 0.2767 -0.472 -0.014 -0.0062 -0.9231 0.8957 0.8052], = [Jan-2012 ... Feb-2016], = M)
In [407]: data.convert(A, func=np.mean) Out[407]: timeseries([-0.394509620575 -0.24462765889 -0.221632512996 -0.453772693384 0.8504806638], dates = [2012 ... 2016], freq = A-DEC)
37
2003-02 2003-03 2003-04 2003-05 2003-06 2003-07 2003-08 2003-09 2003-10 2003-11 2003-12 2004-01 2004-02 Freq: M
0.149748 -0.732339 0.687738 0.176444 0.403310 -0.154951 0.301624 -2.179861 -1.369849 -0.954208 1.462696 -1.743161 -0.826591
In [411]: data.resample(A, how=np.mean) Out[411]: 2000 0.166630 2001 -0.114581 2002 -0.205961 2003 -0.235802 2004 -1.284876 Freq: A-DEC
3.1.3 Plotting
Much of the plotting functionality of scikits.timeseries has been ported and adopted to pandass data structures. For example:
In [412]: rng = period_range(1987Q2, periods=10, freq=Q-DEC) In [413]: data = Series(np.random.randn(10), index=rng) In [414]: plt.figure(); data.plot() Out[414]: <matplotlib.axes.AxesSubplot at 0x88da690>
38
39
2000-01-10 Freq: D
0.274230
In [418]: data.index Out[418]: <class pandas.tseries.index.DatetimeIndex> [2000-01-01 00:00:00, ..., 2000-07-18 00:00:00] Length: 200, Freq: D, Timezone: None In [419]: Out[419]: 2000-01 2000-02 2000-03 2000-04 2000-05 2000-06 2000-07 Freq: M data.resample(M, kind=period) 0.163775 0.026549 -0.089563 -0.079405 0.160348 0.101725 -0.708770
Similarly, resampling from periods to timestamps is possible with an optional interval (start or end) convention:
In [420]: rng = period_range(Jan-2000, periods=50, freq=M) In [421]: data = Series(np.random.randn(50), index=rng) In [422]: resampled = data.resample(A, kind=timestamp, convention=end) In [423]: resampled.index Out[423]: <class pandas.tseries.index.DatetimeIndex> [2000-12-31 00:00:00, ..., 2004-12-31 00:00:00] Length: 5, Freq: A-DEC, Timezone: None
40
CHAPTER
FOUR
PACKAGE OVERVIEW
pandas consists of the following things A set of labeled array data structures, the primary of which are Series/TimeSeries and DataFrame Index objects enabling both simple axis indexing and multi-level / hierarchical axis indexing An integrated group by engine for aggregating and transforming data sets Date range generation (date_range) and custom date offsets enabling the implementation of customized frequencies Input/Output tools: loading tabular data from at les (CSV, delimited, Excel 2003), and saving and loading pandas objects from the fast and efcient PyTables/HDF5 format. Memory-efcent sparse versions of the standard data structures for storing data that is mostly missing or mostly constant (some xed value) Moving window statistics (rolling mean, rolling standard deviation, etc.) Static and moving window linear and panel regression
Series 1D labeled homogeneously-typed array TimeSeries with index containing datetimes Series DataFrame General 2D labeled, size-mutable tabular structure with potentially heterogeneously-typed columns Panel General 3D labeled, also size-mutable array
intended to lend more semantic meaning to the data; i.e., for a particular data set there is likely to be a right way to orient the data. The goal, then, is to reduce the amount of mental effort required to code up data transformations in downstream functions. For example, with tabular data (DataFrame) it is more semantically helpful to think of the index (the rows) and the columns rather than axis 0 and axis 1. And iterating through the columns of the DataFrame thus results in more readable code:
for col in df.columns: series = df[col] # do something with series
4.4 Credits
pandas development began at AQR Capital Management in April 2008. It was open-sourced at the end of 2009. AQR continued to provide resources for development through the end of 2011, and continues to contribute bug reports today. Since January 2012, Lambda Foundry, has been providing development resources, as well as commercial support, training, and consulting for pandas. pandas is only made possible by a group of people around the world like you who have contributed new code, bug reports, xes, comments and ideas. A complete list can be found on Github.
4.6 License
42
======= License ======= pandas is distributed under a 3-clause ("Simplified" or "New") BSD license. Parts of NumPy, SciPy, numpydoc, bottleneck, which all have BSD-compatible licenses, are included. Their licenses follow the pandas license. pandas license ============== Copyright (c) 2011-2012, Lambda Foundry, Inc. and PyData Development Team All rights reserved. Copyright (c) 2008-2011 AQR Capital Management, LLC All rights reserved. Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met: * Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer. * Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following disclaimer in the documentation and/or other materials provided with the distribution. * Neither the name of the copyright holder nor the names of any contributors may be used to endorse or promote products derived from this software without specific prior written permission. THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDER AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. About the Copyright Holders =========================== AQR Capital Management began pandas development in 2008. Development was led by Wes McKinney. AQR released the source under this license in 2009. Wes is now an employee of Lambda Foundry, and remains the pandas project lead. The PyData Development Team is the collection of developers of the PyData project. This includes all of the PyData sub-projects, including pandas. The core team that coordinates development on GitHub can be found here: http://github.com/pydata.
4.6. License
43
Full credits for pandas contributors can be found in the documentation. Our Copyright Policy ==================== PyData uses a shared copyright model. Each contributor maintains copyright over their contributions to PyData. However, it is important to note that these contributions are typically only changes to the repositories. Thus, the PyData source code, in its entirety, is not the copyright of any single person or institution. Instead, it is the collective copyright of the entire PyData Development Team. If individual contributors want to maintain a record of what changes/contributions they have specific copyright on, they should indicate their copyright in the commit message of the change when they commit the change to one of the PyData repositories. With this in mind, the following banner should be used in any source code file to indicate the copyright and license terms: #----------------------------------------------------------------------------# Copyright (c) 2012, PyData Development Team # All rights reserved. # # Distributed under the terms of the BSD Simplified License. # # The full license is in the LICENSE file, distributed with this software. #----------------------------------------------------------------------------NumPy license ------------Copyright (c) 2005-2012, NumPy Developers. All rights reserved. Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met: * Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer. * Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following disclaimer in the documentation and/or other materials provided with the distribution. * Neither the name of the NumPy Developers nor the names of any contributors may be used to endorse or promote products derived from this software without specific prior written permission. THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
44
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. itions of this License Agreement.
SciPy license ------------Copyright (c) 2001, 2002 Enthought, Inc. All rights reserved. Copyright (c) 2003-2012 SciPy Developers. All rights reserved. Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met: a. Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer. b. Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following disclaimer in the documentation and/or other materials provided with the distribution. c. Neither the name of the Enthought nor the names of its contributors may be used to endorse or promote products derived from this software without specific prior written permission.
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. numpydoc license ---------------The numpydoc license is in pandas/doc/sphinxext/LICENSE.txt Bottleneck license -----------------Copyright (c) 2010-2012 Archipel Asset Management AB. All rights reserved. Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met: * Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer. * Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following disclaimer in the
4.6. License
45
documentation and/or other materials provided with the distribution. THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
46
CHAPTER
FIVE
Here is a basic tenet to keep in mind: data alignment is intrinsic. The link between labels and data will not be broken unless done so explicitly by you. Well give a brief intro to the data structures, then consider all of the broad categories of functionality and methods in separate sections. When using pandas, we recommend the following import convention:
import pandas as pd
5.1 Series
Series is a one-dimensional labeled array (technically a subclass of ndarray) capable of holding any data type (integers, strings, oating point numbers, Python objects, etc.). The axis labels are collectively referred to as the index. The basic method to create a Series is to call:
>>> s = Series(data, index=index)
Here, data can be many different things: a Python dict an ndarray a scalar value (like 5) The passed index is a list of axis labels. Thus, this separates into a few cases depending on what data is: From ndarray If data is an ndarray, index must be the same length as data. If no index is passed, one will be created having values [0, ..., len(data) - 1].
47
In [272]: s = Series(randn(5), index=[a, b, c, d, e]) In [273]: s Out[273]: a 0.664 b -0.487 c -0.504 d 0.307 e 1.570 In [274]: s.index Out[274]: Index([a, b, c, d, e], dtype=object) In [275]: Series(randn(5)) Out[275]: 0 -0.431 1 -0.705 2 0.555 3 0.939 4 0.722
Note: Starting in v0.8.0, pandas supports non-unique index values. In previous version, if the index values are not unique an exception will not be raised immediately, but attempting any operation involving the index will later result in an exception. In other words, the Index object containing the labels lazily checks whether the values are unique. The reason for being lazy is nearly all performance-based (there are many instances in computations, like parts of GroupBy, where the index is not used). From dict If data is a dict, if index is passed the values in data corresponding to the labels in the index will be pulled out. Otherwise, an index will be constructed from the sorted keys of the dict, if possible.
In [276]: d = {a : 0., b : 1., c : 2.} In [277]: Series(d) Out[277]: a 0 b 1 c 2 In [278]: Series(d, index=[b, c, d, a]) Out[278]: b 1 c 2 d NaN a 0
Note: NaN (not a number) is the standard missing data marker used in pandas From scalar value If data is a scalar value, an index must be provided. The value will be repeated to match the length of index
In [279]: Series(5., index=[a, b, c, d, e]) Out[279]: a 5 b 5
48
c d e
5 5 5
5.1. Series
49
Using the get method, a missing label will return None or specied default:
In [290]: s.get(f) In [291]: s.get(f, np.nan) Out[291]: nan
A key difference between Series and ndarray is that operations between Series automatically align the data based on label. Thus, you can write computations without giving consideration to whether the Series involved have the same labels.
In [295]: s[1:] + s[:-1] Out[295]: a NaN b -0.974 c -1.009 d 0.613 e NaN
50
The result of an operation between unaligned Series will have the union of the indexes involved. If a label is not found in one Series or the other, the result will be marked as missing (NaN). Being able to write code without doing any explicit data alignment grants immense freedom and exibility in interactive data analysis and research. The integrated data alignment features of the pandas data structures set pandas apart from the majority of related tools for working with labeled data. Note: In general, we chose to make the default result of operations between differently indexed objects yield the union of the indexes in order to avoid loss of information. Having an index label, though the data is missing, is typically important information as part of a computation. You of course have the option of dropping labels with missing data via the dropna function.
The Series name will be assigned automatically in many cases, in particular when taking 1D slices of DataFrame as you will see below.
5.2 DataFrame
DataFrame is a 2-dimensional labeled data structure with columns of potentially different types. You can think of it like a spreadsheet or SQL table, or a dict of Series objects. It is generally the most commonly used pandas object. Like Series, DataFrame accepts many different kinds of input: Dict of 1D ndarrays, lists, dicts, or Series 2-D numpy.ndarray Structured or record ndarray A Series Another DataFrame Along with the data, you can optionally pass index (row labels) and columns (column labels) arguments. If you pass an index and / or columns, you are guaranteeing the index and / or columns of the resulting DataFrame. Thus, a dict of Series plus a specic index will discard all data not matching up to the passed index. If axis labels are not passed, they will be constructed from the input data based on common sense rules.
5.2. DataFrame
51
The row and column labels can be accessed respectively by accessing the index and columns attributes: Note: When a particular set of columns is passed along with a dict of data, the passed columns override the keys in the dict.
In [304]: df.index Out[304]: Index([a, b, c, d], dtype=object) In [305]: df.columns Out[305]: Index([one, two], dtype=object)
52
0 1 2 3
1 2 3 4
4 3 2 1
Note: DataFrame is not intended to work exactly like a 2-dimensional NumPy ndarray.
5.2. DataFrame
53
first second
1 5
2 NaN 10 20
DataFrame.from_items DataFrame.from_items works analogously to the form of the dict constructor that takes a sequence of (key, value) pairs, where the keys are column (or row, in the case of orient=index) names, and the value are the column values (or row values). This can be useful for constructing a DataFrame with the columns in a particular order without having to pass an explicit list of columns:
In [320]: DataFrame.from_items([(A, [1, 2, 3]), (B, [4, 5, 6])]) Out[320]: A B 0 1 4 1 2 5 2 3 6
54
If you pass orient=index, the keys will be the row labels. But in this case you must also pass the desired column names:
In [321]: DataFrame.from_items([(A, [1, 2, 3]), (B, [4, 5, 6])], .....: orient=index, columns=[one, two, three]) .....: Out[321]: one two three A 1 2 3 B 4 5 6
foo bar
5.2. DataFrame
55
b c d
2 3 NaN
When inserting a Series that does not have the same index as the DataFrame, it will be conformed to the DataFrames index:
In [331]: df[one_trunc] = df[one][:2] In [332]: df Out[332]: one flag a 1 False b 2 False c 3 True d NaN False
You can insert raw ndarrays but their length must match the length of the DataFrames index. By default, columns get inserted at the end. The insert function is available to insert at a particular location in the columns:
In [333]: df.insert(1, bar, df[one]) In [334]: df Out[334]: one bar flag a 1 1 False b 2 2 False c 3 3 True d NaN NaN False
Row selection, for example, returns a Series whose index is the columns of the DataFrame:
In [335]: df.xs(b) Out[335]: one 2 bar 2 flag False foo bar one_trunc 2 Name: b In [336]: df.ix[2] Out[336]: one 3 bar 3
56
Note if a DataFrame contains columns of multiple dtypes, the dtype of the row will be chosen to accommodate all of the data types (dtype=object is the most general). For a more exhaustive treatment of more sophisticated label-based indexing and slicing, see the section on indexing. We will address the fundamentals of reindexing / conforming to new sets of lables in the section on reindexing.
D NaN NaN NaN NaN NaN NaN NaN NaN NaN NaN
When doing an operation between DataFrame and Series, the default behavior is to align the Series index on the DataFrame columns, thus broadcasting row-wise. For example:
In [340]: df - df.ix[0] Out[340]: A B C D 0 0.000 0.000 0.000 0.000 1 -0.586 -3.234 -1.119 -2.876 2 -1.136 -3.727 -2.066 -1.500 3 -0.799 -0.803 -1.860 -2.038 4 -1.632 -2.216 -2.384 -1.367 5 -0.485 -2.545 -1.044 -1.190 6 -0.168 0.395 -1.333 -1.060 7 -2.367 -0.799 -3.479 -2.653 8 -2.953 -1.565 -2.251 -2.978 9 -1.615 -1.712 -2.521 -2.106
In the special case of working with time series data, if the Series is a TimeSeries (which it will be automatically if the index contains datetime objects), and the DataFrame index also contains dates, the broadcasting will be column-wise:
In [341]: index = date_range(1/1/2000, periods=8) In [342]: df = DataFrame(randn(8, 3), index=index, .....: columns=[A, B, C])
5.2. DataFrame
57
.....: In [343]: df Out[343]: 2000-01-01 2000-01-02 2000-01-03 2000-01-04 2000-01-05 2000-01-06 2000-01-07 2000-01-08 A -1.209 0.430 1.257 -0.766 -1.566 -0.593 -0.123 -0.389 B -1.257 -0.242 -0.871 -0.219 1.780 -1.059 1.306 0.143 C -0.500 -0.724 -0.544 0.663 -2.139 0.119 -0.129 -1.715
In [344]: type(df[A]) Out[344]: pandas.core.series.TimeSeries In [345]: df - df[A] Out[345]: A B C 2000-01-01 0 -0.048 0.709 2000-01-02 0 -0.672 -1.154 2000-01-03 0 -2.128 -1.801 2000-01-04 0 0.547 1.429 2000-01-05 0 3.346 -0.572 2000-01-06 0 -0.466 0.711 2000-01-07 0 1.429 -0.006 2000-01-08 0 0.532 -1.326
Technical purity aside, this case is so common in practice that supporting the special case is preferable to the alternative of forcing the user to transpose and do column-based alignment like so:
In [346]: (df.T - df[A]).T Out[346]: A B C 2000-01-01 0 -0.048 0.709 2000-01-02 0 -0.672 -1.154 2000-01-03 0 -2.128 -1.801 2000-01-04 0 0.547 1.429 2000-01-05 0 3.346 -0.572 2000-01-06 0 -0.466 0.711 2000-01-07 0 1.429 -0.006 2000-01-08 0 0.532 -1.326
For explicit control over the matching and broadcasting behavior, see the section on exible binary operations. Operations with scalars are just as you would expect:
In [347]: df * 5 + Out[347]: A 2000-01-01 -4.043 2000-01-02 4.149 2000-01-03 8.286 2000-01-04 -1.830 2000-01-05 -5.831 2000-01-06 -0.963 2000-01-07 1.385 2000-01-08 0.054 2 B -4.285 0.789 -2.355 0.907 10.900 -3.296 8.528 2.716 C -0.499 -1.619 -0.719 5.314 -8.693 2.594 1.354 -6.575
58
In [348]: 1 / df Out[348]: A 2000-01-01 -0.827 2000-01-02 2.327 2000-01-03 0.795 2000-01-04 -1.305 2000-01-05 -0.638 2000-01-06 -1.687 2000-01-07 -8.128 2000-01-08 -2.570 In [349]: df ** 4 Out[349]: A 2000-01-01 2.133 2000-01-02 0.034 2000-01-03 2.499 2000-01-04 0.344 2000-01-05 6.018 2000-01-06 0.123 2000-01-07 0.000 2000-01-08 0.023
5.2. DataFrame
59
5.2.10 Transposing
To transpose, access the T attribute (also the transpose function), similar to an ndarray:
# only show the first 5 rows In [356]: df[:5].T Out[356]: 2000-01-01 2000-01-02 2000-01-03 A -1.209 0.430 1.257 B -1.257 -0.242 -0.871 C -0.500 -0.724 -0.544
DataFrame is not intended to be a drop-in replacement for ndarray as its indexing semantics are quite different in places from a matrix.
60
However, using to_string will return a string representation of the DataFrame in tabular form, though it wont always t the console width:
In [364]: print baseball.ix[-20:, :12].to_string() id year stint team lg g ab r 88641 womacto01 2006 2 CHN NL 19 50 6 88643 schilcu01 2006 1 BOS AL 31 2 0 88645 myersmi01 2006 1 NYA AL 62 0 0 88649 helliri01 2006 1 MIL NL 20 3 0 88650 johnsra05 2006 1 NYA AL 33 6 0 88652 finlest01 2006 1 SFN NL 139 426 66 88653 gonzalu01 2006 1 ARI NL 153 586 93 88662 seleaa01 2006 1 LAN NL 28 26 2 89177 francju01 2007 2 ATL NL 15 40 1 89178 francju01 2007 1 NYN NL 40 50 7 89330 zaungr01 2007 1 TOR AL 110 331 43 89333 witasja01 2007 1 TBA AL 3 0 0 89334 williwo02 2007 1 HOU NL 33 59 3 89335 wickmbo01 2007 2 ARI NL 8 0 0 89336 wickmbo01 2007 1 ATL NL 47 0 0 89337 whitero02 2007 1 MIN AL 38 109 8 89338 whiteri01 2007 1 HOU NL 20 1 0 89339 wellsda01 2007 2 LAN NL 7 15 2 89340 wellsda01 2007 1 SDN NL 22 38 1 h 14 1 0 0 1 105 159 5 10 10 80 0 6 0 0 19 0 4 4 X2b 1 0 0 0 0 21 52 1 3 0 24 0 0 0 0 4 0 1 0 X3b 0 0 0 0 0 12 2 0 0 0 1 0 0 0 0 0 0 0 0 hr 1 0 0 0 0 6 15 0 0 1 10 0 1 0 0 4 0 0 0
5.2. DataFrame
61
89341 89343 89345 89347 89348 89352 89354 89355 89359 89360 89361 89363 89365 89366 89367 89368 89370 89371 89372 89374 89375 89378 89381 89382 89383 89384 89385 89388 89389 89396 89398 89400 89402 89406 89410 89411 89412 89420 89421 89425 89426 89429 89430 89431 89438 89439 89442 89445 89450 89451 89452 89460 89462 89463 89464 89465 89466 89467
weathda01 walketo04 wakefti01 vizquom01 villoro01 valenjo03 trachst01 trachst01 timlimi01 thomeji01 thomafr04 tavarju01 sweenma01 sweenma01 suppaje01 stinnke01 stantmi02 stairma01 sprinru01 sosasa01 smoltjo01 sheffga01 seleaa01 seaneru01 schmija01 schilcu01 sandere02 rogerke01 rodriiv01 ramirma02 piazzmi01 perezne01 parkch01 oliveda02 myersmi01 mussimi01 moyerja01 mesajo01 martipe02 maddugr01 mabryjo01 loftoke01 loftoke01 loaizes01 kleskry01 kentje01 jonesto02 johnsra05 hoffmtr01 hernaro01 hernaro01 guarded01 griffke02 greensh01 graffto01 gordoto01 gonzalu01 gomezch02
2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007
1 1 1 1 1 1 2 1 1 1 1 1 2 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 2 1 1 1 1 1 1 1 2 1 1 1 1 1 1 1 2
CIN OAK BOS SFN NYA NYN CHN BAL BOS CHA TOR BOS LAN SFN MIL SLN CIN TOR SLN TEX ATL DET NYN LAN LAN BOS KCA DET DET BOS OAK DET NYN LAA NYA NYA PHI PHI NYN SDN COL CLE TEX LAN SFN LAN DET ARI SDN LAN CLE CIN CIN NYN MIL PHI LAN CLE
NL AL AL NL AL NL NL AL AL AL AL AL NL NL NL NL NL AL NL AL NL AL NL NL NL AL AL AL AL AL AL AL NL AL AL AL NL NL NL NL NL AL AL NL NL NL AL NL NL NL AL NL NL NL NL NL NL AL
67 18 1 145 6 51 4 3 4 130 155 2 30 76 33 26 67 125 72 114 30 133 31 68 6 1 24 1 129 133 83 33 1 5 6 2 33 38 5 33 28 52 84 5 116 136 5 10 60 22 2 15 144 130 86 44 139 19
0 48 2 513 0 166 7 5 0 432 531 4 33 90 61 82 2 357 1 412 54 494 4 1 7 2 73 2 502 483 309 64 1 0 1 2 73 0 9 62 34 173 317 7 362 494 0 15 0 0 0 0 528 446 231 0 464 53
0 5 0 54 0 18 0 0 0 79 63 0 2 18 4 7 0 58 0 53 1 107 0 0 1 0 12 0 50 84 33 5 0 0 0 0 4 0 1 2 4 24 62 0 51 78 0 0 0 0 0 0 78 62 34 0 70 4
0 13 0 126 0 40 1 0 0 119 147 1 9 23 8 13 0 103 0 104 5 131 0 0 1 1 23 0 141 143 85 11 0 0 0 0 9 0 1 9 4 49 96 1 94 149 0 1 0 0 0 0 146 130 55 0 129 15
0 1 0 18 0 11 0 0 0 19 30 0 1 8 0 3 0 28 0 24 1 20 0 0 0 0 7 0 31 33 17 3 0 0 0 0 2 0 1 2 1 9 16 0 27 36 0 0 0 0 0 0 24 30 8 0 23 2
0 0 0 3 0 1 0 0 0 0 0 0 0 0 0 0 0 1 0 1 0 1 0 0 0 0 0 0 3 1 1 0 0 0 0 0 0 0 0 0 0 3 3 0 3 1 0 0 0 0 0 0 1 1 0 0 2 0
0 0 0 4 0 3 0 0 0 35 26 0 0 2 0 1 0 21 0 21 0 25 0 0 1 0 2 0 11 20 8 1 0 0 0 0 0 0 0 0 1 0 7 0 6 20 0 0 0 0 0 0 30 10 9 0 15 0
62
89468 89469 89473 89474 89480 89481 89482 89489 89493 89494 89495 89497 89498 89499 89501 89502 89521 89523 89525 89526 89530 89533 89534
gomezch02 glavito02 floydcl01 finlest01 embreal01 edmonji01 easleda01 delgaca01 cormirh01 coninje01 coninje01 clemero02 claytro01 claytro01 cirilje01 cirilje01 bondsba01 biggicr01 benitar01 benitar01 ausmubr01 aloumo01 alomasa02
2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007 2007
1 1 1 1 1 1 1 1 1 2 1 1 2 1 2 1 1 1 2 1 1 1 1
BAL NYN CHN COL OAK SLN NYN NYN CIN NYN CIN NYA BOS TOR ARI MIN SFN HOU FLO SFN HOU NYN NYN
AL NL NL NL AL NL NL NL NL NL NL AL AL AL NL AL NL NL NL NL NL NL NL
169 56 282 94 0 365 193 538 0 41 215 2 6 189 40 153 340 517 0 0 349 328 22
17 3 40 9 0 39 24 71 0 2 23 0 1 23 6 18 75 68 0 0 38 51 1
10 1 10 3 0 15 6 30 0 2 11 0 0 14 4 9 14 31 0 0 16 19 1
1 0 1 0 0 2 0 0 0 0 1 0 0 0 0 2 0 3 0 0 3 1 0
1 0 9 1 0 12 10 24 0 0 6 0 0 1 0 2 28 10 0 0 3 13 0
The related method get_dtype_counts will return the number of columns of each type:
5.2. DataFrame
63
The columns are also connected to the IPython completion mechanism so they can be tab-completed:
In [5]: df.fo<TAB> df.foo1 df.foo2
5.3 Panel
Panel is a somewhat less-used, but still important container for 3-dimensional data. The term panel data is derived from econometrics and is partially responsible for the name pandas: pan(el)-da(ta)-s. The names for the 3 axes are intended to give some semantic meaning to describing operations involving panel data and, in particular, econometric analysis of panel data. However, for the strict purposes of slicing and dicing a collection of DataFrame objects, you may nd the axis names slightly arbitrary: items: axis 0, each item corresponds to a DataFrame contained inside major_axis: axis 1, it is the index (rows) of each of the DataFrames minor_axis: axis 2, it is the columns of each of the DataFrames Construction of Panels works about like you would expect:
64
Note that the values in the dict need only be convertible to DataFrame. Thus, they can be any of the other valid inputs to DataFrame as per above. One helpful factory method is Panel.from_dict, which takes a dictionary of DataFrames as above, and the following named parameters: Parameter intersect orient Default False items Description drops elements whose indices do not align use minor to use DataFrames columns as panel items
Orient is especially useful for mixed-type DataFrames. If you pass a dict of DataFrame objects with mixed-type columns, all of the data will get upcasted to dtype=object unless you pass orient=minor:
In [375]: df = DataFrame({a: [foo, bar, baz], .....: b: np.random.randn(3)}) .....: In [376]: df Out[376]:
5.3. Panel
65
0 1 2
In [377]: data = {item1: df, item2: df} In [378]: panel = Panel.from_dict(data, orient=minor) In [379]: panel[a] Out[379]: item1 item2 0 foo foo 1 bar bar 2 baz baz In [380]: panel[b] Out[380]: item1 item2 0 0.080597 0.080597 1 -0.000185 -0.000185 2 -0.264704 -0.264704 In [381]: panel[b].dtypes Out[381]: item1 float64 item2 float64
Note: Unfortunately Panel, being less commonly used than Series and DataFrame, has been slightly neglected featurewise. A number of methods and options available in DataFrame are not available in Panel. This will get worked on, of course, in future releases. And faster if you join me in working on the codebase.
66
In [385]: wp[Item1] Out[385]: A B 2000-01-01 -0.519332 -1.765523 2000-01-02 -1.314597 -1.458515 2000-01-03 1.357258 -0.098278 2000-01-04 -1.309989 -1.153000 2000-01-05 -0.289724 -0.996632
The API for insertion and deletion is the same as for DataFrame. And as with DataFrame, if the item is a valid python identier, you can access it as an attribute and tab-complete it in IPython.
5.3.5 Transposing
A Panel can be rearranged using its transpose method (which does not make a copy by default unless the data are heterogeneous):
In [387]: wp.transpose(2, 0, 1) Out[387]: <class pandas.core.panel.Panel> Dimensions: 4 (items) x 3 (major) x 5 (minor) Items: A to D Major axis: Item1 to Item3 Minor axis: 2000-01-01 00:00:00 to 2000-01-05 00:00:00
In [389]: wp.major_xs(wp.major_axis[2]) Out[389]: Item1 Item2 Item3 A 1.357258 -0.177665 -7.639427 B -0.098278 0.490838 -0.200224 C -0.987183 -1.360102 0.725815 D -1.362030 1.592456 -0.855302 In [390]: wp.minor_axis Out[390]: Index([A, B, C, D], dtype=object)
5.3. Panel
67
In [391]: wp.minor_xs(C) Out[391]: Item1 Item2 Item3 2000-01-01 -0.966196 0.071823 -13.452418 2000-01-02 -0.919663 0.214910 -4.279288 2000-01-03 -0.987183 -1.360102 0.725815 2000-01-04 0.606382 -1.890591 -0.320737 2000-01-05 -1.407699 -0.151652 9.282439
two 0.597468 -0.491240 1.281674 -0.099685 -1.823043 -0.779213 -0.949327 0.768043 -0.054860 -1.493561 0.106004 -0.903513 -0.719875 0.301945 1.112546 -0.542770 -2.695540 0.431284 -0.431092 1.666631
three 0.716659 -0.919717 -0.024595 0.068997 0.601797 0.866615 0.092911 -2.606892 0.309303 -0.548401 -2.044772 -1.666264 -1.439775 1.326361 0.221680 1.840992 1.165150 -1.420521 1.616679 -1.030912
68
CHAPTER
SIX
69
To get the actual data inside a data structure, one need only access the values property:
In [11]: s.values Out[11]: array([-1.2667, -0.6215, -0.4416, -0.5841, -0.7332]) In [12]: df.values Out[12]: array([[-0.8895, -0.4644, [ 0.8556, 1.0867, [-0.6869, 0.7678, [ 0.8407, 0.0279, [-0.6334, 1.3072, [-0.5685, -0.7225, [ 0.291 , 1.1988, [ 1.1352, -0.1978,
In [13]: wp.values Out[13]: array([[[ 1.4938, 0.8956, 0.6084, -1.0565], [-0.9734, -0.2901, 0.1716, 0.1655], [ 0.9815, 0.7882, 0.8776, -0.2275], [-1.107 , -0.3737, -0.9902, 0.2331], [ 1.1622, 0.9497, -1.1794, -0.5034]], [[-1.3449, -0.5046, -0.7243, -0.5848], [-0.8746, 1.164 , 1.6359, -1.1039],
70
[-0.0756, 0.2406, -1.7114, -0.9742], [-0.3385, -1.0587, 1.203 , -0.4742], [-0.1466, -1.1496, 0.4858, -0.3914]]])
If a DataFrame or Panel contains homogeneously-typed data, the ndarray can actually be modied in-place, and the changes will be reected in the data structure. For heterogeneous data (e.g. some of the DataFrames 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 oats and integers, the resulting array will be of oat dtype.
71
In [20]: df.sub(row, axis=1) Out[20]: one three two a 2.001853 NaN 0.836368 b 0.000000 0.000000 0.000000 c 0.961343 0.081006 -0.331258 d NaN -0.213803 -1.238650 In [21]: df.sub(column, axis=index) Out[21]: one three two a -1.298036 NaN 0 b -2.463520 -0.877310 0 c -1.170919 -0.465046 0 d NaN 0.147537 0 In [22]: df.sub(column, axis=0) Out[22]: one three two a -1.298036 NaN 0 b -2.463520 -0.877310 0 c -1.170919 -0.465046 0 d NaN 0.147537 0
With Panel, describing the matching behavior is a bit more difcult, 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 [23]: major_mean = wp.mean(axis=major) In [24]: major_mean Out[24]: Item1 Item2 A 0.311415 -0.556026 B 0.393960 -0.261680 C -0.102432 0.177819 D -0.277772 -0.705683 In [25]: wp.sub(major_mean, axis=major) Out[25]: <class pandas.core.panel.Panel> Dimensions: 2 (items) x 5 (major) x 4 (minor) Items: Item1 to Item2 Major axis: 2000-01-01 00:00:00 to 2000-01-05 00:00:00 Minor 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...
72
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 [26]: df Out[26]: one three two a -0.010168 NaN 1.287868 b -2.012021 -0.425811 0.451499 c -1.050678 -0.344805 0.120241 d NaN -0.639613 -0.787151 In [27]: df2 Out[27]: one three two a -0.010168 1.000000 1.287868 b -2.012021 -0.425811 0.451499 c -1.050678 -0.344805 0.120241 d NaN -0.639613 -0.787151 In [28]: df Out[28]: one a -0.020336 b -4.024042 c -2.101355 d NaN + df2 three two NaN 2.575735 -0.851621 0.902999 -0.689609 0.240482 -1.279227 -1.574301
In [29]: df.add(df2, fill_value=0) Out[29]: one three two a -0.020336 1.000000 2.575735 b -4.024042 -0.851621 0.902999 c -2.101355 -0.689609 0.240482 d NaN -1.279227 -1.574301
73
74
Out[38]: A B 0 1 NaN 1 2 2 2 3 3 3 5 4 4 3 6 5 7 8
All such methods have a skipna option signaling whether to exclude missing data (True by default):
In [42]: df.sum(0, skipna=False) Out[42]: one NaN three NaN two 1.072457 In [43]: df.sum(axis=1, skipna=True) Out[43]: a 1.277699 b -1.986332
75
c d
-1.275241 -1.426764
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 [44]: ts_stand = (df - df.mean()) / df.std() In [45]: Out[45]: one three two ts_stand.std() 1 1 1
Note that methods like cumsum and cumprod preserve the location of NA values:
In [48]: df.cumsum() Out[48]: one three a -0.010168 NaN b -2.022189 -0.425811 c -3.072867 -0.770615 d NaN -1.410229
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 count sum mean mad median min max abs prod std var skew kurt quantile cumsum cumprod cummax cummin Description Number of non-null observations Sum of values Mean of values Mean absolute deviation Arithmetic median of values Minimum Maximum Absolute Value Product of values Unbiased standard deviation Unbiased variance Unbiased skewness (3rd moment) Unbiased kurtosis (4th moment) Sample quantile (value at %) Cumulative sum Cumulative product Cumulative maximum Cumulative minimum
Note that by chance some NumPy methods, like mean, std, and sum, will exclude NAs on Series input by default:
76
Series also has a method nunique which will return the number of unique non-null values:
In [51]: series = Series(randn(500)) In [52]: series[20:500] = np.nan In [53]: series[10:20] = 5
For a non-numerical Series object, describe will give a simple summary of the number of unique values and most frequently occurring values:
77
In [61]: s = Series([a, a, b, b, a, a, np.nan, c, d, a]) In [62]: s.describe() Out[62]: count 9 unique 4 top a freq 5
There also is a utility function, value_range which takes a DataFrame and returns a series with the minimum/maximum values in the DataFrame.
When there are multiple rows (or columns) matching the minimum or maximum value, idxmin and idxmax return the rst matching index:
78
In [70]: df3 = DataFrame([2, 1, 1, 3, np.nan], columns=[A], index=list(edcba)) In [71]: df3 Out[71]: A e 2 d 1 c 1 b 3 a NaN In [72]: df3[A].idxmin() Out[72]: d
79
In [78]: arr = np.random.randn(20) In [79]: factor = cut(arr, 4) In [80]: factor Out[80]: Categorical: array([(-0.228, 0.529], (-0.228, 0.529], (-0.987, -0.228], (-0.987, -0.228], (-0.228, 0.529], (-0.987, -0.228], (0.529, 1.285], (-0.228, 0.529], (-0.228, 0.529], (0.529, 1.285], (-0.228, 0.529], (-0.228, 0.529], (-0.987, -0.228], (-0.228, 0.529], (-0.228, 0.529], (1.285, 2.0414], (1.285, 2.0414], (-0.987, -0.228], (-0.228, 0.529], (1.285, 2.0414]], dtype=object) Levels (4): Index([(-0.987, -0.228], (-0.228, 0.529], (0.529, 1.285], (1.285, 2.0414]], dtype=object) In [81]: factor = cut(arr, [-5, -1, 0, 1, 5]) In [82]: factor Out[82]: Categorical: array([(0, 1], (0, 1], (-1, 0], (-1, 0], (-1, 0], (-1, 0], (0, 1], (0, 1], (-1, 0], (0, 1], (0, 1], (0, 1], (-1, 0], (-1, 0], (0, 1], (1, 5], (1, 5], (-1, 0], (0, 1], (1, 5]], 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 [83]: arr = np.random.randn(30) In [84]: factor = qcut(arr, [0, .25, .5, .75, 1]) In [85]: factor Out[85]: Categorical: array([(-0.713, -0.0994], [-2.00807, -0.713], (0.714, 3.0441], (-0.0994, 0.714], (-0.0994, 0.714], (-0.0994, 0.714], (0.714, 3.0441], [-2.00807, -0.713], (0.714, 3.0441], (-0.0994, 0.714], (-0.0994, 0.714], [-2.00807, -0.713], (0.714, 3.0441], (0.714, 3.0441], (-0.713, -0.0994], (0.714, 3.0441], (-0.713, -0.0994], [-2.00807, -0.713], [-2.00807, -0.713], (-0.713, -0.0994], (0.714, 3.0441], [-2.00807, -0.713], [-2.00807, -0.713], [-2.00807, -0.713], (-0.713, -0.0994], (-0.0994, 0.714], (0.714, 3.0441], (-0.713, -0.0994], (-0.713, -0.0994], (-0.0994, 0.714]], dtype=object) Levels (4): Index([[-2.00807, -0.713], (-0.713, -0.0994], (-0.0994, 0.714], (0.714, 3.0441]], dtype=object) In [86]: value_counts(factor) Out[86]: [-2.00807, -0.713] 8 (0.714, 3.0441] 8 (-0.713, -0.0994] 7 (-0.0994, 0.714] 7
80
In [90]: df.apply(np.cumsum) Out[90]: one three two a -0.010168 NaN 1.287868 b -2.022189 -0.425811 1.739367 c -3.072867 -0.770615 1.859608 d NaN -1.410229 1.072457 In [91]: df.apply(np.exp) Out[91]: one three two a 0.989883 NaN 3.625048 b 0.133718 0.653240 1.570665 c 0.349701 0.708359 1.127769 d NaN 0.527496 0.455140
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 [92]: tsdf = DataFrame(randn(1000, 3), columns=[A, B, C], ....: index=date_range(1/1/2000, periods=1000)) ....: In [93]: tsdf.apply(lambda x: x.index[x.dropna().argmax()]) Out[93]: A 2000-08-25 00:00:00 B 2000-05-03 00:00:00 C 2000-06-30 00:00:00
You may also pass additional arguments and keyword arguments to the apply method. For instance, consider the following function you would like to apply:
81
Another useful feature is the ability to pass Series methods to carry out some Series operation on each column or row:
In [94]: tsdf Out[94]: A B C 2000-01-01 0.874043 0.983288 -0.311235 2000-01-02 -0.360482 -1.620920 -0.965061 2000-01-03 0.215272 1.386449 1.121756 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 2.472721 1.112620 2.047303 2000-01-09 0.201261 1.304873 0.418457 2000-01-10 1.424382 1.501215 0.571983 In [95]: tsdf.apply(Series.interpolate) Out[95]: A B C 2000-01-01 0.874043 0.983288 -0.311235 2000-01-02 -0.360482 -1.620920 -0.965061 2000-01-03 0.215272 1.386449 1.121756 2000-01-04 0.666762 1.331683 1.306865 2000-01-05 1.118252 1.276917 1.491974 2000-01-06 1.569742 1.222152 1.677084 2000-01-07 2.021232 1.167386 1.862193 2000-01-08 2.472721 1.112620 2.047303 2000-01-09 0.201261 1.304873 0.418457 2000-01-10 1.424382 1.501215 0.571983
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, exible functionality for grouping by some criterion, applying, and combining the results into a Series, DataFrame, etc.
82
Series.map has an additional feature which is that it can be used to easily link or map values dened by a secondary series. This is closely related to merging/joining functionality:
In [99]: s = Series([six, seven, six, seven, six], ....: index=[a, b, c, d, e]) ....: In [100]: t = Series({six : 6., seven : 7.}) In [101]: s Out[101]: a six b seven c six d seven e six In [102]: s.map(t) Out[102]: a 6 b 7 c 6 d 7 e 6
83
d e
-0.064418 -1.881446
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 [106]: df Out[106]: one three two a -0.010168 NaN 1.287868 b -2.012021 -0.425811 0.451499 c -1.050678 -0.344805 0.120241 d NaN -0.639613 -0.787151 In [107]: df.reindex(index=[c, f, b], columns=[three, two, one]) Out[107]: three two one c -0.344805 0.120241 -1.050678 f NaN NaN NaN b -0.425811 0.451499 -2.012021
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 [108]: rs = s.reindex(df.index) In [109]: rs Out[109]: a 0.850486 b 0.319157 c 1.080624 d -0.064418 In [110]: rs.index is df.index Out[110]: True
This means that the reindexed Seriess index is the same Python object as the DataFrames 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.
84
6.6.2 Reindexing with reindex_axis 6.6.3 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 objects index join=right: use the passed objects index join=inner: intersect the indexes It returns a tuple with both of the reindexed Series:
In [114]: s = Series(randn(5), index=[a, b, c, d, e]) In [115]: s1 = s[:4] In [116]: s2 = s[1:] In [117]: s1.align(s2) Out[117]: (a 0.253571 b -0.161640 c 1.950726 d 0.208047 e NaN,
85
a b c d e
In [118]: s1.align(s2, join=inner) Out[118]: (b -0.161640 c 1.950726 d 0.208047, b -0.161640 c 1.950726 d 0.208047) In [119]: s1.align(s2, join=left) Out[119]: (a 0.253571 b -0.161640 c 1.950726 d 0.208047, a NaN b -0.161640 c 1.950726 d 0.208047)
For DataFrames, the join method will be applied to both the index and the columns by default:
In [120]: df.align(df2, join=inner) Out[120]: ( one two a -0.010168 1.287868 b -2.012021 0.451499 c -1.050678 0.120241, one two a 1.014121 0.667998 b -0.987732 -0.168370 c -0.026389 -0.499628)
You can also pass an axis option to only align on the specied axis:
In [121]: df.align(df2, join=inner, axis=0) Out[121]: ( one three two a -0.010168 NaN 1.287868 b -2.012021 -0.425811 0.451499 c -1.050678 -0.344805 0.120241, one two a 1.014121 0.667998 b -0.987732 -0.168370 c -0.026389 -0.499628)
If you pass a Series to DataFrame.align, you can choose to align both objects either on the DataFrames index or columns using the axis argument:
In [122]: df.align(df2.ix[0], axis=1) Out[122]: ( one three two a -0.010168 NaN 1.287868
86
b -2.012021 -0.425811 0.451499 c -1.050678 -0.344805 0.120241 d NaN -0.639613 -0.787151, one 1.014121 three NaN two 0.667998 Name: a)
Other ll 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 ll methods on a simple TimeSeries:
In [123]: rng = date_range(1/3/2000, periods=8) In [124]: ts = Series(randn(8), index=rng) In [125]: ts2 = ts[[0, 3, 6]] In [126]: ts Out[126]: 2000-01-03 2000-01-04 2000-01-05 2000-01-06 2000-01-07 2000-01-08 2000-01-09 2000-01-10 Freq: D
In [127]: ts2 Out[127]: 2000-01-03 0.577077 2000-01-06 1.267783 2000-01-09 -0.522009 In [128]: ts2.reindex(ts.index) Out[128]: 2000-01-03 0.577077 2000-01-04 NaN 2000-01-05 NaN 2000-01-06 1.267783 2000-01-07 NaN 2000-01-08 NaN 2000-01-09 -0.522009 2000-01-10 NaN
87
Freq: D In [129]: ts2.reindex(ts.index, method=ffill) Out[129]: 2000-01-03 0.577077 2000-01-04 0.577077 2000-01-05 0.577077 2000-01-06 1.267783 2000-01-07 1.267783 2000-01-08 1.267783 2000-01-09 -0.522009 2000-01-10 -0.522009 Freq: D In [130]: ts2.reindex(ts.index, method=bfill) Out[130]: 2000-01-03 0.577077 2000-01-04 1.267783 2000-01-05 1.267783 2000-01-06 1.267783 2000-01-07 -0.522009 2000-01-08 -0.522009 2000-01-09 -0.522009 2000-01-10 NaN Freq: D
Note the same result could have been achieved using llna:
In [131]: ts2.reindex(ts.index).fillna(method=ffill) Out[131]: 2000-01-03 0.577077 2000-01-04 0.577077 2000-01-05 0.577077 2000-01-06 1.267783 2000-01-07 1.267783 2000-01-08 1.267783 2000-01-09 -0.522009 2000-01-10 -0.522009 Freq: D
Note these methods generally assume that the indexes are sorted. They may be modied in the future to be a bit more exible but as time series data is ordered most of the time anyway, this has not been a major priority.
88
In [134]: df.drop([one], axis=1) Out[134]: three two a NaN 1.287868 b -0.425811 0.451499 c -0.344805 0.120241 d -0.639613 -0.787151
Note that the following also works, but is a bit less obvious / clean:
In [135]: df.reindex(df.index - [a, d]) Out[135]: one three two b -2.012021 -0.425811 0.451499 c -1.050678 -0.344805 0.120241
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 [138]: df.rename(columns={one : foo, two : bar}, .....: index={a : apple, b : banana, d : durian}) .....: Out[138]: foo three bar apple -0.010168 NaN 1.287868 banana -2.012021 -0.425811 0.451499 c -1.050678 -0.344805 0.120241 durian NaN -0.639613 -0.787151
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.
89
6.7 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 [139]: for col in df: .....: print col .....: one three two
6.7.1 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 [140]: for item, frame in wp.iteritems(): .....: print item .....: print frame .....: Item1 A B C D 2000-01-01 1.493806 0.895554 0.608352 -1.056488 2000-01-02 -0.973447 -0.290059 0.171564 0.165464 2000-01-03 0.981531 0.788226 0.877571 -0.227514 2000-01-04 -1.107011 -0.373661 -0.990247 0.233063 2000-01-05 1.162199 0.949740 -1.179397 -0.503387 Item2 A B C D 2000-01-01 -1.344904 -0.504629 -0.724263 -0.584786 2000-01-02 -0.874568 1.164009 1.635948 -1.103886 2000-01-03 -0.075590 0.240562 -1.711394 -0.974197 2000-01-04 -0.338477 -1.058711 1.203003 -0.474163 2000-01-05 -0.146593 -1.149633 0.485802 -0.391383
6.7.2 iterrows
New in v0.7 is the ability to iterate efciently through rows of a DataFrame. It returns an iterator yielding each index value along with a Series containing the data in each row:
90
In [141]: for row_index, row in df2.iterrows(): .....: print %s\n%s % (row_index, row) .....: a one 1.014121 two 0.667998 Name: a b one -0.987732 two -0.168370 Name: b c one -0.026389 two -0.499628 Name: c
6.7.3 itertuples
This method will return an iterator yielding a tuple for each row in the DataFrame. The rst element of the tuple will be the rows corresponding index value, while the remaining values are the row values proper. For instance,
In [147]: for r in df2.itertuples(): print r (0, 1, 4) (1, 2, 5) (2, 3, 6)
91
are accessed via the Seriess str attribute and generally have names matching the equivalent (scalar) build-in string methods:
In [148]: s = Series([A, B, C, Aaba, Baca, np.nan, CABA, dog, cat]) In [149]: s.str.lower() Out[149]: 0 a 1 b 2 c 3 aaba 4 baca 5 NaN 6 caba 7 dog 8 cat In [150]: s.str.upper() Out[150]: 0 A 1 B 2 C 3 AABA 4 BACA 5 NaN 6 CABA 7 DOG 8 CAT In [151]: s.str.len() Out[151]: 0 1 1 1 2 1 3 4 4 4 5 NaN 6 4 7 3 8 3
92
Methods like contains, startswith, and endswith takes an extra na arguement so missing values can be considered True or False:
In [159]: s4 = Series([A, B, C, Aaba, Baca, np.nan, CABA, dog, cat]) In [160]: s4.str.contains(A, na=False) Out[160]: 0 True 1 False 2 False 3 True 4 False 5 NaN 6 True 7 False 8 False
93
Method cat split get join contains replace repeat pad center slice slice_replace count startswith endswidth findall match len strip rstrip lstrip lower upper
Description Concatenate strings Split strings on delimiter Index into each element (retrieve i-th element) Join strings in each element of the Series with passed separator Return boolean array if each string contains pattern/regex Replace occurrences of pattern/regex with some other string Duplicate values (s.str.repeat(3) equivalent to x * 3) Add whitespace to left, right, or both sides of strings Equivalent to pad(side=both) Slice each string in the Series Replace slice in each string with passed value Count occurrences of pattern Equivalent to str.startswith(pat) for each element Equivalent to str.endswith(pat) for each element Compute list of all occurrences of pattern/regex for each string Call re.match on each element, returning matched groups as list Compute string lengths Equivalent to str.strip Equivalent to str.rstrip Equivalent to str.lstrip Equivalent to str.lower Equivalent to str.upper
94
0.120241 0.451499
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 [165]: df.sort_index(by=two) Out[165]: one three two d NaN -0.639613 -0.787151 c -1.050678 -0.344805 0.120241 b -2.012021 -0.425811 0.451499 a -0.010168 NaN 1.287868
Series has the method order (analogous to Rs order function) which sorts by value, with special treatment of NA values via the na_last argument:
In [168]: s[2] = np.nan In [169]: s.order() Out[169]: 0 A 3 Aaba 1 B 4 Baca 6 CABA 8 cat 7 dog 2 NaN 5 NaN In [170]: s.order(na_last=False) Out[170]: 2 NaN 5 NaN 0 A 3 Aaba 1 B 4 Baca 6 CABA 8 cat 7 dog
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.
95
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.
96
In [180]: df.dtypes Out[180]: a object b object c object d object In [181]: converted = df.convert_objects() In [182]: converted.dtypes Out[182]: a float64 b float64 c float64 d object
In [184]: 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 le:
In [185]: load(foo.pickle) Out[185]: a b c 0 -0.4866361 -0.4972392 -2.025424 1 -0.08565091 0.1053943 0.4535188 2 -0.2709468 -0.4031711 -1.775506 3 0.3699833 0.1122724 -0.8405974 4 2.115236 -1.3988 -0.161103 5 0.7263931 -0.6230891 0.05823471
There is also a save function which takes any object as its rst argument:
In [186]: save(df, foo.pickle) In [187]: load(foo.pickle) Out[187]: a b c 0 -0.4866361 -0.4972392 -2.025424 1 -0.08565091 0.1053943 0.4535188 2 -0.2709468 -0.4031711 -1.775506
97
3 4 5
The set_printoptions function has a number of options for controlling how oating 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 t the DataFrames string representation into the current terminal width, and defaulting to the summary view otherwise.
98
CHAPTER
SEVEN
7.1 Basics
As mentioned when introducing the data structures in the last section, the primary function of indexing with [] (a.k.a. __getitem__ for those familiar with implementing class behavior in Python) is selecting out lower-dimensional slices. Thus, Series: series[label] returns a scalar value DataFrame: frame[colname] returns a Series corresponding to the passed column name Panel: panel[itemname] returns a DataFrame corresponding to the passed item name Here we construct a simple time series data set to use for illustrating the indexing functionality:
In [552]: dates = np.asarray(date_range(1/1/2000, periods=8)) In [553]: df = DataFrame(randn(8, 4), index=dates, columns=[A, B, C, D]) In [554]: df Out[554]: 2000-01-01 2000-01-02 2000-01-03 2000-01-04 2000-01-05 2000-01-06 2000-01-07 2000-01-08 A 0.469112 1.212112 -0.861849 0.721555 -0.424972 -0.673690 0.404705 -0.370647 B -0.282863 -0.173215 -2.104569 -0.706771 0.567020 0.113648 0.577046 -1.157892 C -1.509059 0.119209 -0.494929 -1.039575 0.276232 -1.478427 -1.715002 -1.344312 D -1.135632 -1.044236 1.071804 0.271860 -1.087401 0.524988 -1.039268 0.844885
99
In [555]: panel = Panel({one : df, two : df - df.mean()}) In [556]: panel Out[556]: <class pandas.core.panel.Panel> Dimensions: 2 (items) x 8 (major) x 4 (minor) Items: one to two Major axis: 2000-01-01 00:00:00 to 2000-01-08 00:00:00 Minor axis: A to D
Note: None of the indexing functionality is time series specic unless specically stated. Thus, as per above, we have the most basic indexing using []:
In [557]: s = df[A] In [558]: s[dates[5]] Out[558]: -0.67368970808837059 In [559]: panel[two] Out[559]: A B 2000-01-01 0.409571 0.113086 2000-01-02 1.152571 0.222735 2000-01-03 -0.921390 -1.708620 2000-01-04 0.662014 -0.310822 2000-01-05 -0.484513 0.962970 2000-01-06 -0.733231 0.509598 2000-01-07 0.345164 0.972995 2000-01-08 -0.430188 -0.761943
There is an analogous set_value method which has the additional capability of enlarging an object. This method always returns a reference to the object it modied, which in the case of enlargement, will be a new object:
In [562]: df.set_value(dates[5], E, 7) Out[562]: A B C D E 2000-01-01 0.469112 -0.282863 -1.509059 -1.135632 NaN 2000-01-02 1.212112 -0.173215 0.119209 -1.044236 NaN 2000-01-03 -0.861849 -2.104569 -0.494929 1.071804 NaN 2000-01-04 0.721555 -0.706771 -1.039575 0.271860 NaN 2000-01-05 -0.424972 0.567020 0.276232 -1.087401 NaN 2000-01-06 -0.673690 0.113648 -1.478427 0.524988 7
100
2000-01-07 0.404705 0.577046 -1.715002 -1.039268 NaN 2000-01-08 -0.370647 -1.157892 -1.344312 0.844885 NaN
If you are using the IPython environment, you may also use tab-completion to see the accessible columns of a DataFrame. You can pass a list of columns to [] to select columns in that order: If a column is not contained in the DataFrame, an exception will be raised. Multiple columns can also be set in this manner:
In [564]: df Out[564]: 2000-01-01 2000-01-02 2000-01-03 2000-01-04 2000-01-05 2000-01-06 2000-01-07 2000-01-08 A 0.469112 1.212112 -0.861849 0.721555 -0.424972 -0.673690 0.404705 -0.370647 B -0.282863 -0.173215 -2.104569 -0.706771 0.567020 0.113648 0.577046 -1.157892 C -1.509059 0.119209 -0.494929 -1.039575 0.276232 -1.478427 -1.715002 -1.344312 D -1.135632 -1.044236 1.071804 0.271860 -1.087401 0.524988 -1.039268 0.844885
In [565]: df[[B, A]] = df[[A, B]] In [566]: df Out[566]: 2000-01-01 2000-01-02 2000-01-03 2000-01-04 2000-01-05 2000-01-06 2000-01-07 2000-01-08 A -0.282863 -0.173215 -2.104569 -0.706771 0.567020 0.113648 0.577046 -1.157892 B 0.469112 1.212112 -0.861849 0.721555 -0.424972 -0.673690 0.404705 -0.370647 C -1.509059 0.119209 -0.494929 -1.039575 0.276232 -1.478427 -1.715002 -1.344312 D -1.135632 -1.044236 1.071804 0.271860 -1.087401 0.524988 -1.039268 0.844885
You may nd this useful for applying a transform (in-place) to a subset of the columns.
major_xs and minor_xs functions for retrieving slices as DataFrames for a given major_axis or minor_axis label, respectively.
In [567]: date = dates[5] In [568]: df.xs(date) Out[568]: A 0.113648 B -0.673690 C -1.478427 D 0.524988 Name: 2000-01-06 00:00:00 In [569]: panel.major_xs(date) Out[569]: one two A -0.673690 -0.733231 B 0.113648 0.509598 C -1.478427 -0.580194 D 0.524988 0.724113 In [570]: panel.minor_xs(A) Out[570]: one two 2000-01-01 0.469112 0.409571 2000-01-02 1.212112 1.152571 2000-01-03 -0.861849 -0.921390 2000-01-04 0.721555 0.662014 2000-01-05 -0.424972 -0.484513 2000-01-06 -0.673690 -0.733231 2000-01-07 0.404705 0.345164 2000-01-08 -0.370647 -0.430188
102
Out[573]: 2000-01-08 2000-01-07 2000-01-06 2000-01-05 2000-01-04 2000-01-03 2000-01-02 2000-01-01 Name: A
With DataFrame, slicing inside of [] slices the rows. This is provided largely as a convenience since it is such a common operation.
In [577]: df[:3] Out[577]: A B C D 2000-01-01 -0.282863 0.469112 -1.509059 -1.135632 2000-01-02 -0.173215 1.212112 0.119209 -1.044236 2000-01-03 -2.104569 -0.861849 -0.494929 1.071804 In [578]: df[::-1] Out[578]: 2000-01-08 2000-01-07 2000-01-06 2000-01-05 2000-01-04 2000-01-03 2000-01-02 2000-01-01 A -1.157892 0.577046 0.113648 0.567020 -0.706771 -2.104569 -0.173215 -0.282863 B -0.370647 0.404705 -0.673690 -0.424972 0.721555 -0.861849 1.212112 0.469112 C -1.344312 -1.715002 -1.478427 0.276232 -1.039575 -0.494929 0.119209 -1.509059 D 0.844885 -1.039268 0.524988 -1.087401 0.271860 1.071804 -1.044236 -1.135632
7.1. Basics
103
In [580]: s[(s < 0) & (s > -0.5)] Out[580]: 2000-01-01 -0.282863 2000-01-02 -0.173215 Name: A
You may select rows from a DataFrame using a boolean vector the same length as the DataFrames index (for example, something derived from one of the columns of the DataFrame):
In [581]: df[df[A] > 0] Out[581]: A B C D 2000-01-05 0.567020 -0.424972 0.276232 -1.087401 2000-01-06 0.113648 -0.673690 -1.478427 0.524988 2000-01-07 0.577046 0.404705 -1.715002 -1.039268
Consider the isin method of Series, which returns a boolean vector that is true wherever the Series elements exist in the passed list. This allows you to select rows where one or more columns have values you want:
In [582]: df2 = DataFrame({a : [one, one, two, three, two, one, six], .....: b : [x, y, y, x, y, x, x], .....: c : randn(7)}) .....: In [583]: df2[df2[a].isin([one, two])] Out[583]: a b c 0 one x 1.075770 1 one y -0.109050 2 two y 1.643563 4 two y 0.357021 5 one x -0.674600
List comprehensions and map method of Series can also be used to produce more complex criteria:
# only want two or three In [584]: criterion = df2[a].map(lambda x: x.startswith(t)) In [585]: Out[585]: a 2 two 3 three 4 two df2[criterion] b c y 1.643563 x -1.469388 y 0.357021
# equivalent but slower In [586]: df2[[x.startswith(t) for x in df2[a]]] Out[586]: a b c 2 two y 1.643563 3 three x -1.469388 4 two y 0.357021
104
# Multiple criteria In [587]: df2[criterion & (df2[b] == x)] Out[587]: a b c 3 three x -1.469388
Note, with the advanced indexing ix method, you may select along more than one axis using boolean vectors combined with other indexing expressions.
In [590]: df2[df2 < 0] = 0 In [591]: df2 Out[591]: 2000-01-01 2000-01-02 2000-01-03 2000-01-04 2000-01-05 2000-01-06 2000-01-07 2000-01-08 A 0.000000 0.000000 0.000000 0.000000 0.567020 0.113648 0.577046 0.000000 B 0.469112 1.212112 0.000000 0.721555 0.000000 0.000000 0.404705 0.000000 C 0.000000 0.119209 0.000000 0.000000 0.276232 0.000000 0.000000 0.000000 D 0.000000 0.000000 1.071804 0.271860 0.000000 0.524988 0.000000 0.844885
Note that such an operation requires that the boolean DataFrame is indexed exactly the same.
7.1. Basics
105
In [595]: index[positions] Out[595]: Index([969, 658, 195], dtype=int64) In [596]: index.take(positions) Out[596]: Index([969, 658, 195], dtype=int64) In [597]: ser = Series(randn(10)) In [598]: ser.ix[positions] Out[598]: 0 -0.968914 9 -1.131345 3 1.247642 In [599]: ser.take(positions) Out[599]: 0 -0.968914 9 -1.131345 3 1.247642
For DataFrames, the given indices should be a 1d list or ndarray that species row or column positions.
In [600]: frm = DataFrame(randn(5, 3)) In [601]: frm.take([1, 4, 3]) Out[601]: 0 1 2 1 -0.932132 1.956030 0.017587 4 -0.077118 -0.408530 -0.862495 3 -1.143704 0.215897 1.193555 In [602]: frm.take([0, 2], axis=1) Out[602]: 0 2 0 -0.089329 -0.945867 1 -0.932132 0.017587 2 -0.016692 0.254161 3 -1.143704 1.193555 4 -0.077118 -0.862495
It is important to note that the take method on pandas objects are not intended to work on boolean indices and may return unexpected results.
In [603]: arr = randn(10) In [604]: arr.take([False, False, True, True]) Out[604]: array([ 1.3461, 1.3461, 1.5118, 1.5118]) In [605]: arr[[0, 1]] Out[605]: array([ 1.3461,
1.5118])
In [606]: ser = Series(randn(10)) In [607]: ser.take([False, False, True, True]) Out[607]: 0 -0.105381 0 -0.105381 1 -0.532532
106
-0.532532
Finally, as a small note on performance, because the take method handles a narrower range of inputs, it can offer performance that is a good deal faster than fancy indexing.
7.1. Basics
107
108
-0.173215 1.212112 0.119209 -1.044236 -2.104569 -0.861849 -0.494929 1.071804 -0.706771 0.000000 0.000000 0.271860 0.567020 0.000000 0.000000 -1.087401 0.113648 0.000000 0.000000 0.524988 0.577046 0.404705 -1.715002 -1.039268 -1.157892 -0.370647 -1.344312 0.844885
Slicing with labels is semantically slightly different because the slice start and stop are inclusive in the label-based case:
In [625]: date1, date2 = dates[[2, 4]] In [626]: print date1, date2 1970-01-11 232:00:00 1970-01-11 24:00:00 In [627]: df.ix[date1:date2] Out[627]: Empty DataFrame Columns: [A, B, C, D] Index: [] In [628]: df[A].ix[date1:date2] Out[628]: TimeSeries([], dtype=float64)
Getting and setting rows in a DataFrame, especially by their location, is much easier:
In [629]: df2 = df[:5].copy() In [630]: df2.ix[3] Out[630]: A -0.706771
109
B 0.721555 C -1.039575 D 0.271860 Name: 2000-01-04 00:00:00 In [631]: df2.ix[3] = np.arange(len(df2.columns)) In [632]: df2 Out[632]: A B C D 2000-01-01 -0.282863 0.469112 -1.509059 -1.135632 2000-01-02 -0.173215 1.212112 0.119209 -1.044236 2000-01-03 -2.104569 -0.861849 -0.494929 1.071804 2000-01-04 0.000000 1.000000 2.000000 3.000000 2000-01-05 0.567020 -0.424972 0.276232 -1.087401
Column or row selection can be combined as you would expect with arrays of labels or even boolean vectors:
In [633]: df.ix[df[A] > 0, B] Out[633]: 2000-01-05 -0.424972 2000-01-06 -0.673690 2000-01-07 0.404705 Name: B In [634]: df.ix[date1:date2, B] Out[634]: TimeSeries([], dtype=float64) In [635]: df.ix[date1, B] Out[635]: -0.86184896334779992
Slicing with labels is closely related to the truncate method which does precisely .ix[start:stop] but returns a copy (for legacy reasons).
110
This deliberate decision was made to prevent ambiguities and subtle bugs (many users reported nding bugs when the API change was made to stop falling back on position-based indexing).
In [642]: df2.ix[2] = np.nan In [643]: print df2 A 2000-01-01 -0.282863 2000-01-02 -0.173215 2000-01-03 NaN 2000-01-04 -0.706771 B C D 0.469112 -1.509059 -1.135632 1.212112 0.119209 -1.044236 NaN NaN NaN 0.721555 -1.039575 0.271860 foo bar bar NaN bar
111
Starting with pandas 0.5, the name, if set, will be shown in the console display:
In [650]: index = Index(range(5), name=rows) In [651]: columns = Index([A, B, C], name=cols) In [652]: df = DataFrame(np.random.randn(5, 3), index=index, columns=columns) In [653]: df Out[653]: cols A B C rows 0 0.192451 0.629675 -1.425966 1 1.857704 -1.193545 0.677510 2 -0.153931 0.520091 -1.475051 3 0.722570 -0.322646 -1.601631 4 0.778033 -0.289342 0.233141 In [654]: df[A] Out[654]: rows 0 0.192451 1 1.857704 2 -0.153931 3 0.722570 4 0.778033 Name: A
112
113
In [662]: tuples = zip(*arrays) In [663]: tuples Out[663]: [(bar, one), (bar, two), (baz, one), (baz, two), (foo, one), (foo, two), (qux, one), (qux, two)] In [664]: index = MultiIndex.from_tuples(tuples, names=[first, second]) In [665]: s = Series(randn(8), index=index) In [666]: s Out[666]: first second bar one two baz one two foo one two qux one two
As a convenience, you can pass a list of arrays directly into Series or DataFrame to construct a MultiIndex automatically:
In [667]: arrays = [np.array([bar, bar, baz, baz, foo, foo, qux, qux]), .....: np.array([one, two, one, two, one, two, one, two])] .....: In [668]: s = Series(randn(8), index=arrays) In [669]: s Out[669]: bar one -0.880984 two 0.997289 baz one -1.693316 two -0.179129 foo one -1.598062 two 0.936914 qux one 0.912560 two -1.003401 In [670]: df = DataFrame(randn(8, 4), index=arrays) In [671]: df Out[671]: 0 1 2 3 bar one 1.632781 -0.724626 0.178219 0.310610 two -0.108002 -0.974226 -1.147708 -2.281374 baz one 0.760010 -0.742532 1.533318 2.495362 two -0.432771 -0.068954 0.043520 0.112246
114
foo one 0.871721 two 0.187483 qux one 2.141616 two -0.479010
-0.816064 -0.784880 1.030659 -1.933946 0.377312 0.734122 -0.011225 0.048869 -1.360687 -0.859661 -0.231595 -0.527750
All of the MultiIndex constructors accept a names argument which stores string names for the levels themselves. If no names are provided, some arbitrary ones will be assigned:
In [672]: index.names Out[672]: [first, second]
This index can back any axis of a pandas object, and the number of levels of the index is up to you:
In [673]: df = DataFrame(randn(3, 8), index=[A, B, C], columns=index) In [674]: df Out[674]: <class pandas.core.frame.DataFrame> Index: 3 entries, A to C Data columns: (bar, one)3 non-null values (bar, two)3 non-null values (baz, one)3 non-null values (baz, two)3 non-null values (foo, one)3 non-null values (foo, two)3 non-null values (qux, one)3 non-null values (qux, two)3 non-null values dtypes: float64(8) In [675]: DataFrame(randn(6, 6), Out[675]: first bar second one two first second bar one -1.993606 -1.927385 two 0.455264 -0.030740 baz one 0.323586 -0.641630 two 0.318587 2.089075 foo one -2.029766 0.792652 two 0.095031 -0.270099 index=index[:6], columns=index[:6]) baz one foo one
two
two
-2.027924 1.624972 0.551135 3.059267 0.935716 1.061192 -2.107852 0.199905 -0.587514 0.053897 0.194889 -0.381994 -0.728293 -0.090255 -0.748199 1.318931 0.461007 -0.542749 -0.305384 -0.479195 -0.707140 -0.773882 0.229453 0.304418
Weve sparsied the higher levels of the indexes to make the console output a bit easier on the eyes. Its worth keeping in mind that theres nothing preventing you from using tuples as atomic labels on an axis:
In [676]: Series(randn(8), index=tuples) Out[676]: (bar, one) 0.736135 (bar, two) -0.859631 (baz, one) -0.424100 (baz, two) -0.776114 (foo, one) 1.279293 (foo, two) 0.943798 (qux, one) -1.001859 (qux, two) 0.306546
The reason that the MultiIndex matters is that it can allow you to do grouping, selection, and reshaping operations as we will describe below and in subsequent areas of the documentation. As you will see in later sections, you can nd
115
yourself working with hierarchically-indexed data without creating a MultiIndex explicitly yourself. However, when loading data from a le, you may wish to generate your own MultiIndex when preparing the data set. Note that how the index is displayed by be controlled using the multi_sparse option in pandas.set_printoptions:
In [677]: pd.set_printoptions(multi_sparse=False) In [678]: df Out[678]: <class pandas.core.frame.DataFrame> Index: 3 entries, A to C Data columns: (bar, one)3 non-null values (bar, two)3 non-null values (baz, one)3 non-null values (baz, two)3 non-null values (foo, one)3 non-null values (foo, two)3 non-null values (qux, one)3 non-null values (qux, two)3 non-null values dtypes: float64(8) In [679]: pd.set_printoptions(multi_sparse=True)
116
Out[684]: A -1.296337 B 1.469725 C -0.938794 Name: one In [685]: s[qux] Out[685]: one 0.912560 two -1.003401
reindex can be called with another MultiIndex or even a list or array of tuples:
In [688]: s.reindex(index[:3]) Out[688]: first second bar one -0.880984 two 0.997289 baz one -1.693316 In [689]: s.reindex([(foo, two), (bar, one), (qux, one), (baz, one)]) Out[689]: foo two 0.936914 bar one -0.880984 qux one 0.912560 baz one -1.693316
117
118
foo
one two
The following does not work, and its not clear if it should or not:
>>> df.ix[[bar, qux]]
The code for implementing .ix makes every attempt to do the right thing but as you use it you may uncover corner cases or unintuitive behavior. If you do nd something like this, do not hesitate to report the issue or ask on the mailing list.
119
1.047010 0.242737
In [704]: print df2.reindex(df.index, level=0) 0 1 one y -0.598972 0.242737 x -0.598972 0.242737 zero y -0.345075 1.047010 x -0.345075 1.047010 In [705]: df_aligned, df2_aligned = df.align(df2, level=0) In [706]: print df_aligned 0 1 one y 0.307453 -0.906534 x -1.505397 1.392009 zero y -0.027793 -0.631023 x -0.662357 2.725042 In [707]: print df2_aligned 0 1 one y -0.598972 0.242737 x -0.598972 0.242737 zero y -0.345075 1.047010 x -0.345075 1.047010
120
foo qux
In [712]: s.sortlevel(1) Out[712]: bar one -1.847240 baz one 0.819712 foo one -2.107728 qux one -0.529247 bar two -1.590742 baz two 0.614656 foo two -0.156479 qux two -1.696377
Note, you may also pass a level name to sortlevel if the MultiIndex levels are named.
In [713]: s.index.names = [L1, L2] In [714]: s.sortlevel(level=L1) Out[714]: L1 L2 bar one -1.847240 two -1.590742 baz one 0.819712 two 0.614656 foo one -2.107728 two -0.156479 qux one -0.529247 two -1.696377 In [715]: s.sortlevel(level=L2) Out[715]: L1 L2 bar one -1.847240 baz one 0.819712 foo one -2.107728 qux one -0.529247 bar two -1.590742 baz two 0.614656 foo two -0.156479 qux two -1.696377
Some indexing will work even if the data are not sorted, but will be rather inefcient and will also return a copy of the data rather than a view:
In [716]: s[qux] Out[716]: L2 one -0.529247 two -1.696377 In [717]: s.sortlevel(1)[qux] Out[717]: L2 one -0.529247 two -1.696377
121
On higher dimensional objects, you can sort any of the other axes by level if they have a MultiIndex:
In [718]: df.T.sortlevel(1, axis=1) Out[718]: zero one zero one x x y y 0 -0.662357 -1.505397 -0.027793 0.307453 1 2.725042 1.392009 -0.631023 -0.906534
The MultiIndex object has code to explicity check the sort depth. Thus, if you try to index at a depth at which the index is not sorted, it will raise an exception. Here is a concrete example to illustrate this:
In [719]: tuples = [(a, a), (a, b), (b, a), (b, b)] In [720]: idx = MultiIndex.from_tuples(tuples) In [721]: idx.lexsort_depth Out[721]: 2 In [722]: reordered = idx[[1, 0, 3, 2]] In [723]: reordered.lexsort_depth Out[723]: 1 In [724]: s = Series(randn(4), index=reordered) In [725]: s.ix[a:a] Out[725]: a b -0.488326 a 0.851918
However:
>>> s.ix[(a, b):(b, a)] Exception: MultiIndex lexsort depth 1, key was length 2
122
You can probably guess that the labels determine which unique element is identied with that location at each layer of the index. Its important to note that sortedness is determined solely from the integer labels and does not check (or care) whether the levels themselves are sorted. Fortunately, the constructors from_tuples and from_arrays ensure that this is true, but if you compute the levels and labels yourself, please be careful.
d 1 2 3 4
123
In [734]: indexed1 = data.set_index(c) In [735]: indexed1 Out[735]: a b d c z bar one 1 y bar two 2 x foo one 3 w foo two 4 In [736]: indexed2 = data.set_index([a, b]) In [737]: indexed2 Out[737]: c d a b bar one z 1 two y 2 foo one x 3 two w 4
The append keyword option allow you to keep the existing index and append the given columns to a MultiIndex:
In [738]: frame = data.set_index(c, drop=False) In [739]: frame = frame.set_index([a, b], append=True) In [740]: frame Out[740]: c d c a b z bar one z 1 y bar two y 2 x foo one x 3 w foo two w 4
Other options in set_index allow you not drop the index columns or to add the index in-place (without creating a new object):
In [741]: data.set_index(c, drop=False) Out[741]: a b c d c z bar one z 1 y bar two y 2 x foo one x 3 w foo two w 4 In [742]: df = data.set_index([a, b], inplace=True) In [743]: data Out[743]: c d a b bar one z 1 two y 2 foo one x 3 two w 4
124
The output is more similar to a SQL table or a record array. The names for the columns derived from the index are the ones stored in the names attribute. You can use the level keyword to remove only a portion of the index:
In [746]: frame Out[746]: c d c a b z bar one z 1 y bar two y 2 x foo one x 3 w foo two w 4 In [747]: frame.reset_index(level=1) Out[747]: a c d c b z one bar z 1 y two bar y 2 x one foo x 3 w two foo w 4
reset_index takes an optional parameter drop which if true simply discards the index, instead of putting index values in the DataFrames columns. Note: The reset_index method used to be called delevel which is now deprecated.
125
126
CHAPTER
EIGHT
COMPUTATIONAL TOOLS
8.1 Statistical functions
8.1.1 Percent Change
Both Series and DataFrame has a method pct_change to compute the percent change over a given number of periods (using fill_method to ll NA/null values).
In [191]: ser = Series(randn(8)) In [192]: ser.pct_change() Out[192]: 0 NaN 1 -1.602976 2 4.334938 3 -0.247456 4 -2.067345 5 -1.142903 6 -1.688214 7 -9.759729 In [193]: df = DataFrame(randn(10, 4)) In [194]: df.pct_change(periods=3) Out[194]: 0 1 2 3 0 NaN NaN NaN NaN 1 NaN NaN NaN NaN 2 NaN NaN NaN NaN 3 -0.218320 -1.054001 1.987147 -0.510183 4 -0.439121 -1.816454 0.649715 -4.822809 5 -0.127833 -3.042065 -5.866604 -1.776977 6 -2.596833 -1.959538 -2.111697 -3.798900 7 -0.117826 -2.169058 0.036094 -0.067696 8 2.492606 -1.357320 -1.205802 -1.558697 9 -1.012977 2.324558 -1.003744 -0.371806
8.1.2 Covariance
The Series object has a method cov to compute covariance between series (excluding NA/null values).
127
Analogously, DataFrame has a method cov to compute pairwise covariances among the series in the DataFrame, also excluding NA/null values.
In [198]: frame = DataFrame(randn(1000, 5), columns=[a, b, c, d, e]) In [199]: frame.cov() Out[199]: a b c d e a 1.000882 -0.003177 -0.002698 -0.006889 0.031912 b -0.003177 1.024721 0.000191 0.009212 0.000857 c -0.002698 0.000191 0.950735 -0.031743 -0.005087 d -0.006889 0.009212 -0.031743 1.002983 -0.047952 e 0.031912 0.000857 -0.005087 -0.047952 1.042487
8.1.3 Correlation
Several methods for computing correlations are provided. Several kinds of correlation methods are provided: Method name pearson (default) kendall spearman Description Standard correlation coefcient Kendall Tau correlation coefcient Spearman rank correlation coefcient
Note that non-numeric columns will be automatically excluded from the correlation calculation. A related method corrwith is implemented on DataFrame to compute the correlation between like-labeled Series contained in different DataFrame objects.
128
In [205]: index = [a, b, c, d, e] In [206]: columns = [one, two, three, four] In [207]: df1 = DataFrame(randn(5, 4), index=index, columns=columns) In [208]: df2 = DataFrame(randn(4, 4), index=index[:4], columns=columns) In [209]: df1.corrwith(df2) Out[209]: one 0.803464 two 0.142469 three -0.498774 four 0.806420 In [210]: df2.corrwith(df1, axis=1) Out[210]: a 0.011572 b 0.388066 c -0.335819 d 0.232412 e NaN
rank is also a DataFrame method and can rank either the rows (axis=0) or the columns (axis=1). NaN values are excluded from the ranking.
In [214]: df = DataFrame(np.random.randn(10, 6)) In [215]: df[4] = df[2][:5] # some ties In [216]: df Out[216]: 0 1 0 0.085011 -0.459422 1 -0.557052 0.775425 2 0.815695 -0.295737 3 1.465947 0.021757 4 -0.678378 0.091855 5 -0.190285 0.187520 6 -0.776678 -2.082637 7 -1.295037 0.367656
2 3 4 5 -1.660917 -1.913019 -1.660917 0.833479 0.003794 0.555351 0.003794 -1.169977 -0.534290 0.068917 -0.534290 -0.513855 0.523224 -0.439297 0.523224 -0.959568 1.337956 0.792551 1.337956 0.711776 -0.355562 1.730964 NaN -1.362312 -0.165877 0.357163 NaN 0.631662 -1.886797 -0.531790 NaN 1.270408
129
8 9
1.106052 0.309979
In [217]: df.rank(1) Out[217]: 0 1 2 3 4 0 5 4 2.5 1 2.5 1 2 6 3.5 5 3.5 2 6 4 1.5 5 1.5 3 6 3 4.5 2 4.5 4 1 2 5.5 4 5.5 5 3 4 2.0 5 NaN 6 2 1 3.0 4 NaN 7 2 4 1.0 3 NaN 8 4 3 2.0 5 NaN 9 2 4 3.0 1 NaN
5 6 1 3 1 3 1 5 5 1 5
rank optionally takes a parameter ascending which by default is true; when false, data is reverse-ranked, with larger values assigned a smaller rank. rank supports different tie-breaking methods, specied with the method parameter: average : average rank of tied group min : lowest rank in the group max : highest rank in the group first : ranks assigned in the order they appear in the array Note: These methods are signicantly faster (around 10-20x) than scipy.stats.rankdata.
130
Generally these methods all have the same interface. The binary operators (e.g. rolling_corr) take two Series or DataFrames. Otherwise, they all accept the following arguments: window: size of moving window min_periods: threshold of non-null data points to require (otherwise result is NA) freq: optionally specify a frequency string or DateOffset to pre-conform the data to. Note that prior to pandas v0.8.0, a keyword argument time_rule was used instead of freq that referred to the legacy time rule constants These functions can be applied to ndarrays or Series objects:
In [218]: ts = Series(randn(1000), index=date_range(1/1/2000, periods=1000)) In [219]: ts = ts.cumsum() In [220]: ts.plot(style=k--) Out[220]: <matplotlib.axes.AxesSubplot at 0x5598150> In [221]: rolling_mean(ts, 60).plot(style=k) Out[221]: <matplotlib.axes.AxesSubplot at 0x5598150>
They can also be applied to DataFrame objects. This is really just syntactic sugar for applying the moving window operator to all of the DataFrames columns:
In [222]: df = DataFrame(randn(1000, 4), index=ts.index, .....: columns=[A, B, C, D]) .....: In [223]: df = df.cumsum() In [224]: rolling_sum(df, 60).plot(subplots=True) Out[224]: array([Axes(0.125,0.747826;0.775x0.152174),
131
The rolling_apply function takes an extra func argument and performs generic rolling computations. The func argument should be a single function that produces a single value from an ndarray input. Suppose we wanted to compute the mean absolute deviation on a rolling basis:
In [225]: mad = lambda x: np.fabs(x - x.mean()).mean() In [226]: rolling_apply(ts, 60, mad).plot(style=k) Out[226]: <matplotlib.axes.AxesSubplot at 0x5ad8e50>
132
You can efciently retrieve the time series of correlations between two columns using ix indexing:
133
Like the rolling_ functions, the following methods are included in the pandas namespace or can be located in pandas.stats.moments.
134
Function expanding_count expanding_sum expanding_mean expanding_median expanding_min expanding_max expanding_std expanding_var expanding_skew expanding_kurt expanding_quantile expanding_apply expanding_cov expanding_corr expanding_corr_pairwise
Description Number of non-null observations Sum of values Mean of values Arithmetic median of values Minimum Maximum Unbiased standard deviation Unbiased variance Unbiased skewness (3rd moment) Unbiased kurtosis (4th moment) Sample quantile (value at %) Generic apply Unbiased covariance (binary) Correlation (binary) Pairwise correlation of DataFrame columns
Aside from not having a window parameter, these functions have the same interfaces as their rolling_ counterpart. Like above, the parameters they all accept are: min_periods: threshold of non-null data points to require. Defaults to minimum needed to compute statistic. No NaNs will be output once min_periods non-null data points have been seen. freq: optionally specify a frequency string or DateOffset to pre-conform the data to. Note that prior to pandas v0.8.0, a keyword argument time_rule was used instead of freq that referred to the legacy time rule constants Note: The output of the rolling_ and expanding_ functions do not return a NaN if there are at least min_periods non-null values in the current window. This differs from cumsum, cumprod, cummax, and cummin, which return NaN in the output wherever a NaN is encountered in the input. An expanding window statistic will be more stable (and less responsive) than its rolling window counterpart as the increasing window size decreases the relative impact of an individual data point. As an example, here is the expanding_mean output for the previous time series dataset:
In [234]: ts.plot(style=k--) Out[234]: <matplotlib.axes.AxesSubplot at 0x61a4750> In [235]: expanding_mean(ts).plot(style=k) Out[235]: <matplotlib.axes.AxesSubplot at 0x61a4750>
135
You can pass one or the other to these functions but not both. Span corresponds to what is commonly called a 20day EW moving average for example. Center of mass has a more physical interpretation. For example, span = 20 corresponds to com = 9.5. Here is the list of functions available: Function ewma ewmvar ewmstd ewmcorr ewmcov Description EW moving average EW moving variance EW moving standard deviation EW moving correlation EW moving covariance
136
Note: The EW functions perform a standard adjustment to the initial observations whereby if there are fewer observations than called for in the span, those observations are reweighted accordingly.
137
Generally speaking, the ols works by being given a y (response) object and an x (predictors) object. These can take many forms: y: a Series, ndarray, or DataFrame (panel model) x: Series, DataFrame, dict of Series, dict of DataFrame or Panel Based on the types of y and x, the model will be inferred to either a panel model or a regular linear model. If the y variable is a DataFrame, the result will be a panel model. In this case, the x variable must either be a Panel, or a dict of DataFrame (which will be coerced into a Panel).
138
If we had passed a Series instead of a DataFrame with the single GOOG column, the model would have assigned the generic name x to the sole right-hand side variable. We can do a moving window regression to see how the relationship changes over time:
In [249]: model = ols(y=rets[AAPL], x=rets.ix[:, [GOOG]], .....: window=250) .....: # just plot the coefficient for GOOG In [250]: model.beta[GOOG].plot() Out[250]: <matplotlib.axes.AxesSubplot at 0x6dffed0>
It looks like there are some outliers rolling in and out of the window in the above regression, inuencing the results. We could perform a simple winsorization at the 3 STD level to trim the impact of outliers:
In [251]: winz = rets.copy() In [252]: std_1year = rolling_std(rets, 250, min_periods=20) # cap at 3 * 1 year standard deviation In [253]: cap_level = 3 * np.sign(winz) * std_1year In [254]: winz[np.abs(winz) > 3 * std_1year] = cap_level In [255]: winz_model = ols(y=winz[AAPL], x=winz.ix[:, [GOOG]],
139
.....: .....:
window=250)
In [256]: model.beta[GOOG].plot(label="With outliers") Out[256]: <matplotlib.axes.AxesSubplot at 0x6deef50> In [257]: winz_model.beta[GOOG].plot(label="Winsorized"); plt.legend(loc=best) Out[257]: <matplotlib.legend.Legend at 0x7b92410>
So in this simple example we see the impact of winsorization is actually quite signicant. Note the correlation after winsorization remains high:
In [258]: winz.corrwith(rets) Out[258]: AAPL 0.994804 GOOG 0.972083 MSFT 0.998329
Multiple regressions can be run by passing a DataFrame with multiple columns for the predictors x:
In [259]: ols(y=winz[AAPL], x=winz.drop([AAPL], axis=1)) Out[259]: -------------------------Summary of Regression Analysis------------------------Formula: Y ~ <GOOG> + <MSFT> + <intercept> Number of Observations: 726 Number of Degrees of Freedom: 3 R-squared: 0.3812 Adj R-squared: 0.3795 Rmse: 0.0130 F-stat (2, 723): 222.7034, p-value: 0.0000 Degrees of Freedom: model 2, resid 723 -----------------------Summary of Estimated Coefficients-----------------------Variable Coef Std Err t-stat p-value CI 2.5% CI 97.5%
140
-------------------------------------------------------------------------------GOOG 0.4783 0.0369 12.95 0.0000 0.4059 0.5507 MSFT 0.3031 0.0412 7.36 0.0000 0.2224 0.3838 intercept 0.0013 0.0005 2.63 0.0086 0.0003 0.0022 ---------------------------------End of Summary---------------------------------
In a panel model, we can insert dummy (0-1) variables for the entities involved (here, each of the stocks) to account the a entity-specic effect (intercept):
In [263]: fe_model = ols(y=volume, x={return : np.abs(rets)}, .....: entity_effects=True) .....: In [264]: fe_model Out[264]: -------------------------Summary of Regression Analysis------------------------Formula: Y ~ <return> + <FE_GOOG> + <FE_MSFT> + <intercept> Number of Observations: 2178 Number of Degrees of Freedom: 4 R-squared: 0.7385 Adj R-squared: 0.7382 Rmse: 0.1395 F-stat (3, 2174): 2046.7338, p-value: 0.0000 Degrees of Freedom: model 3, resid 2174 -----------------------Summary of Estimated Coefficients-----------------------Variable Coef Std Err t-stat p-value CI 2.5% CI 97.5% --------------------------------------------------------------------------------
141
return 4.5910 0.2656 17.28 0.0000 4.0704 5.1116 FE_GOOG -0.1574 0.0073 -21.49 0.0000 -0.1718 -0.1431 FE_MSFT 0.3925 0.0073 53.45 0.0000 0.3781 0.4069 intercept 0.1341 0.0062 21.70 0.0000 0.1220 0.1462 ---------------------------------End of Summary---------------------------------
Because we ran the regression with an intercept, one of the dummy variables must be dropped or the design matrix will not be full rank. If we do not use an intercept, all of the dummy variables will be included:
In [265]: fe_model = ols(y=volume, x={return : np.abs(rets)}, .....: entity_effects=True, intercept=False) .....: In [266]: fe_model Out[266]: -------------------------Summary of Regression Analysis------------------------Formula: Y ~ <return> + <FE_AAPL> + <FE_GOOG> + <FE_MSFT> Number of Observations: 2178 Number of Degrees of Freedom: 4 R-squared: 0.7385 Adj R-squared: 0.7382 Rmse: 0.1395 F-stat (4, 2174): 2046.7338, p-value: 0.0000 Degrees of Freedom: model 3, resid 2174 -----------------------Summary of Estimated Coefficients-----------------------Variable Coef Std Err t-stat p-value CI 2.5% CI 97.5% -------------------------------------------------------------------------------return 4.5910 0.2656 17.28 0.0000 4.0704 5.1116 FE_AAPL 0.1341 0.0062 21.70 0.0000 0.1220 0.1462 FE_GOOG -0.0233 0.0060 -3.88 0.0001 -0.0352 -0.0115 FE_MSFT 0.5266 0.0059 89.61 0.0000 0.5151 0.5381 ---------------------------------End of Summary---------------------------------
We can also include time effects, which demeans the data cross-sectionally at each point in time (equivalent to including dummy variables for each date). More mathematical care must be taken to properly compute the standard errors in this case:
In [267]: te_model = ols(y=volume, x={return : np.abs(rets)}, .....: time_effects=True, entity_effects=True) .....: In [268]: te_model Out[268]: -------------------------Summary of Regression Analysis------------------------Formula: Y ~ <return> + <FE_GOOG> + <FE_MSFT> Number of Observations: 2178 Number of Degrees of Freedom: 729 R-squared: 0.8153 Adj R-squared: 0.7226 Rmse: 0.1343 F-stat (3, 1449): 8.7885, p-value: 0.0000 Degrees of Freedom: model 728, resid 1449 -----------------------Summary of Estimated Coefficients-----------------------Variable Coef Std Err t-stat p-value CI 2.5% CI 97.5% -------------------------------------------------------------------------------return 3.7810 0.3861 9.79 0.0000 3.0243 4.5377 FE_GOOG -0.1583 0.0071 -22.43 0.0000 -0.1722 -0.1445 FE_MSFT 0.3907 0.0071 55.03 0.0000 0.3768 0.4046 ---------------------------------End of Summary---------------------------------
142
Here the intercept (the mean term) is dropped by default because it will be 0 according to the model assumptions, having subtracted off the group means.
143
144
CHAPTER
NINE
three four -2.400634 bar -1.386071 bar 1.500571 bar -0.374279 bar -0.551865 bar
two
three four
five
145
a 0.059117 1.138469 -2.400634 b NaN NaN NaN c -0.280853 0.025653 -1.386071 d NaN NaN NaN e 0.863937 0.252462 1.500571 f 1.053202 -2.338595 -0.374279 g NaN NaN NaN h -2.359958 -1.157886 -0.551865
Summary: NaN, inf, -inf, and None (in object arrays) are all considered missing by the isnull and notnull functions. 146 Chapter 9. Working with missing data
two three 1.138469 -2.400634 NaN NaN 0.025653 -1.386071 NaN NaN 0.252462 1.500571
In [946]: a + b Out[946]: one three a 0.118234 NaN b NaN NaN c -0.561707 NaN d NaN NaN e 1.727874 NaN
The descriptive statistics and computational methods discussed in the data structure overview (and listed here and here) are all written to account for missing data. For example: When summing data, NA (missing) values will be treated as zero If the data are all NA, the result will be NA Methods like cumsum and cumprod ignore NA values, but preserve them in the resulting arrays
In [947]: df Out[947]: one two a 0.059117 1.138469 b NaN NaN c -0.280853 0.025653 d NaN NaN e 0.863937 0.252462 f 1.053202 -2.338595 g NaN NaN h -2.359958 -1.157886
147
c d e f g h
In [950]: df.cumsum() Out[950]: one two a 0.059117 1.138469 b NaN NaN c -0.221736 1.164122 d NaN NaN e 0.642200 1.416584 f 1.695403 -0.922011 g NaN NaN h -0.664556 -2.079897
three four -2.400634 bar NaN NaN -1.386071 bar NaN NaN 1.500571 bar -0.374279 bar NaN NaN -0.551865 bar
In [952]: df2.fillna(0) Out[952]: one two three four a 0.059117 1.138469 -2.400634 bar b 0.000000 0.000000 0.000000 0 c -0.280853 0.025653 -1.386071 bar d 0.000000 0.000000 0.000000 0 e 0.863937 0.252462 1.500571 bar f 1.053202 -2.338595 -0.374279 bar
148
0 bar
0 False
In [953]: df2[four].fillna(missing) Out[953]: a bar b missing c bar d missing e bar f bar g missing h bar Name: four
Fill gaps forward or backward Using the same lling arguments as reindexing, we can propagate non-null values forward or backward:
In [954]: df Out[954]: one two a 0.059117 1.138469 b NaN NaN c -0.280853 0.025653 d NaN NaN e 0.863937 0.252462 f 1.053202 -2.338595 g NaN NaN h -2.359958 -1.157886
In [955]: df.fillna(method=pad) Out[955]: one two three a 0.059117 1.138469 -2.400634 b 0.059117 1.138469 -2.400634 c -0.280853 0.025653 -1.386071 d -0.280853 0.025653 -1.386071 e 0.863937 0.252462 1.500571 f 1.053202 -2.338595 -0.374279 g 1.053202 -2.338595 -0.374279 h -2.359958 -1.157886 -0.551865
Limit the amount of lling If we only want consecutive gaps lled up to a certain number of data points, we can use the limit keyword:
In [956]: df Out[956]: one two three a 0.059117 1.138469 -2.400634 b NaN NaN NaN c NaN NaN NaN d NaN NaN NaN e 0.863937 0.252462 1.500571 f 1.053202 -2.338595 -0.374279 g NaN NaN NaN h -2.359958 -1.157886 -0.551865 In [957]: df.fillna(method=pad, limit=1)
149
Out[957]: one two three a 0.059117 1.138469 -2.400634 b 0.059117 1.138469 -2.400634 c NaN NaN NaN d NaN NaN NaN e 0.863937 0.252462 1.500571 f 1.053202 -2.338595 -0.374279 g 1.053202 -2.338595 -0.374279 h -2.359958 -1.157886 -0.551865
To remind you, these are the available lling methods: Method pad / fll bll / backll Action Fill values forward Fill values backward
With time series data, using pad/fll is extremely common so that the last known value is available at every time point.
150
dropna is presently only implemented for Series and DataFrame, but will be eventually added to Panel. Series.dropna is a simpler method as it only has one axis to consider. DataFrame.dropna has considerably more options, which can be examined in the API.
9.3.3 Interpolation
A linear interpolate method has been implemented on Series. The default interpolation assumes equally spaced points.
In [962]: ts.count() Out[962]: 61 In [963]: ts.head() Out[963]: 2000-01-31 0.469112 2000-02-29 NaN 2000-03-31 NaN 2000-04-28 NaN 2000-05-31 NaN Freq: BM In [964]: ts.interpolate().count() Out[964]: 100 In [965]: ts.interpolate().head() Out[965]: 2000-01-31 0.469112 2000-02-29 0.435428 2000-03-31 0.401743 2000-04-28 0.368059 2000-05-31 0.334374 Freq: BM In [966]: ts.interpolate().plot() Out[966]: <matplotlib.axes.AxesSubplot at 0xc184a10>
151
152
In [970]: ser Out[970]: 0 0 1 NaN 10 10 In [971]: ser.interpolate() Out[971]: 0 0 1 5 10 10 In [972]: ser.interpolate(method=values) Out[972]: 0 0 1 1 10 10
153
In [977]: df = DataFrame({a: [0, 1, 2, 3, 4], b: [5, 6, 7, 8, 9]}) In [978]: df.replace({a: 0, b: 5}, 100) Out[978]: a b 0 100 100 1 1 6 2 2 7 3 3 8 4 4 9
Instead of replacing with specied values, you can treat all given values as missing and interpolate over them:
In [979]: ser.replace([1, 2, 3], method=pad) Out[979]: 0 0 1 0 2 0 3 0 4 4
154
5 6 7
Ordinarily NumPy will complain if you try to use an object array (even if it contains boolean values) instead of a boolean array to get or set values from an ndarray (e.g. selecting values based on some criteria). If a boolean vector contains NAs, an exception will be generated:
In [986]: reindexed = s.reindex(range(8)).fillna(0) In [987]: reindexed[crit] --------------------------------------------------------------------------ValueError Traceback (most recent call last) <ipython-input-987-2da204ed1ac7> in <module>() ----> 1 reindexed[crit] /home/chang/tmp/doc_pandas/pandas/core/series.py in __getitem__(self, key) 488 # special handling of boolean data with NAs stored in object 489 # arrays. Since we cant represent NA with dtype=bool --> 490 if _is_bool_indexer(key): 491 key = self._check_bool_indexer(key) 492 key = np.asarray(key, dtype=bool) /home/chang/tmp/doc_pandas/pandas/core/common.py in _is_bool_indexer(key) 579 if not lib.is_bool_array(key): 580 if isnull(key).any(): --> 581 raise ValueError(cannot index with vector containing 582 NA / NaN values) 583 return False ValueError: cannot index with vector containing NA / NaN values
However, these can be lled in using llna and it will work ne:
In [988]: reindexed[crit.fillna(False)] Out[988]: 2 1.314232 4 0.690579 6 0.995761 7 2.396780 In [989]: reindexed[crit.fillna(True)] Out[989]: 1 0.000000 2 1.314232 3 0.000000 4 0.690579 5 0.000000 6 0.995761 7 2.396780
155
156
CHAPTER
TEN
We aim to make operations like this natural and easy to express using pandas. Well address each area of GroupBy functionality then provide some non-trivial examples / use cases.
157
The mapping can be specied many different ways: A Python function, to be called on each of the axis labels A list or NumPy array of the same length as the selected axis A dict or Series, providing a label -> group name mapping For DataFrame objects, a string indicating a column to be used to group. Of course df.groupby(A) is just syntactic sugar for df.groupby(df[A]), but it makes life simpler A list of any of the above things Collectively we refer to the grouping objects as the keys. For example, consider the following DataFrame:
In [454]: df = DataFrame({A : [foo, bar, foo, bar, .....: foo, bar, foo, foo], .....: B : [one, one, two, three, .....: two, two, one, three], .....: C : randn(8), D : randn(8)}) .....: In [455]: df Out[455]: A B 0 foo one 1 bar one 2 foo two 3 bar three 4 foo two 5 bar two 6 foo one 7 foo three
These will split the DataFrame on its index (rows). We could also split by the columns:
In [458]: def get_letter_type(letter): .....: if letter.lower() in aeiou: .....: return vowel .....: else: .....: return consonant .....: In [459]: grouped = df.groupby(get_letter_type, axis=1)
Starting with 0.8, pandas Index objects now supports duplicate values. If a non-unique index is used as the group key in a groupby operation, all values for the same index value will be considered to be in one group and thus the output of aggregation functions will only contain unique index values:
In [460]: lst = [1, 2, 3, 1, 2, 3] In [461]: s = Series([1, 2, 3, 10, 20, 30], lst) In [462]: grouped = s.groupby(level=0) In [463]: grouped.first()
158
Note that no splitting occurs until its needed. Creating the GroupBy object only veries that youve passed a valid mapping. Note: Many kinds of complicated data manipulations can be expressed in terms of GroupBy operations (though cant be guaranteed to be the most efcient). You can get quite creative with the label mapping functions.
Calling the standard Python len function on the GroupBy object just returns the length of the groups dict, so it is largely just a convenience:
In [468]: grouped = df.groupby([A, B]) In [469]: grouped.groups Out[469]: {(bar, one): [1], (bar, three): [3], (bar, two): [5], (foo, one): [0, 6], (foo, three): [7], (foo, two): [2, 4]} In [470]: len(grouped) Out[470]: 6
By default the group keys are sorted during the groupby operation. You may however pass sort=False for potential speedups:
In [471]: df2 = DataFrame({X : [B, B, A, A], Y : [1, 2, 3, 4]})
159
In [475]: grouped = s.groupby(level=0) In [476]: grouped.sum() Out[476]: first bar 0.142048 baz -0.811169 foo -0.560041 qux -0.953439
If the MultiIndex has names specied, these can be passed instead of the level number:
In [477]: s.groupby(level=second).sum() Out[477]: second one -2.300857 two 0.118256
The aggregation functions such as sum will take the level parameter directly. Additionally, the resulting index will be named according to the chosen level:
In [478]: s.sum(level=second) Out[478]: second one -2.300857 two 0.118256
160
In [479]: s Out[479]: first second bar doo baz foo qux bee bop bop
In [480]: s.groupby(level=[first,second]).sum() Out[480]: first second bar doo 0.981751 baz bee -2.754270 foo bop -1.528539 qux bop -0.499427
This is mainly syntactic sugar for the alternative and much more verbose:
In [484]: df[C].groupby(df[A]) Out[484]: <pandas.core.groupby.SeriesGroupBy at 0xb6f9490>
Additionally this method avoids recomputing the internal grouping information derived from the passed key.
161
two -0.173215 -0.706771 B C D one 0.469112 -0.861849 two -1.509059 -0.494929 two 1.212112 0.721555 one 0.119209 -1.039575 three -1.044236 0.271860
In the case of grouping by multiple keys, the group name will be a tuple:
In [487]: for name, group in df.groupby([A, B]): .....: print name .....: print group .....: (bar, one) A B C D 1 bar one -0.282863 -2.104569 (bar, three) A B C D 3 bar three -1.135632 1.071804 (bar, two) A B C D 5 bar two -0.173215 -0.706771 (foo, one) A B C D 0 foo one 0.469112 -0.861849 6 foo one 0.119209 -1.039575 (foo, three) A B C D 7 foo three -1.044236 0.27186 (foo, two) A B C D 2 foo two -1.509059 -0.494929 4 foo two 1.212112 0.721555
Its standard Python-fu but remember you can unpack the tuple in the for loop statement if you wish: for (k1, k2), group in grouped:.
10.3 Aggregation
Once the GroupBy object has been created, several methods are available to perform a computation on the grouped data. An obvious one is aggregation via the aggregate or equivalently agg method:
In [488]: grouped = df.groupby(A) In [489]: grouped.aggregate(np.sum) Out[489]: C D A bar -1.591710 -1.739537 foo -0.752861 -1.402938 In [490]: grouped = df.groupby([A, B]) In [491]: grouped.aggregate(np.sum) Out[491]:
162
-0.282863 -2.104569 -1.135632 1.071804 -0.173215 -0.706771 0.588321 -1.901424 -1.044236 0.271860 -0.296946 0.226626
As you can see, the result of the aggregation will have the group names as the new index along the grouped axis. In the case of multiple keys, the result is a MultiIndex by default, though this can be changed by using the as_index option:
In [492]: grouped = df.groupby([A, B], as_index=False) In [493]: grouped.aggregate(np.sum) Out[493]: A B C D 0 bar one -0.282863 -2.104569 1 bar three -1.135632 1.071804 2 bar two -0.173215 -0.706771 3 foo one 0.588321 -1.901424 4 foo three -1.044236 0.271860 5 foo two -0.296946 0.226626 In [494]: df.groupby(A, as_index=False).sum() Out[494]: A C D 0 bar -1.591710 -1.739537 1 foo -0.752861 -1.402938
Note that you could use the reset_index DataFrame function to achieve the same result as the column names are stored in the resulting MultiIndex:
In [495]: df.groupby([A, B]).sum().reset_index() Out[495]: A B C D 0 bar one -0.282863 -2.104569 1 bar three -1.135632 1.071804 2 bar two -0.173215 -0.706771 3 foo one 0.588321 -1.901424 4 foo three -1.044236 0.271860 5 foo two -0.296946 0.226626
Another simple aggregation example is to compute the size of each group. This is included in GroupBy as the size method. It returns a Series whose index are the group names and whose values are the sizes of each group.
In [496]: grouped.size() Out[496]: A B bar one 1 three 1 two 1 foo one 2 three 1 two 2
10.3. Aggregation
163
If a dict is passed, the keys will be used to name the columns. Otherwise the functions name (stored in the function object) will be used.
In [499]: grouped[D].agg({result1 : np.sum, .....: result2 : np.mean}) .....: Out[499]: result2 result1 A bar -0.579846 -1.739537 foo -0.280588 -1.402938
On a grouped DataFrame, you can pass a list of functions to apply to each column, which produces an aggregated result with a hierarchical index:
In [500]: grouped.agg([np.sum, np.mean, np.std]) Out[500]: C D sum mean std sum mean A bar -1.591710 -0.530570 0.526860 -1.739537 -0.579846 foo -0.752861 -0.150572 1.113308 -1.402938 -0.280588
Passing a dict of functions has different behavior by default, see the next section.
The function names can also be strings. In order for a string to be valid it must be either implemented on GroupBy or available via dispatching:
In [502]: grouped.agg({C : sum, D : std}) Out[502]: C D A
164
1.591986 0.753219
Of course sum and mean are implemented on pandas objects, so the above code would work even without the special versions via dispatching (see below).
10.4 Transformation
The transform method returns an object that is indexed the same (same size) as the one being grouped. Thus, the passed transform function should return a result that is the same size as the group chunk. For example, suppose we wished to standardize the data within each group:
In [505]: index = date_range(10/1/1999, periods=1100) In [506]: ts = Series(np.random.normal(0.5, 2, 1100), index) In [507]: ts = rolling_mean(ts, 100, 100).dropna() In [508]: ts.head() Out[508]: 2000-01-08 0.536925 2000-01-09 0.494448 2000-01-10 0.496114 2000-01-11 0.443475 2000-01-12 0.474744 Freq: D In [509]: ts.tail() Out[509]: 2002-09-30 0.978859 2002-10-01 0.994704 2002-10-02 0.953789 2002-10-03 0.932345
10.4. Transformation
165
2002-10-04 Freq: D
0.915581
In [510]: key = lambda x: x.year In [511]: zscore = lambda x: (x - x.mean()) / x.std() In [512]: transformed = ts.groupby(key).transform(zscore)
We would expect the result to now have mean 0 and standard deviation 1 within each group, which we can easily check:
# Original Data In [513]: grouped = ts.groupby(key) In [514]: grouped.mean() Out[514]: 2000 0.416344 2001 0.416987 2002 0.599380 In [515]: grouped.std() Out[515]: 2000 0.174755 2001 0.309640 2002 0.266172 # Transformed Data In [516]: grouped_trans = transformed.groupby(key) In [517]: grouped_trans.mean() Out[517]: 2000 -5.108881e-16 2001 -3.808217e-16 2002 -8.577174e-17 In [518]: grouped_trans.std() Out[518]: 2000 1 2001 1 2002 1
We can also visually compare the original and transformed data sets.
In [519]: compare = DataFrame({Original: ts, Transformed: transformed}) In [520]: compare.plot() Out[520]: <matplotlib.axes.AxesSubplot at 0x8658550>
166
Another common data transform is to replace missing data with the group mean.
In [521]: data_df Out[521]: <class pandas.core.frame.DataFrame> Int64Index: 1000 entries, 0 to 999 Data columns: A 908 non-null values B 953 non-null values C 820 non-null values dtypes: float64(3) In [522]: countries = np.array([US, UK, GR, JP]) In [523]: key = countries[np.random.randint(0, 4, 1000)] In [524]: grouped = data_df.groupby(key) # Non-NA count in each group In [525]: grouped.count() Out[525]: A B C GR 219 223 194 JP 238 250 211 UK 228 239 213 US 223 241 202 In [526]: f = lambda x: x.fillna(x.mean()) In [527]: transformed = grouped.transform(f)
We can verify that the group means have not changed in the transformed data and that the transformed data contains no NAs.
In [528]: grouped_trans = transformed.groupby(key) In [529]: grouped.mean() # original group means
10.4. Transformation
167
Out[529]: A B C GR 0.093655 -0.004978 -0.049883 JP -0.067605 0.025828 0.006752 UK -0.054246 0.031742 0.068974 US 0.084334 -0.013433 0.056589 In [530]: grouped_trans.mean() # transformation did not change group means Out[530]: A B C GR 0.093655 -0.004978 -0.049883 JP -0.067605 0.025828 0.006752 UK -0.054246 0.031742 0.068974 US 0.084334 -0.013433 0.056589 In [531]: grouped.count() # original has some missing data points Out[531]: A B C GR 219 223 194 JP 238 250 211 UK 228 239 213 US 223 241 202 In [532]: grouped_trans.count() # counts after transformation Out[532]: A B C GR 234 234 234 JP 264 264 264 UK 251 251 251 US 251 251 251 In [533]: grouped_trans.size() # Verify non-NA count equals group size Out[533]: GR 234 JP 264 UK 251 US 251
But, its rather verbose and can be untidy if you need to pass additional arguments. Using a bit of metaprogramming cleverness, GroupBy now has the ability to dispatch method calls to the groups:
168
What is actually happening here is that a function wrapper is being generated. When invoked, it takes any passed arguments and invokes the function with any arguments on each group (in the above example, the std function). The results are then combined together much in the style of agg and transform (it actually uses apply to infer the gluing, documented next). This enables some operations to be carried out rather succinctly:
In [537]: tsdf = DataFrame(randn(1000, 3), .....: index=date_range(1/1/2000, periods=1000), .....: columns=[A, B, C]) .....: In [538]: tsdf.ix[::2] = np.nan In [539]: grouped = tsdf.groupby(lambda x: x.year) In [540]: grouped.fillna(method=pad) Out[540]: <class pandas.core.frame.DataFrame> DatetimeIndex: 1000 entries, 2000-01-01 00:00:00 to 2002-09-26 00:00:00 Freq: D Data columns: A 998 non-null values B 998 non-null values C 998 non-null values dtypes: float64(3)
In this example, we chopped the collection of time series into yearly chunks then independently called llna on the groups.
169
# could also just call .describe() In [543]: grouped[C].apply(lambda x: x.describe()) Out[543]: A bar count 3.000000 mean -0.530570 std 0.526860 min -1.135632 25% -0.709248 50% -0.282863 75% -0.228039 max -0.173215 foo count 5.000000 mean -0.150572 std 1.113308 min -1.509059 25% -1.044236 50% 0.119209 75% 0.469112 max 1.212112
170
5 6 7
Supposed we wished to compute the standard deviation grouped by the A column. There is a slight problem, namely that we dont care about the data in column B. We refer to this as a nuisance column. If the passed aggregation function cant be applied to some columns, the troublesome columns will be (silently) dropped. Thus, this does not pose any problems:
In [548]: df.groupby(A).std() Out[548]: C D A bar 0.526860 1.591986 foo 1.113308 0.753219
171
172
CHAPTER
ELEVEN
2 -1.509059 0.119209 -0.494929 -1.039575 0.276232 -1.478427 -1.715002 -1.344312 1.643563 -1.776904
3 -1.135632 -1.044236 1.071804 0.271860 -1.087401 0.524988 -1.039268 0.844885 -1.469388 -0.968914
# break it into pieces In [833]: pieces = [df[:3], df[3:7], df[7:]] In [834]: concatenated = concat(pieces) In [835]: concatenated Out[835]: 0 1 2 0 0.469112 -0.282863 -1.509059 1 1.212112 -0.173215 0.119209 2 -0.861849 -2.104569 -0.494929 3 0.721555 -0.706771 -1.039575 4 -0.424972 0.567020 0.276232 5 -0.673690 0.113648 -1.478427 6 0.404705 0.577046 -1.715002
173
7 -0.370647 -1.157892 -1.344312 0.844885 8 1.075770 -0.109050 1.643563 -1.469388 9 0.357021 -0.674600 -1.776904 -0.968914
Like its sibling function on ndarrays, numpy.concatenate, pandas.concat takes a list or dict of homogeneously-typed objects and concatenates them with some congurable handling of what to do with the other axes:
concat(objs, axis=0, join=outer, join_axes=None, ignore_index=False, keys=None, levels=None, names=None, verify_integrity=False)
objs: list or dict of Series, DataFrame, or Panel objects. If a dict is passed, the sorted keys will be used as the keys argument, unless it is passed, in which case the values will be selected (see below) axis: {0, 1, ...}, default 0. The axis to concatenate along join: {inner, outer}, default outer. How to handle indexes on other axis(es). Outer for union and inner for intersection join_axes: list of Index objects. Specic indexes to use for the other n - 1 axes instead of performing inner/outer set logic keys: sequence, default None. Construct hierarchical index using the passed keys as the outermost level If multiple levels passed, should contain tuples. levels : list of sequences, default None. If keys passed, specic levels to use for the resulting MultiIndex. Otherwise they will be inferred from the keys names: list, default None. Names for the levels in the resulting hierarchical index verify_integrity: boolean, default False. Check whether the new concatenated axis contains duplicates. This can be very expensive relative to the actual data concatenation ignore_index : boolean, default False. If True, do not use the index values on the concatenation axis. The resulting axis will be labeled 0, ..., n - 1. This is useful if you are concatenating objects where the concatenation axis does not have meaningful indexing information. Without a little bit of context and example many of these arguments dont make much sense. Lets take the above example. Suppose we wanted to associate specic keys with each of the pieces of the chopped up DataFrame. We can do this using the keys argument:
In [836]: concatenated = concat(pieces, keys=[first, second, third]) In [837]: concatenated Out[837]: 0 1 first 0 0.469112 -0.282863 1 1.212112 -0.173215 2 -0.861849 -2.104569 second 3 0.721555 -0.706771 4 -0.424972 0.567020 5 -0.673690 0.113648 6 0.404705 0.577046 third 7 -0.370647 -1.157892 8 1.075770 -0.109050 9 0.357021 -0.674600
2 -1.509059 0.119209 -0.494929 -1.039575 0.276232 -1.478427 -1.715002 -1.344312 1.643563 -1.776904
3 -1.135632 -1.044236 1.071804 0.271860 -1.087401 0.524988 -1.039268 0.844885 -1.469388 -0.968914
As you can see (if youve read the rest of the documentation), the resulting objects index has a hierarchical index. This means that we can now do stuff like select out each chunk by key:
174
In [838]: concatenated.ix[second] Out[838]: 0 1 2 3 3 0.721555 -0.706771 -1.039575 0.271860 4 -0.424972 0.567020 0.276232 -1.087401 5 -0.673690 0.113648 -1.478427 0.524988 6 0.404705 0.577046 -1.715002 -1.039268
Its not a stretch to see how this can be very useful. More detail on this functionality below.
In [842]: concat([df.ix[:7, [a, b]], df.ix[2:-2, [c]], .....: df.ix[-7:, [d]]], axis=1) .....: Out[842]: a b c d 3xT4w 0.410835 0.813850 0.132003 -0.827317 ElqrR 1.431256 1.340309 -1.170299 -0.226169 FCrRo -0.013960 -0.362543 NaN NaN OxQ8d NaN NaN 0.974466 -2.006747 TVX4p NaN NaN NaN -0.727707 ZNR5B 0.895717 0.805244 -1.206412 NaN bJNQO -0.076467 -1.187678 1.130127 -1.436737 kVfbD NaN NaN NaN -1.219217
175
0.413738 1.607920
NaN 1.024180
NaN 0.569605
Note that the row indexes have been unioned and sorted. Here is the same thing with join=inner:
In [843]: concat([df.ix[:7, [a, b]], df.ix[2:-2, [c]], .....: df.ix[-7:, [d]]], axis=1, join=inner) .....: Out[843]: a b c d ElqrR 1.431256 1.340309 -1.170299 -0.226169 3xT4w 0.410835 0.813850 0.132003 -0.827317 bJNQO -0.076467 -1.187678 1.130127 -1.436737 yCtf4 -1.413681 1.607920 1.024180 0.569605
Lastly, suppose we just wanted to reuse the exact index from the original DataFrame:
In [844]: concat([df.ix[:7, [a, b]], df.ix[2:-2, [c]], .....: df.ix[-7:, [d]]], axis=1, join_axes=[df.index]) .....: Out[844]: a b c d xA2g1 -1.294524 0.413738 NaN NaN FCrRo -0.013960 -0.362543 NaN NaN ZNR5B 0.895717 0.805244 -1.206412 NaN ElqrR 1.431256 1.340309 -1.170299 -0.226169 3xT4w 0.410835 0.813850 0.132003 -0.827317 bJNQO -0.076467 -1.187678 1.130127 -1.436737 yCtf4 -1.413681 1.607920 1.024180 0.569605 OxQ8d NaN NaN 0.974466 -2.006747 kVfbD NaN NaN NaN -1.219217 TVX4p NaN NaN NaN -0.727707
In the case of DataFrame, the indexes must be disjoint but the columns do not need to be:
176
In [849]: df = DataFrame(randn(6, 4), index=date_range(1/1/2000, periods=6), .....: columns=[A, B, C, D]) .....: In [850]: df1 = df.ix[:3] In [851]: df2 = df.ix[3:, :3] In [852]: df1 Out[852]: A B C 2000-01-01 0.176444 0.403310 -0.154951 2000-01-02 -2.179861 -1.369849 -0.954208 2000-01-03 -1.743161 -0.826591 -0.345352 In [853]: df2 Out[853]: A B C 2000-01-04 0.690579 0.995761 2.396780 2000-01-05 3.357427 -0.317441 -1.236269 2000-01-06 -0.487602 -0.082240 -2.182937 In [854]: df1.append(df2) Out[854]: A B 2000-01-01 0.176444 0.403310 2000-01-02 -2.179861 -1.369849 2000-01-03 -1.743161 -0.826591 2000-01-04 0.690579 0.995761 2000-01-05 3.357427 -0.317441 2000-01-06 -0.487602 -0.082240 D 0.301624 1.462696 1.314232
Note: Unlike list.append method, which appends to the original list and returns nothing, append here does not modify df1 and returns its copy with df2 appended.
177
In [862]: df2 Out[862]: A B C D 0 0.082423 -0.055758 0.536580 -0.489682 1 0.369374 -0.034571 -2.484478 -0.281461 2 0.030711 0.109121 1.126203 -0.977349
178
2 3 -1.282782 0.781836 2.353925 0.583787 0.758527 1.729689 -1.340896 1.846883 -1.717693 0.888782 1.171216 0.520260 -0.303421 -0.858447 0.384316 1.574159 0.473424 -0.242861 0.650776 -1.461665
# break it into pieces In [867]: pieces = [df.ix[:, [0, 1]], df.ix[:, [2]], df.ix[:, [3]]] In [868]: result = concat(pieces, axis=1, keys=[one, two, three]) In [869]: result Out[869]: one 0 1 0 1.474071 -0.064034 1 -1.071357 0.441153 2 0.221471 -0.744471 3 -0.964980 -0.845696 4 -1.328865 1.682706 5 0.228440 0.901805 6 -1.197071 -1.066969 7 0.306996 -0.028665 8 1.588931 0.476720 9 -0.014805 -0.284319
two three 2 3 -1.282782 0.781836 2.353925 0.583787 0.758527 1.729689 -1.340896 1.846883 -1.717693 0.888782 1.171216 0.520260 -0.303421 -0.858447 0.384316 1.574159 0.473424 -0.242861 0.650776 -1.461665
You can also pass a dict to concat in which case the dict keys will be used for the keys argument (unless other keys are specied):
In [870]: pieces = {one: df.ix[:, [0, 1]], .....: two: df.ix[:, [2]], .....: three: df.ix[:, [3]]} .....: In [871]: concat(pieces, axis=1) Out[871]: one three two 0 1 3 2 0 1.474071 -0.064034 0.781836 -1.282782 1 -1.071357 0.441153 0.583787 2.353925 2 0.221471 -0.744471 1.729689 0.758527 3 -0.964980 -0.845696 1.846883 -1.340896 4 -1.328865 1.682706 0.888782 -1.717693 5 0.228440 0.901805 0.520260 1.171216 6 -1.197071 -1.066969 -0.858447 -0.303421
179
In [872]: concat(pieces, keys=[three, two]) Out[872]: 2 3 three 0 NaN 0.781836 1 NaN 0.583787 2 NaN 1.729689 3 NaN 1.846883 4 NaN 0.888782 5 NaN 0.520260 6 NaN -0.858447 7 NaN 1.574159 8 NaN -0.242861 9 NaN -1.461665 two 0 -1.282782 NaN 1 2.353925 NaN 2 0.758527 NaN 3 -1.340896 NaN 4 -1.717693 NaN 5 1.171216 NaN 6 -0.303421 NaN 7 0.384316 NaN 8 0.473424 NaN 9 0.650776 NaN
The MultiIndex created has levels that are constructed from the passed keys and the columns of the DataFrame pieces:
In [873]: result.columns.levels Out[873]: [Index([one, two, three], dtype=object), Index([0, 1, 2, 3], dtype=int64)]
If you wish to specify other levels (as will occasionally be the case), you can do so using the levels argument:
In [874]: result = concat(pieces, axis=1, keys=[one, two, three], .....: levels=[[three, two, one, zero]], .....: names=[group_key]) .....: In [875]: result Out[875]: group_key one 0 0 1.474071 1 -1.071357 2 0.221471 3 -0.964980 4 -1.328865 5 0.228440 6 -1.197071 7 0.306996 8 1.588931 9 -0.014805
1 -0.064034 0.441153 -0.744471 -0.845696 1.682706 0.901805 -1.066969 -0.028665 0.476720 -0.284319
two three 2 3 -1.282782 0.781836 2.353925 0.583787 0.758527 1.729689 -1.340896 1.846883 -1.717693 0.888782 1.171216 0.520260 -0.303421 -0.858447 0.384316 1.574159 0.473424 -0.242861 0.650776 -1.461665
In [876]: result.columns.levels Out[876]: [Index([three, two, one, zero], dtype=object), Index([0, 1, 2, 3], dtype=int64)]
180
Yes, this is fairly esoteric, but is actually necessary for implementing things like GroupBy where the order of a categorical variable is meaningful.
C D -0.693921 1.613616 -0.496922 0.306389 -1.561819 -0.260838 -0.902937 0.068159 -1.144073 0.861209 -1.069094 -1.099248 0.661084 0.379319 -1.056652 0.533946
You should use ignore_index with this method to instruct DataFrame to discard its index. If you wish to preserve the index, you should construct an appropriately-indexed DataFrame and append or concatenate those objects. You can also pass a list of dicts or Series:
In [881]: df = DataFrame(np.random.randn(5, 4), .....: columns=[foo, bar, baz, qux]) .....: In [882]: dicts = [{foo: 1, bar: 2, baz: 3, peekaboo: 4}, .....: {foo: 5, bar: 6, baz: 7, peekaboo: 8}] .....: In [883]: result = df.append(dicts, ignore_index=True) In [884]: result Out[884]: bar baz foo 0 0.040403 -0.507516 -1.226970 1 -1.934370 -1.652499 0.394500 2 0.576897 1.146000 -0.896484
181
Heres a description of what each argument is for: left: A DataFrame object right: Another DataFrame object on: Columns (names) to join on. Must be found in both the left and right DataFrame objects. If not passed and left_index and right_index are False, the intersection of the columns in the DataFrames will be inferred to be the join keys left_on: Columns from the left DataFrame to use as keys. Can either be column names or arrays with length equal to the length of the DataFrame right_on: Columns from the right DataFrame to use as keys. Can either be column names or arrays with length equal to the length of the DataFrame left_index: If True, use the index (row labels) from the left DataFrame as its join key(s). In the case of a DataFrame with a MultiIndex (hierarchical), the number of levels must match the number of join keys from the right DataFrame right_index: Same usage as left_index for the right DataFrame how: One of left, right, outer, inner. Defaults to inner. See below for more detailed description of each method sort: Sort the result DataFrame by the join keys in lexicographical order. Defaults to True, setting to False will improve performance substantially in many cases suffixes: A tuple of string sufxes to apply to overlapping columns. Defaults to (.x, .y). copy: Always copy data (default True) from the passed DataFrame objects, even when reindexing is not necessary. Cannot be avoided in many cases but may improve performance / memory usage. The cases where copying can be avoided are somewhat pathological but this option is provided nonetheless. merge is a function in the pandas namespace, and it is also available as a DataFrame instance method, with the calling DataFrame being implicitly considered the left object in the join. The related DataFrame.join method, uses merge internally for the index-on-index and index-on-column(s) joins, but joins on indexes by default rather than trying to join on common columns (the default behavior for merge). If you are joining on index, you may wish to use DataFrame.join to save yourself some typing.
182
183
0 1 2 3 4
3 NaN 1 1 2
6 7 4 5 NaN
In [893]: merge(left, right, how=inner) Out[893]: key1 key2 lval rval 0 bar one 3 6 1 foo one 1 4 2 foo one 1 5
The how argument to merge species how to determine which keys are to be included in the resulting table. If a key combination does not appear in either the left or right tables, the values in the joined table will be NA. Here is a summary of the how options and their SQL equivalent names: Merge method left right outer inner SQL Join Name LEFT OUTER JOIN RIGHT OUTER JOIN FULL OUTER JOIN INNER JOIN Description Use keys from left frame only Use keys from right frame only Use union of keys from both frames Use intersection of keys from both frames
184
A B C D 1 -2.461467 -1.553902 2.015523 -1.833722 2 1.771740 -0.670027 0.049307 -0.521493 3 -3.201750 0.792716 0.146111 1.903247 4 -0.747169 -0.309038 0.393876 1.861468 5 0.936527 1.255746 -2.655452 1.219492 6 0.062297 -0.110388 NaN NaN 7 0.077849 0.629498 NaN NaN In [900]: df1.join(df2, how=outer) Out[900]: A B C D 0 NaN NaN 0.377953 0.493672 1 -2.461467 -1.553902 2.015523 -1.833722 2 1.771740 -0.670027 0.049307 -0.521493 3 -3.201750 0.792716 0.146111 1.903247 4 -0.747169 -0.309038 0.393876 1.861468 5 0.936527 1.255746 -2.655452 1.219492 6 0.062297 -0.110388 NaN NaN 7 0.077849 0.629498 NaN NaN In [901]: df1.join(df2, how=inner) Out[901]: A B C D 1 -2.461467 -1.553902 2.015523 -1.833722 2 1.771740 -0.670027 0.049307 -0.521493 3 -3.201750 0.792716 0.146111 1.903247 4 -0.747169 -0.309038 0.393876 1.861468 5 0.936527 1.255746 -2.655452 1.219492
The data alignment here is on the indexes (row labels). This same behavior can be achieved using merge plus additional arguments instructing it to use the indexes:
In [902]: merge(df1, df2, left_index=True, right_index=True, how=outer) Out[902]: A B C D 0 NaN NaN 0.377953 0.493672 1 -2.461467 -1.553902 2.015523 -1.833722 2 1.771740 -0.670027 0.049307 -0.521493 3 -3.201750 0.792716 0.146111 1.903247 4 -0.747169 -0.309038 0.393876 1.861468 5 0.936527 1.255746 -2.655452 1.219492 6 0.062297 -0.110388 NaN NaN 7 0.077849 0.629498 NaN NaN
Obviously you can choose whichever form you nd more convenient. For many-to-one joins (where one of the DataFrames is already indexed by the join key), using join may be more convenient. Here is a simple example:
185
In [903]: df[key] = [foo, bar] * 4 In [904]: to_join = DataFrame(randn(2, 2), index=[bar, foo], .....: columns=[j1, j2]) .....: In [905]: df Out[905]: A B C D 0 -0.308853 -0.681087 0.377953 0.493672 1 -2.461467 -1.553902 2.015523 -1.833722 2 1.771740 -0.670027 0.049307 -0.521493 3 -3.201750 0.792716 0.146111 1.903247 4 -0.747169 -0.309038 0.393876 1.861468 5 0.936527 1.255746 -2.655452 1.219492 6 0.062297 -0.110388 -1.184357 -0.558081 7 0.077849 0.629498 -1.035260 -0.438229 In [906]: to_join Out[906]: j1 j2 bar 0.503703 0.413086 foo -1.139050 0.660342 In [907]: df.join(to_join, on=key) Out[907]: A B C D 0 -0.308853 -0.681087 0.377953 0.493672 1 -2.461467 -1.553902 2.015523 -1.833722 2 1.771740 -0.670027 0.049307 -0.521493 3 -3.201750 0.792716 0.146111 1.903247 4 -0.747169 -0.309038 0.393876 1.861468 5 0.936527 1.255746 -2.655452 1.219492 6 0.062297 -0.110388 -1.184357 -0.558081 7 0.077849 0.629498 -1.035260 -0.438229
In [908]: merge(df, to_join, left_on=key, right_index=True, .....: how=left, sort=False) .....: Out[908]: A B C D key j1 j2 0 -0.308853 -0.681087 0.377953 0.493672 foo -1.139050 0.660342 1 -2.461467 -1.553902 2.015523 -1.833722 bar 0.503703 0.413086 2 1.771740 -0.670027 0.049307 -0.521493 foo -1.139050 0.660342 3 -3.201750 0.792716 0.146111 1.903247 bar 0.503703 0.413086 4 -0.747169 -0.309038 0.393876 1.861468 foo -1.139050 0.660342 5 0.936527 1.255746 -2.655452 1.219492 bar 0.503703 0.413086 6 0.062297 -0.110388 -1.184357 -0.558081 foo -1.139050 0.660342 7 0.077849 0.629498 -1.035260 -0.438229 bar 0.503703 0.413086
186
In [910]: to_join = DataFrame(np.random.randn(10, 3), index=index, .....: columns=[j_one, j_two, j_three]) .....: # a little relevant example with NAs In [911]: key1 = [bar, bar, bar, foo, foo, baz, baz, qux, .....: qux, snap] .....: In [912]: key2 = [two, one, three, one, two, one, two, two, .....: three, one] .....: In [913]: data = np.random.randn(len(key1)) In [914]: data = DataFrame({key1 : key1, key2 : key2, .....: data : data}) .....: In [915]: data Out[915]: data key1 0 -1.004168 bar 1 -1.377627 bar 2 0.499281 bar 3 -1.405256 foo 4 0.162565 foo 5 -0.067785 baz 6 -1.260006 baz 7 -1.132896 qux 8 -2.006481 qux 9 0.301016 snap In [916]: to_join Out[916]: j_one first second foo one two three bar one two baz two three qux one two three 0.464794 0.683758 1.032814 1.515707 1.397431 -0.135950 0.281151 -0.851985 -1.537770 -0.390201 j_two -0.309337 -0.643834 -1.290493 -0.276487 1.503874 -0.730327 -1.298915 -1.106952 0.555759 1.207122 j_three -0.649593 0.421287 0.787872 -0.223762 -0.478905 -0.033277 -2.819487 -0.937731 -2.277282 0.178690
key2 two one three one two one two two three one
Now this can be joined by passing the two key column names:
In [917]: data.join(to_join, on=[key1, key2]) Out[917]: data key1 key2 j_one j_two j_three 0 -1.004168 bar two 1.397431 1.503874 -0.478905 1 -1.377627 bar one 1.515707 -0.276487 -0.223762 2 0.499281 bar three NaN NaN NaN 3 -1.405256 foo one 0.464794 -0.309337 -0.649593 4 0.162565 foo two 0.683758 -0.643834 0.421287 5 -0.067785 baz one NaN NaN NaN
187
two -0.135950 -0.730327 -0.033277 two -1.537770 0.555759 -2.277282 three -0.390201 1.207122 0.178690 one NaN NaN NaN
The default for DataFrame.join is to perform a left join (essentially a VLOOKUP operation, for Excel users), which uses only the keys found in the calling DataFrame. Other join types, for example inner join, can be just as easily performed:
In [918]: data.join(to_join, on=[key1, key2], how=inner) Out[918]: data key1 key2 j_one j_two j_three 0 -1.004168 bar two 1.397431 1.503874 -0.478905 1 -1.377627 bar one 1.515707 -0.276487 -0.223762 3 -1.405256 foo one 0.464794 -0.309337 -0.649593 4 0.162565 foo two 0.683758 -0.643834 0.421287 6 -1.260006 baz two -0.135950 -0.730327 -0.033277 7 -1.132896 qux two -1.537770 0.555759 -2.277282 8 -2.006481 qux three -0.390201 1.207122 0.178690
As you can see, this drops any rows where there was no match.
lvalue 1 2 3 1 2 3
188
In [923]: B Out[923]: key rvalue 0 b 1 1 c 2 2 d 3 In [924]: ordered_merge(A, B, fill_method=ffill, left_by=group) Out[924]: group key lvalue rvalue 0 a a 1 NaN 1 a b 1 1 2 a c 2 2 3 a d 2 3 4 a e 3 3 5 b a 1 NaN 6 b b 1 1 7 b c 2 2 8 b d 2 3 9 b e 3 3
189
Note that this method only takes values from the right DataFrame if they are missing in the left DataFrame. A related method, update, alters non-NA values inplace:
In [933]: df1.update(df2) In [934]: df1 Out[934]: 0 1 2 0 NaN 3.0 5.0 1 -42.6 NaN -8.2 2 -5.0 1.6 4.0
190
CHAPTER
TWELVE
For the curious here is how the above DataFrame was created:
import pandas.util.testing as tm; tm.N = 3 def unpivot(frame): N, K = frame.shape data = {value : frame.values.ravel(F), variable : np.asarray(frame.columns).repeat(N), date : np.tile(np.asarray(frame.index), K)} return DataFrame(data, columns=[date, variable, value]) df = unpivot(tm.makeTimeDataFrame())
But suppose we wish to do time series operations with the variables. A better representation would be where the columns are the unique variables and an index of dates identies individual observations. To reshape the data into this form, use the pivot function:
191
In [1003]: df.pivot(index=date, columns=variable, values=value) Out[1003]: variable A B C D date 2000-01-03 0.469112 -1.135632 0.119209 -2.104569 2000-01-04 -0.282863 1.212112 -1.044236 -0.494929 2000-01-05 -1.509059 -0.173215 -0.861849 1.071804
If the values argument is omitted, and the input DataFrame has more than one column of values which are not used as column or index inputs to pivot, then the resulting pivoted DataFrame will have hierarchical columns whose topmost level indicates the respective value column:
In [1004]: df[value2] = df[value] * 2 In [1005]: pivoted = df.pivot(date, variable) In [1006]: pivoted Out[1006]: <class pandas.core.frame.DataFrame> DatetimeIndex: 3 entries, 2000-01-03 00:00:00 to 2000-01-05 00:00:00 Data columns: (value, A) 3 non-null values (value, B) 3 non-null values (value, C) 3 non-null values (value, D) 3 non-null values (value2, A)3 non-null values (value2, B)3 non-null values (value2, C)3 non-null values (value2, D)3 non-null values dtypes: float64(8)
You of course can then select subsets from the pivoted DataFrame:
In [1007]: pivoted[value2] Out[1007]: variable A B C D date 2000-01-03 0.938225 -2.271265 0.238417 -4.209138 2000-01-04 -0.565727 2.424224 -2.088472 -0.989859 2000-01-05 -3.018117 -0.346429 -1.723698 2.143608
Note that this returns a view on the underlying data in the case where the data are homogeneously-typed.
192
In [1008]: tuples = zip(*[[bar, ......: foo, ......: [one, ......: one, ......:
In [1009]: index = MultiIndex.from_tuples(tuples, names=[first, second]) In [1010]: df = DataFrame(randn(8, 2), index=index, columns=[A, B]) In [1011]: df2 = df[:4] In [1012]: df2 Out[1012]: A B first second bar one 0.721555 -0.706771 two -1.039575 0.271860 baz one -0.424972 0.567020 two 0.276232 -1.087401
The stack function compresses a level in the DataFrames columns to produce either: A Series, in the case of a simple column Index A DataFrame, in the case of a MultiIndex in the columns If the columns have a MultiIndex, you can choose which level to stack. The stacked level becomes the new lowest level in a MultiIndex on the columns:
In [1013]: stacked = df2.stack() In [1014]: stacked Out[1014]: first second bar one A B two A B baz one A B two A B
With a stacked DataFrame or Series (having a MultiIndex as the index), the inverse operation of stack is unstack, which by default unstacks the last level:
In [1015]: stacked.unstack() Out[1015]: A B first second bar one 0.721555 -0.706771 two -1.039575 0.271860 baz one -0.424972 0.567020 two 0.276232 -1.087401 In [1016]: stacked.unstack(1) Out[1016]: second one two first
193
bar baz
In [1017]: stacked.unstack(0) Out[1017]: first bar baz second one A 0.721555 -0.424972 B -0.706771 0.567020 two A -1.039575 0.276232 B 0.271860 -1.087401
If the indexes have names, you can use the level names instead of specifying the level numbers:
In [1018]: stacked.unstack(second) Out[1018]: second one two first bar A 0.721555 -1.039575 B -0.706771 0.271860 baz A -0.424972 0.276232 B 0.567020 -1.087401
You may also stack or unstack more than one level at a time by passing a list of levels, in which case the end result is as if each level in the list were processed individually. These functions are intelligent about handling missing data and do not expect each subgroup within the hierarchical index to have the same set of labels. They also can handle the index being unsorted (but you can make it sorted by calling sortlevel, of course). Here is a more complex example:
In [1019]: columns = MultiIndex.from_tuples([(A, cat), (B, dog), ......: (B, cat), (A, dog)], ......: names=[exp, animal]) ......: In [1020]: df = DataFrame(randn(8, 4), index=index, columns=columns) In [1021]: df2 = df.ix[[0, 1, 2, 4, 5, 7]] In [1022]: df2 Out[1022]: exp A animal cat first second bar one -0.370647 two 1.075770 baz one 0.357021 foo one -0.013960 two 0.895717 qux two 0.410835
As mentioned above, stack can be called with a level argument to select which level in the columns to stack:
In [1023]: df2.stack(exp) Out[1023]: animal cat first second exp
dog
194
bar
one two
baz foo
qux
two
A B A B A B A B A B A B
-0.370647 -1.344312 1.075770 1.643563 0.357021 -1.776904 -0.013960 -0.006154 0.895717 -1.206412 0.410835 0.132003
0.844885 -1.157892 -1.469388 -0.109050 -0.968914 -0.674600 -0.923061 -0.362543 2.565646 0.805244 -0.827317 0.813850
In [1024]: df2.stack(animal) Out[1024]: exp A B first second animal bar one cat -0.370647 -1.344312 dog 0.844885 -1.157892 two cat 1.075770 1.643563 dog -1.469388 -0.109050 baz one cat 0.357021 -1.776904 dog -0.968914 -0.674600 foo one cat -0.013960 -0.006154 dog -0.923061 -0.362543 two cat 0.895717 -1.206412 dog 2.565646 0.805244 qux two cat 0.410835 0.132003 dog -0.827317 0.813850
Unstacking when the columns are a MultiIndex is also careful about doing the right thing:
In [1025]: df[:3].unstack(0) Out[1025]: <class pandas.core.frame.DataFrame> Index: 2 entries, one to two Data columns: (A, cat, bar2 non-null values (A, cat, baz1 non-null values (B, dog, bar2 non-null values (B, dog, baz1 non-null values (B, cat, bar2 non-null values (B, cat, baz1 non-null values (A, dog, bar2 non-null values (A, dog, baz1 non-null values dtypes: float64(8) In [1026]: df2.unstack(1) Out[1026]: <class pandas.core.frame.DataFrame> Index: 4 entries, bar to qux Data columns: (A, cat, one3 non-null values (A, cat, two3 non-null values (B, dog, one3 non-null values (B, dog, two3 non-null values (B, cat, one3 non-null values (B, cat, two3 non-null values
195
In [1029]: melt(cheese, id_vars=[first, last]) Out[1029]: first last variable value 0 John Doe height 5.5 1 Mary Bo height 6.0 2 John Doe weight 130.0 3 Mary Bo weight 150.0
196
first second bar one -0.857479 two 1.359666 baz one -0.709942 two -0.508931 foo one -0.010057 two -0.155347 qux one 0.130479 two 0.271419
# same result, another way In [1032]: df.groupby(level=1, axis=1).mean() Out[1032]: animal cat dog first second bar one -0.857479 -0.156504 two 1.359666 -0.789219 baz one -0.709942 -0.821757 two -0.508931 -0.029148 foo one -0.010057 -0.642802 two -0.155347 1.685445 qux one 0.130479 0.557070 two 0.271419 -0.006733 In [1033]: df.stack().groupby(level=1).mean() Out[1033]: exp A B second one 0.016301 -0.644049 two 0.110588 0.346200 In [1034]: df.mean().unstack(0) Out[1034]: exp A B animal cat 0.311433 -0.431481 dog -0.184544 0.133632
197
......: ......: ......: In [1036]: Out[1036]: A 0 one 1 one 2 two 3 three 4 one 5 one 6 two 7 three 8 one 9 one 10 two 11 three 12 one 13 one 14 two 15 three 16 one 17 one 18 two 19 three 20 one 21 one 22 two 23 three df B A B C A B C A B C A B C A B C A B C A B C A B C C foo foo foo bar bar bar foo foo foo bar bar bar foo foo foo bar bar bar foo foo foo bar bar bar
D : np.random.randn(24), E : np.random.randn(24)})
D -0.076467 -1.187678 1.130127 -1.436737 -1.413681 1.607920 1.024180 0.569605 0.875906 -2.211372 0.974466 -2.006747 -0.410001 -0.078638 0.545952 -1.219217 -1.226825 0.769804 -1.281247 -0.727707 -0.121306 -0.097883 0.695775 0.341734
E 0.959726 -1.110336 -0.619976 0.149748 -0.732339 0.687738 0.176444 0.403310 -0.154951 0.301624 -2.179861 -1.369849 -0.954208 1.462696 -1.743161 -0.826591 -0.345352 1.314232 0.690579 0.995761 2.396780 0.014871 3.357427 -0.317441
198
<class pandas.core.frame.DataFrame> Index: 3 entries, A to C Data columns: (D, one, bar)3 non-null values (D, one, foo)3 non-null values (D, three, bar2 non-null values (D, three, foo1 non-null values (D, two, bar)1 non-null values (D, two, foo)2 non-null values (E, one, bar)3 non-null values (E, one, foo)3 non-null values (E, three, bar2 non-null values (E, three, foo1 non-null values (E, two, bar)1 non-null values (E, two, foo)2 non-null values dtypes: float64(12)
The result object is a DataFrame having potentially hierarchical indexes on the rows and columns. If the values column name is not given, the pivot table will include all of the data that can be aggregated in an additional level of hierarchy in the columns:
In [1040]: pivot_table(df, rows=[A, B], cols=[C]) Out[1040]: D E C bar foo bar foo A B one A -1.154627 -0.243234 0.158248 0.002759 B -1.320253 -0.633158 -0.538846 0.176180 C 1.188862 0.377300 1.000985 1.120915 three A -1.327977 NaN -0.338421 NaN B NaN -0.079051 NaN 0.699535 C -0.832506 NaN -0.843645 NaN two A NaN -0.128534 NaN 0.433512 B 0.835120 NaN 0.588783 NaN C NaN 0.838040 NaN -1.181568
You can render a nice output of the table omitting the missing values by calling to_string if you wish:
In [1041]: table = pivot_table(df, rows=[A, B], cols=[C]) In [1042]: print table.to_string(na_rep=) D E C bar foo bar foo A B one A -1.154627 -0.243234 0.158248 0.002759 B -1.320253 -0.633158 -0.538846 0.176180 C 1.188862 0.377300 1.000985 1.120915 three A -1.327977 -0.338421 B -0.079051 0.699535 C -0.832506 -0.843645 two A -0.128534 0.433512 B 0.835120 0.588783 C 0.838040 -1.181568
199
200
C All
NaN 1.294620
0.413074 0.824989
0.413074 1.064129
NaN 1.403041
0.794212 1.188419
0.794212 1.248988
12.6 Tiling
The cut function computes groupings for the values of the input array and is often used to transform continuous variables to discrete or categorical variables:
In [1049]: ages = np.array([10, 15, 13, 12, 23, 25, 28, 59, 60]) In [1050]: cut(ages, bins=3) Out[1050]: Categorical: array([(9.95, 26.667], (9.95, 26.667], (9.95, 26.667], (9.95, 26.667], (9.95, 26.667], (9.95, 26.667], (26.667, 43.333], (43.333, 60], (43.333, 60]], dtype=object) Levels (3): Index([(9.95, 26.667], (26.667, 43.333], (43.333, 60]], dtype=object)
If the bins keyword is an integer, then equal-width bins are formed. Alternatively we can specify custom bin-edges:
In [1051]: cut(ages, bins=[0, 18, 35, 70]) Out[1051]: Categorical: array([(0, 18], (0, 18], (0, 18], (0, 18], (18, 35], (18, 35], (18, 35], (35, 70], (35, 70]], dtype=object) Levels (3): Index([(0, 18], (18, 35], (35, 70]], dtype=object)
12.6. Tiling
201
202
CHAPTER
THIRTEEN
203
converted.head() 00:00:00 00:45:00 01:30:00 02:15:00 03:00:00 0.469112 0.469112 -0.282863 -1.509059 -1.135632
Resample:
# Daily means In [1081]: ts.resample(D, how=mean) Out[1081]: 2011-01-01 0.469112 2011-01-02 -0.322252 2011-01-03 -0.317244 2011-01-04 0.083412 Freq: D
However, in many cases it is more natural to associate things like change variables with a time span instead. For example:
In [1086]: periods = PeriodIndex([Period(2012-01), Period(2012-02), ......: Period(2012-03)]) ......: In [1087]: ts = Series(np.random.randn(3), periods) In [1088]: type(ts.index) Out[1088]: pandas.tseries.period.PeriodIndex In [1089]: ts Out[1089]: 2012-01 -1.219217 2012-02 -1.226825 2012-03 0.769804 Freq: M
204
Starting with 0.8, pandas allows you to capture both representations and convert between them. Under the hood, pandas represents timestamps using instances of Timestamp and sequences of timestamps using instances of DatetimeIndex. For regular time spans, pandas uses Period objects for scalar values and PeriodIndex for sequences of spans. Better support for irregular intervals with arbitrary start and end points are forth-coming in future releases.
Practically, this becomes very cumbersome because we often need a very long index with a large number of timestamps. If we need timestamps on a regular frequency, we can use the pandas functions date_range and bdate_range to create timestamp indexes.
In [1095]: index = date_range(2000-1-1, periods=1000, freq=M) In [1096]: index Out[1096]: <class pandas.tseries.index.DatetimeIndex> [2000-01-31 00:00:00, ..., 2083-04-30 00:00:00] Length: 1000, Freq: M, Timezone: None In [1097]: index = bdate_range(2012-1-1, periods=250) In [1098]: index Out[1098]: <class pandas.tseries.index.DatetimeIndex> [2012-01-02 00:00:00, ..., 2012-12-14 00:00:00] Length: 250, Freq: B, Timezone: None
Convenience functions like date_range and bdate_range utilize a variety of frequency aliases. The default frequency for date_range is a calendar day while the default for bdate_range is a business day
In [1099]: start = datetime(2011, 1, 1) In [1100]: end = datetime(2012, 1, 1) In [1101]: rng = date_range(start, end)
205
In [1102]: rng Out[1102]: <class pandas.tseries.index.DatetimeIndex> [2011-01-01 00:00:00, ..., 2012-01-01 00:00:00] Length: 366, Freq: D, Timezone: None In [1103]: rng = bdate_range(start, end) In [1104]: rng Out[1104]: <class pandas.tseries.index.DatetimeIndex> [2011-01-03 00:00:00, ..., 2011-12-30 00:00:00] Length: 260, Freq: B, Timezone: None
date_range and bdate_range makes it easy to generate a range of dates using various combinations of parameters like start, end, periods, and freq:
In [1105]: date_range(start, end, freq=BM) Out[1105]: <class pandas.tseries.index.DatetimeIndex> [2011-01-31 00:00:00, ..., 2011-12-30 00:00:00] Length: 12, Freq: BM, Timezone: None In [1106]: date_range(start, end, freq=W) Out[1106]: <class pandas.tseries.index.DatetimeIndex> [2011-01-02 00:00:00, ..., 2012-01-01 00:00:00] Length: 53, Freq: W-SUN, Timezone: None In [1107]: bdate_range(end=end, periods=20) Out[1107]: <class pandas.tseries.index.DatetimeIndex> [2011-12-05 00:00:00, ..., 2011-12-30 00:00:00] Length: 20, Freq: B, Timezone: None In [1108]: bdate_range(start=start, periods=20) Out[1108]: <class pandas.tseries.index.DatetimeIndex> [2011-01-03 00:00:00, ..., 2011-01-28 00:00:00] Length: 20, Freq: B, Timezone: None
The start and end dates are strictly inclusive. So it will not generate any dates outside of those dates if specied.
13.2.1 DatetimeIndex
One of the main uses for DatetimeIndex is as an index for pandas objects. The DatetimeIndex class contains many timeseries related optimizations: A large range of dates for various offsets are pre-computed and cached under the hood in order to make generating subsequent date ranges very fast (just have to grab a slice) Fast shifting using the shift and tshift method on pandas objects Unioning of overlapping DatetimeIndex objects with the same frequency is very fast (important for fast data alignment) Quick access to date elds via properties such as year, month, etc.
206
Regularization functions like snap and very fast asof logic DatetimeIndex can be used like a regular index and offers all of its intelligent functionality like selection, slicing, etc.
In [1109]: rng = date_range(start, end, freq=BM) In [1110]: ts = Series(randn(len(rng)), index=rng) In [1111]: ts.index Out[1111]: <class pandas.tseries.index.DatetimeIndex> [2011-01-31 00:00:00, ..., 2011-12-30 00:00:00] Length: 12, Freq: BM, Timezone: None In [1112]: ts[:5].index Out[1112]: <class pandas.tseries.index.DatetimeIndex> [2011-01-31 00:00:00, ..., 2011-05-31 00:00:00] Length: 5, Freq: BM, Timezone: None In [1113]: ts[::2].index Out[1113]: <class pandas.tseries.index.DatetimeIndex> [2011-01-31 00:00:00, ..., 2011-11-30 00:00:00] Length: 6, Freq: 2BM, Timezone: None
You can pass in dates and strings that parses to dates as indexing parameters:
In [1114]: ts[1/31/2011] Out[1114]: -1.2812473076599531 In [1115]: ts[datetime(2011, 12, 25):] Out[1115]: 2011-12-30 0.687738 Freq: BM In [1116]: ts[10/31/2011:12/31/2011] Out[1116]: 2011-10-31 0.149748 2011-11-30 -0.732339 2011-12-30 0.687738 Freq: BM
To provide convenience for accessing longer time series, you can also pass in the year or year and month as strings:
In [1118]: ts[2011] Out[1118]: 2011-01-31 -1.281247 2011-02-28 -0.727707 2011-03-31 -0.121306
207
2011-04-29 2011-05-31 2011-06-30 2011-07-29 2011-08-31 2011-09-30 2011-10-31 2011-11-30 2011-12-30 Freq: BM
Even complicated fancy indexing that breaks the DatetimeIndexs frequency regularity will result in a DatetimeIndex (but frequency is lost):
In [1120]: ts[[0, 2, 6]].index Out[1120]: <class pandas.tseries.index.DatetimeIndex> [2011-01-31 00:00:00, ..., 2011-07-29 00:00:00] Length: 3, Freq: None, Timezone: None
DatetimeIndex objects has all the basic functionality of regular Index objects and a smorgasbord of advanced timeseries-specic methods for easy frequency processing. See Also: Reindexing methods Note: While pandas does not force you to have a sorted date index, some of these methods may have unexpected or incorrect behavior if the dates are unsorted. So please be careful.
208
Class name DateOffset BDay Week WeekOfMonth MonthEnd MonthBegin BMonthEnd BMonthBegin QuarterEnd QuarterBegin BQuarterEnd BQuarterBegin YearEnd YearBegin BYearEnd BYearBegin Hour Minute Second Milli Micro
Description Generic offset class, defaults to 1 calendar day business day (weekday) one week, optionally anchored on a day of the week the x-th day of the y-th week of each month calendar month end calendar month begin business month end business month begin calendar quarter end calendar quarter begin business quarter end business quarter begin calendar year end calendar year begin business year end business year begin one hour one minute one second one millisecond one microsecond
The basic DateOffset takes the same arguments as dateutil.relativedelta, which works like:
In [1121]: d = datetime(2008, 8, 18) In [1122]: d + relativedelta(months=4, days=5) Out[1122]: datetime.datetime(2008, 12, 23, 0, 0)
The key features of a DateOffset object are: it can be added / subtracted to/from a datetime object to obtain a shifted date it can be multiplied by an integer (positive or negative) so that the increment will be applied multiple times it has rollforward and rollback methods for moving a date forward or backward to the next or previous offset date Subclasses of DateOffset dene the apply function which dictates custom date increment logic, such as adding business days:
class BDay(DateOffset): """DateOffset increments between business days""" def apply(self, other): ... In [1125]: d - 5 * BDay() Out[1125]: datetime.datetime(2008, 8, 11, 0, 0) In [1126]: d + BMonthEnd() Out[1126]: datetime.datetime(2008, 8, 29, 0, 0)
209
The rollforward and rollback methods do exactly what you would expect:
In [1127]: d Out[1127]: datetime.datetime(2008, 8, 18, 0, 0) In [1128]: offset = BMonthEnd() In [1129]: offset.rollforward(d) Out[1129]: datetime.datetime(2008, 8, 29, 0, 0) In [1130]: offset.rollback(d) Out[1130]: datetime.datetime(2008, 7, 31, 0, 0)
Its denitely worth exploring the pandas.tseries.offsets module and the various docstrings for the classes.
210
Description business day frequency calendar day frequency weekly frequency month end frequency business month end frequency month start frequency business month start frequency quarter end frequency business quarter endfrequency quarter start frequency business quarter start frequency year end frequency business year end frequency year start frequency business year start frequency hourly frequency minutely frequency secondly frequency milliseonds microseconds
211
These can be used as arguments to date_range, bdate_range, constructors for DatetimeIndex, as well as various other timeseries-related functions in pandas.
212
Legacy Time Rule WEEKDAY EOM W@MON W@TUE W@WED W@THU W@FRI W@SAT W@SUN Q@JAN Q@FEB Q@MAR A@JAN A@FEB A@MAR A@APR A@MAY A@JUN A@JUL A@AUG A@SEP A@OCT A@NOV A@DEC min ms us: U
Offset Alias B BM W-MON W-TUE W-WED W-THU W-FRI W-SAT W-SUN BQ-JAN BQ-FEB BQ-MAR BA-JAN BA-FEB BA-MAR BA-APR BA-MAY BA-JUN BA-JUL BA-AUG BA-SEP BA-OCT BA-NOV BA-DEC T L
As you can see, legacy quarterly and annual frequencies are business quarter and business year ends. Please also note the legacy time rule for milliseconds ms versus the new offset alias for month start MS. This means that offset alias parsing is case sensitive.
The shift method accepts an freq argument which can accept a DateOffset class or other timedelta-like object
213
Rather than changing the alignment of the data and the index, DataFrame and TimeSeries objects also have a tshift convenience method that changes all the dates in the index by a specied number of offsets:
In [1144]: ts.tshift(5, freq=D) Out[1144]: 2011-02-05 -1.281247 2011-03-05 -0.727707 2011-04-05 -0.121306 2011-05-04 -0.097883 2011-06-05 0.695775
Note that with tshift, the leading entry is no longer NaN because the data is not being realigned.
214
asfreq provides a further convenience so you can specify an interpolation method for any gaps that may appear after the frequency conversion
In [1149]: ts.asfreq(BDay(), method=pad) Out[1149]: 2010-01-01 0.176444 2010-01-04 0.176444 2010-01-05 0.176444 2010-01-06 0.403310 2010-01-07 0.403310 2010-01-08 0.403310 2010-01-11 -0.154951 Freq: B
The resample function is very exible and allows you to specify many different parameters to control the frequency conversion and resampling operation. The how parameter can be a function name or numpy array function that takes an array and produces aggregated values:
In [1153]: ts.resample(5Min) # default is mean Out[1153]: 2012-01-01 00:00:00 230.00000 2012-01-01 00:05:00 258.20202 Freq: 5T In [1154]: ts.resample(5Min, how=ohlc) Out[1154]: open high low close 2012-01-01 00:00:00 230 230 230 230 2012-01-01 00:05:00 202 492 0 214 In [1155]: ts.resample(5Min, how=np.max) Out[1155]: 2012-01-01 00:00:00 230
215
492
Any function available via dispatching can be given to the how parameter by name, including sum, mean, std, max, min, median, first, last, ohlc. For downsampling, closed can be set to left or right to specify which end of the interval is closed:
In [1156]: ts.resample(5Min, closed=right) Out[1156]: 2012-01-01 00:00:00 230.00000 2012-01-01 00:05:00 258.20202 Freq: 5T In [1157]: ts.resample(5Min, closed=left) Out[1157]: 2012-01-01 00:05:00 257.92 Freq: 5T
For upsampling, the fill_method and limit parameters can be specied to interpolate over the gaps that are created:
# from secondly to every 250 milliseconds In [1158]: ts[:2].resample(250L) Out[1158]: 2012-01-01 00:00:00 230 2012-01-01 00:00:00.250000 NaN 2012-01-01 00:00:00.500000 NaN 2012-01-01 00:00:00.750000 NaN 2012-01-01 00:00:01 202 Freq: 250L In [1159]: Out[1159]: 2012-01-01 2012-01-01 2012-01-01 2012-01-01 2012-01-01 Freq: 250L In [1160]: Out[1160]: 2012-01-01 2012-01-01 2012-01-01 2012-01-01 2012-01-01 Freq: 250L ts[:2].resample(250L, fill_method=pad) 00:00:00 00:00:00.250000 00:00:00.500000 00:00:00.750000 00:00:01 230 230 230 230 202
ts[:2].resample(250L, fill_method=pad, limit=2) 00:00:00 00:00:00.250000 00:00:00.500000 00:00:00.750000 00:00:01 230 230 230 NaN 202
Parameters like label and loffset are used to manipulate the resulting labels. label species whether the result is labeled with the beginning or the end of the interval. loffset performs a time adjustment on the output labels.
In [1161]: ts.resample(5Min) # by default label=right Out[1161]: 2012-01-01 00:00:00 230.00000 2012-01-01 00:05:00 258.20202 Freq: 5T
216
In [1162]: ts.resample(5Min, label=left) Out[1162]: 2011-12-31 23:55:00 230.00000 2012-01-01 00:00:00 258.20202 Freq: 5T In [1163]: ts.resample(5Min, label=left, loffset=1s) Out[1163]: 2011-12-31 23:55:01 230.00000 2012-01-01 00:00:01 258.20202
The axis parameter can be set to 0 or 1 and allows you to resample the specied axis for a DataFrame. kind can be set to timestamp or period to convert the resulting index to/from time-stamp and time-span representations. By default resample retains the input representation. convention can be set to start or end when resampling period data (detail below). It species how low frequency periods are converted to higher frequency periods. Note that 0.8 marks a watershed in the timeseries functionality in pandas. In previous versions, resampling had to be done using a combination of date_range, groupby with asof, and then calling an aggregation function on the grouped object. This was not nearly convenient or performant as the new pandas timeseries API.
13.6.1 Period
A Period represents a span of time (e.g., a day, a month, a quarter, etc). It can be created using a frequency alias:
In [1164]: Period(2012, freq=A-DEC) Out[1164]: Period(2012, A-DEC) In [1165]: Period(2012-1-1, freq=D) Out[1165]: Period(2012-01-01, D) In [1166]: Period(2012-1-1 19:00, freq=H) Out[1166]: Period(2012-01-01 19:00, H)
Unlike time stamped data, pandas does not support frequencies at multiples of DateOffsets (e.g., 3Min) for periods. Adding and subtracting integers from periods shifts the period by its own frequency.
In [1167]: p = Period(2012, freq=A-DEC) In [1168]: p + 1 Out[1168]: Period(2013, A-DEC) In [1169]: p - 3 Out[1169]: Period(2009, A-DEC)
Taking the difference of Period instances with the same frequency will return the number of frequency units between them:
217
Just like DatetimeIndex, a PeriodIndex can also be used to index pandas objects:
In [1174]: Series(randn(len(prng)), prng) Out[1174]: 2011-01 0.301624 2011-02 -1.460489 2011-03 0.610679 2011-04 1.195856 2011-05 -0.008820 2011-06 -0.045729 2011-07 -1.051015 2011-08 -0.422924 2011-09 -0.028361 2011-10 -0.782386 2011-11 0.861980 2011-12 1.438604 2012-01 -0.525492 Freq: M
218
We can convert it to a monthly frequency. Using the how parameter, we can specify whether to return the starting or ending month:
In [1177]: p.asfreq(M, how=start) Out[1177]: Period(2011-01, M) In [1178]: p.asfreq(M, how=end) Out[1178]: Period(2011-12, M)
Converting to a super-period (e.g., annual frequency is a super-period of quarterly frequency) automatically returns the super-period that includes the input period:
In [1181]: p = Period(2011-12, freq=M) In [1182]: p.asfreq(A-NOV) Out[1182]: Period(2012, A-NOV)
Note that since we converted to an annual frequency that ends the year in November, the monthly period of December 2011 is actually in the 2012 A-NOV period. Period conversions with anchored frequencies are particularly useful for working with various quarterly data common to economics, business, and other elds. Many organizations dene quarters relative to the month in which their scal year start and ends. Thus, rst quarter of 2011 could start in 2010 or a few months into 2011. Via anchored frequencies, pandas works all quarterly frequencies Q-JAN through Q-DEC. Q-DEC dene regular calendar quarters:
In [1183]: p = Period(2012Q1, freq=Q-DEC) In [1184]: p.asfreq(D, s) Out[1184]: Period(2012-01-01, D) In [1185]: p.asfreq(D, e) Out[1185]: Period(2012-03-31, D)
219
In [1189]: rng = date_range(1/1/2012, periods=5, freq=M) In [1190]: ts = Series(randn(len(rng)), index=rng) In [1191]: ts Out[1191]: 2012-01-31 -1.684469 2012-02-29 0.550605 2012-03-31 0.091955 2012-04-30 0.891713 2012-05-31 0.807078 Freq: M In [1192]: ps = ts.to_period() In [1193]: ps Out[1193]: 2012-01 -1.684469 2012-02 0.550605 2012-03 0.091955 2012-04 0.891713 2012-05 0.807078 Freq: M In [1194]: ps.to_timestamp() Out[1194]: 2012-01-01 -1.684469 2012-02-01 0.550605 2012-03-01 0.091955 2012-04-01 0.891713 2012-05-01 0.807078 Freq: MS
Remember that s and e can be used to return the timestamps at the start or end of the period:
In [1195]: ps.to_timestamp(D, how=s) Out[1195]: 2012-01-01 -1.684469 2012-02-01 0.550605 2012-03-01 0.091955 2012-04-01 0.891713 2012-05-01 0.807078 Freq: MS
Converting between period and timestamp enables some convenient arithmetic functions to be used. In the following example, we convert a quarterly frequency with year ending in November to 9am of the end of the month following the quarter end:
In [1196]: prng = period_range(1990Q1, 2000Q4, freq=Q-NOV) In [1197]: ts = Series(randn(len(prng)), prng) In [1198]: ts.index = (prng.asfreq(M, e) + 1).asfreq(H, s) + 9 In [1199]: Out[1199]: 1990-03-01 1990-06-01 1990-09-01 ts.head() 09:00 09:00 09:00 0.221441 -0.113139 -1.812900
220
-0.053708 -0.114574
To supply the time zone, you can use the tz keyword to date_range and other functions:
In [1202]: rng_utc = date_range(3/6/2012 00:00, periods=10, freq=D, tz=UTC) In [1203]: print(rng_utc.tz) UTC
Timestamps, like Pythons datetime.datetime object can be either time zone naive or time zone aware. Naive time series and DatetimeIndex objects can be localized using tz_localize:
In [1204]: ts = Series(randn(len(rng)), rng) In [1205]: ts_utc = ts.tz_localize(UTC) In [1206]: Out[1206]: 2012-03-06 2012-03-07 2012-03-08 2012-03-09 2012-03-10 2012-03-11 2012-03-12 2012-03-13 2012-03-14 2012-03-15 2012-03-16 2012-03-17 2012-03-18 2012-03-19 2012-03-20 Freq: D ts_utc 00:00:00+00:00 00:00:00+00:00 00:00:00+00:00 00:00:00+00:00 00:00:00+00:00 00:00:00+00:00 00:00:00+00:00 00:00:00+00:00 00:00:00+00:00 00:00:00+00:00 00:00:00+00:00 00:00:00+00:00 00:00:00+00:00 00:00:00+00:00 00:00:00+00:00 -0.114722 0.168904 -0.048048 0.801196 1.392071 -0.048788 -0.808838 -1.003677 -0.160766 1.758853 0.729195 1.359732 2.006296 0.870210 0.043464
You can use the tz_convert method to convert pandas objects to convert tz-aware data to another time zone:
In [1207]: Out[1207]: 2012-03-05 2012-03-06 2012-03-07 2012-03-08 2012-03-09 2012-03-10 ts_utc.tz_convert(US/Eastern) 19:00:00-05:00 19:00:00-05:00 19:00:00-05:00 19:00:00-05:00 19:00:00-05:00 19:00:00-05:00 -0.114722 0.168904 -0.048048 0.801196 1.392071 -0.048788
221
2012-03-11 2012-03-12 2012-03-13 2012-03-14 2012-03-15 2012-03-16 2012-03-17 2012-03-18 2012-03-19 Freq: D
Under the hood, all timestamps are stored in UTC. Scalar values from a DatetimeIndex with a time zone will have their elds (day, hour, minute) localized to the time zone. However, timestamps with the same UTC value are still considered to be equal even if they are in different time zones:
In [1208]: rng_eastern = rng_utc.tz_convert(US/Eastern) In [1209]: rng_berlin = rng_utc.tz_convert(Europe/Berlin) In [1210]: rng_eastern[5] Out[1210]: <Timestamp: 2012-03-10 19:00:00-0500 EST, tz=US/Eastern> In [1211]: rng_berlin[5] Out[1211]: <Timestamp: 2012-03-11 01:00:00+0100 CET, tz=Europe/Berlin> In [1212]: rng_eastern[5] == rng_berlin[5] Out[1212]: True
Like Series, DataFrame, and DatetimeIndex, Timestamps can be converted to other time zones using tz_convert:
In [1213]: rng_eastern[5] Out[1213]: <Timestamp: 2012-03-10 19:00:00-0500 EST, tz=US/Eastern> In [1214]: rng_berlin[5] Out[1214]: <Timestamp: 2012-03-11 01:00:00+0100 CET, tz=Europe/Berlin> In [1215]: rng_eastern[5].tz_convert(Europe/Berlin) Out[1215]: <Timestamp: 2012-03-11 01:00:00+0100 CET, tz=Europe/Berlin>
Operations between TimeSeries in difcult time zones will yield UTC TimeSeries, aligning the data on the UTC timestamps:
In [1218]: eastern = ts_utc.tz_convert(US/Eastern) In [1219]: berlin = ts_utc.tz_convert(Europe/Berlin) In [1220]: result = eastern + berlin In [1221]: result Out[1221]: 2012-03-06 00:00:00+00:00
-0.229443
222
2012-03-07 2012-03-08 2012-03-09 2012-03-10 2012-03-11 2012-03-12 2012-03-13 2012-03-14 2012-03-15 2012-03-16 2012-03-17 2012-03-18 2012-03-19 2012-03-20 Freq: D
00:00:00+00:00 00:00:00+00:00 00:00:00+00:00 00:00:00+00:00 00:00:00+00:00 00:00:00+00:00 00:00:00+00:00 00:00:00+00:00 00:00:00+00:00 00:00:00+00:00 00:00:00+00:00 00:00:00+00:00 00:00:00+00:00 00:00:00+00:00
0.337809 -0.096096 1.602392 2.784142 -0.097575 -1.617677 -2.007353 -0.321532 3.517706 1.458389 2.719465 4.012592 1.740419 0.086928
In [1222]: result.index Out[1222]: <class pandas.tseries.index.DatetimeIndex> [2012-03-06 00:00:00, ..., 2012-03-20 00:00:00] Length: 15, Freq: D, Timezone: UTC
223
224
CHAPTER
FOURTEEN
225
If the index consists of dates, it calls gcf().autofmt_xdate() to try to format the x-axis nicely as per above. The method takes a number of arguments for controlling the look of the plot:
In [1227]: plt.figure(); ts.plot(style=k--, label=Series); plt.legend() Out[1227]: <matplotlib.legend.Legend at 0xdc74650>
226
You may set the legend argument to False to hide the legend, which is shown by default.
In [1231]: df.plot(legend=False) Out[1231]: <matplotlib.axes.AxesSubplot at 0xe2364d0>
Some other options are available, like plotting each Series on a different axis:
In [1232]: df.plot(subplots=True, figsize=(8, 8)); plt.legend(loc=best) Out[1232]: <matplotlib.legend.Legend at 0xdc685d0>
227
228
You can plot one column versus another using the x and y keywords in DataFrame.plot:
In [1236]: plt.figure() Out[1236]: <matplotlib.figure.Figure at 0xfbe3f50> In [1237]: df3 = DataFrame(np.random.randn(1000, 2), columns=[B, C]).cumsum() In [1238]: df3[A] = Series(range(len(df))) In [1239]: df3.plot(x=A, y=B) Out[1239]: <matplotlib.axes.AxesSubplot at 0xf82fd90>
229
230
Note that the columns plotted on the secondary y-axis is automatically marked with (right) in the legend. To turn off the automatic marking, use the mark_right=False keyword:
In [1245]: plt.figure() Out[1245]: <matplotlib.figure.Figure at 0x10765f10> In [1246]: df.plot(secondary_y=[A, B], mark_right=False) Out[1246]: <matplotlib.axes.AxesSubplot at 0x103563d0>
231
232
If you have more than one plot that needs to be suppressed, the use method in pandas.plot_params can be used in a with statement:
In [1251]: import pandas as pd In [1252]: plt.figure() Out[1252]: <matplotlib.figure.Figure at 0x118f9c50> In [1253]: with pd.plot_params.use(x_compat, True): ......: df.A.plot(color=r) ......: df.B.plot(color=g) ......: df.C.plot(color=b) ......:
233
234
Calling a DataFrames plot method with kind=bar produces a multiple bar plot:
In [1260]: df2 = DataFrame(np.random.rand(10, 4), columns=[a, b, c, d]) In [1261]: df2.plot(kind=bar);
235
236
14.2.2 Histograms
In [1261]: plt.figure(); In [1261]: df[A].diff().hist() Out[1261]: <matplotlib.axes.AxesSubplot at 0x139231d0>
For a DataFrame, hist plots the histograms of the columns on multiple subplots:
In [1262]: plt.figure() Out[1262]: <matplotlib.figure.Figure at 0x14086fd0> In [1263]: df.diff().hist(color=k, alpha=0.5, bins=50) Out[1263]: array([[Axes(0.125,0.552174;0.336957x0.347826), Axes(0.563043,0.552174;0.336957x0.347826)], [Axes(0.125,0.1;0.336957x0.347826), Axes(0.563043,0.1;0.336957x0.347826)]], dtype=object)
237
New since 0.9.2, the by keyword can be specied to plot grouped histograms:
In [1264]: data = Series(np.random.randn(1000)) In [1265]: data.hist(by=np.random.randint(0, 4, 1000)) Out[1265]: array([[Axes(0.1,0.6;0.347826x0.3), Axes(0.552174,0.6;0.347826x0.3)], [Axes(0.1,0.15;0.347826x0.3), Axes(0.552174,0.15;0.347826x0.3)]], dtype=object)
14.2.3 Box-Plotting
DataFrame has a boxplot method which allows you to visualize the distribution of values within each column. For instance, here is a boxplot representing ve trials of 10 observations of a uniform random variable on [0,1).
In [1266]: df = DataFrame(np.random.rand(10,5))
238
You can create a stratied boxplot using the by keyword argument to create groupings. For instance,
In [1268]: df = DataFrame(np.random.rand(10,2), columns=[Col1, Col2] ) In [1269]: df[X] = Series([A,A,A,A,A,B,B,B,B,B]) In [1270]: plt.figure(); In [1270]: bp = df.boxplot(by=X)
You can also pass a subset of columns to plot, as well as group by multiple columns:
239
In [1271]: df = DataFrame(np.random.rand(10,3), columns=[Col1, Col2, Col3]) In [1272]: df[X] = Series([A,A,A,A,A,B,B,B,B,B]) In [1273]: df[Y] = Series([A,B,A,B,A,B,A,B,A,B]) In [1274]: plt.figure(); In [1274]: bp = df.boxplot(column=[Col1,Col2], by=[X,Y])
method
in
240
New in 0.8.0 You can create density plots using the Series/DataFrame.plot and setting kind=kde:
In [1278]: ser = Series(np.random.randn(1000)) In [1279]: ser.plot(kind=kde) Out[1279]: <matplotlib.axes.AxesSubplot at 0x180ce510>
241
242
243
244
245
246
14.2.10 RadViz
RadViz is a way of visualizing multi-variate data. It is based on a simple spring tension minimization algorithm. Basically you set up a bunch of points in a plane. In our case they are equally spaced on a unit circle. Each point represents a single attribute. You then pretend that each sample in the data set is attached to each of these points by a spring, the stiffness of which is proportional to the numerical value of that attribute (they are normalized to unit interval). The point in the plane, where our sample settles to (where the forces acting on our sample are at an equilibrium) is where a dot representing our sample will be drawn. Depending on which class that sample belongs it will be colored differently. Note: The Iris dataset is available here.
In [1301]: from pandas import read_csv In [1302]: from pandas.tools.plotting import radviz
247
In [1303]: data = read_csv(data/iris.data) In [1304]: plt.figure() Out[1304]: <matplotlib.figure.Figure at 0x195fa5d0> In [1305]: radviz(data, Name) Out[1305]: <matplotlib.axes.AxesSubplot at 0x195fa450>
248
CHAPTER
FIFTEEN
249
index_col: column number, column name, or list of column numbers/names, to use as the index (row labels) of the resulting DataFrame. By default, it will number the rows without using any column, unless there is one more data column than there are headers, in which case the rst column is taken as the index. names: List of column names to use. If passed, header will be implicitly set to None. na_values: optional list of strings to recognize as NaN (missing values), either in addition to or in lieu of the default set. keep_default_na: whether to include the default set of missing values in addition to the ones specied in na_values parse_dates: if True then index will be parsed as dates (False by default). You can specify more complicated options to parse a subset of columns or a combination of columns into a single date column (list of ints or names, list of lists, or dict) [1, 2, 3] -> try parsing columns 1, 2, 3 each as a separate date column [[1, 3]] -> combine columns 1 and 3 and parse as a single date column {foo : [1, 3]} -> parse columns 1, 3 as date and call result foo keep_date_col: if True, then date component columns passed into parse_dates will be retained in the output (False by default). date_parser: function to use to parse strings into datetime objects. If parse_dates is True, it defaults to the very robust dateutil.parser. Specifying this implicitly sets parse_dates as True. You can also use functions from community supported date converters from date_converters.py dayfirst: if True then uses the DD/MM international/European date format (This is False by default) thousands: sepcies the thousands separator. If not None, then parser will try to look for it in the output and parse relevant data to integers. Because it has to essentially scan through the data again, this causes a signicant performance hit so only use if necessary. comment: denotes the start of a comment and ignores the rest of the line. Currently line commenting is not supported. nrows: Number of rows to read out of the le. Useful to only read a small portion of a large le iterator: If True, return a TextParser to enable reading a le into memory piece by piece chunksize: An number of rows to be used to chunk a le into pieces. Will cause an TextParser object to be returned. More on this below in the section on iterating and chunking skip_footer: number of lines to skip at bottom of le (default 0) converters: a dictionary of functions for converting values in certain columns, where keys are either integers or column labels encoding: a string representing the encoding to use if the contents are non-ascii verbose: show number of NA values inserted in non-numeric columns squeeze: if True then output with only one column is turned into Series Consider a typical CSV le containing, in this case, some time series data:
In [749]: print open(foo.csv).read() date,A,B,C 20090101,a,1,2 20090102,b,3,4 20090103,c,4,5
The default for read_csv is to create a DataFrame with simple numbered rows:
250
In the case of indexed data, you can pass the column number or column name you wish to use as the index:
In [751]: read_csv(foo.csv, index_col=0) Out[751]: A B C date 20090101 a 1 2 20090102 b 3 4 20090103 c 4 5 In [752]: read_csv(foo.csv, index_col=date) Out[752]: A B C date 20090101 a 1 2 20090102 b 3 4 20090103 c 4 5
The dialect keyword gives greater exibility in specifying the le format. By default it uses the Excel dialect but you can specify either the dialect name or a csv.Dialect instance. Suppose you had data with unenclosed quotes:
In [754]: print data label1,label2,label3 index1,"a,c,e index2,b,d,f
By default, read_csv uses the Excel dialect and treats the double quote as the quote character, which causes it to fail when it nds a newline before it nds the closing double quote. We can get around this using dialect
In [755]: dia = csv.excel() In [756]: dia.quoting = csv.QUOTE_NONE In [757]: read_csv(StringIO(data), dialect=dia) Out[757]: label1 label2 label3 index1 "a c e index2 b d f
251
The parsers make every attempt to do the right thing and not be very fragile. Type inference is a pretty big deal. So if a column can be coerced to integer dtype without altering the contents, it will do so. Any non-numeric columns will come through as object dtype as with the rest of pandas objects.
# These are python datetime objects In [760]: df.index Out[760]: <class pandas.tseries.index.DatetimeIndex> [2009-01-01 00:00:00, ..., 2009-01-03 00:00:00] Length: 3, Freq: None, Timezone: None
It is often the case that we may want to store date and time data separately, or store various date elds separately. the parse_dates keyword can be used to specify a combination of columns to parse the dates and/or times from. You can specify a list of column lists to parse_dates, the resulting date columns will be prepended to the output (so as to not affect the existing column order) and the new column names will be the concatenation of the component column names:
In [761]: print open(tmp.csv).read() KORD,19990127, 19:00:00, 18:56:00, 0.8100 KORD,19990127, 20:00:00, 19:56:00, 0.0100 KORD,19990127, 21:00:00, 20:56:00, -0.5900 KORD,19990127, 21:00:00, 21:18:00, -0.9900 KORD,19990127, 22:00:00, 21:56:00, -0.5900 KORD,19990127, 23:00:00, 22:56:00, -0.5900 In [762]: df = read_csv(tmp.csv, header=None, parse_dates=[[1, 2], [1, 3]]) In [763]: df Out[763]: 0 1 2 3 4 5 1999-01-27 1999-01-27 1999-01-27 1999-01-27 1999-01-27 1999-01-27 X1_X2 19:00:00 20:00:00 21:00:00 21:00:00 22:00:00 23:00:00 1999-01-27 1999-01-27 1999-01-27 1999-01-27 1999-01-27 1999-01-27 X1_X3 18:56:00 19:56:00 20:56:00 21:18:00 21:56:00 22:56:00 X0 KORD KORD KORD KORD KORD KORD X4 0.81 0.01 -0.59 -0.99 -0.59 -0.59
252
By default the parser removes the component date columns, but you can choose to retain them via the keep_date_col keyword:
In [764]: df = read_csv(tmp.csv, header=None, parse_dates=[[1, 2], [1, 3]], .....: keep_date_col=True) .....: In [765]: df Out[765]: <class pandas.core.frame.DataFrame> Int64Index: 6 entries, 0 to 5 Data columns: X1_X2 6 non-null values X1_X3 6 non-null values X0 6 non-null values X1 6 non-null values X2 6 non-null values X3 6 non-null values X4 6 non-null values dtypes: float64(1), object(6)
Note that if you wish to combine multiple columns into a single date column, a nested list must be used. In other words, parse_dates=[1, 2] indicates that the second and third columns should each be parsed as separate date columns while parse_dates=[[1, 2]] means the two columns should be parsed into a single column. You can also use a dict to specify custom name columns:
In [766]: date_spec = {nominal: [1, 2], actual: [1, 3]} In [767]: df = read_csv(tmp.csv, header=None, parse_dates=date_spec) In [768]: df Out[768]: 0 1 2 3 4 5 1999-01-27 1999-01-27 1999-01-27 1999-01-27 1999-01-27 1999-01-27 nominal 19:00:00 20:00:00 21:00:00 21:00:00 22:00:00 23:00:00 1999-01-27 1999-01-27 1999-01-27 1999-01-27 1999-01-27 1999-01-27 actual 18:56:00 19:56:00 20:56:00 21:18:00 21:56:00 22:56:00 X0 KORD KORD KORD KORD KORD KORD X4 0.81 0.01 -0.59 -0.99 -0.59 -0.59
It is important to remember that if multiple text columns are to be parsed into a single date column, then a new column is prepended to the data. The index_col specication is based off of this new set of columns rather than the original data columns:
In [769]: date_spec = {nominal: [1, 2], actual: [1, 3]} In [770]: df = read_csv(tmp.csv, header=None, parse_dates=date_spec, .....: index_col=0) #index is the nominal column .....: In [771]: df Out[771]: actual nominal 1999-01-27 1999-01-27 1999-01-27 1999-01-27 19:00:00 20:00:00 21:00:00 21:00:00 1999-01-27 1999-01-27 1999-01-27 1999-01-27 18:56:00 19:56:00 20:56:00 21:18:00 X0 X4
253
Note: When passing a dict as the parse_dates argument, the order of the columns prepended is not guaranteed, because dict objects do not impose an ordering on their keys. On Python 2.7+ you may use collections.OrderedDict instead of a regular dict if this matters to you. Because of this, when using a dict for parse_dates in conjunction with the index_col argument, its best to specify index_col as a column label rather then as an index on the resulting frame.
You can explore the date parsing functionality in date_converters.py and add your own. We would love to turn this module into a community supported set of date/time parsers. To get you started, date_converters.py contains functions to parse dual date and time columns, year/month/day columns, and year/month/day/hour/minute/second columns. It also contains a generic_parser function so you can curry it with a function that deals with a single date rather than the entire array.
254
0 1 2
5 10 15
a b c
15.2.5 Comments
Sometimes comments or meta data may be included in a le:
In [786]: print open(tmp.csv).read() ID,level,category Patient1,123000,x # really unpleasant
255
256
In order to parse this le into a DataFrame, we simply need to supply the column specications to the read_fwf function along with the le name:
#Column specifications are a list of half-intervals In [796]: colspecs = [(0, 6), (8, 20), (21, 33), (34, 43)] In [797]: df = read_fwf(bar.csv, colspecs=colspecs, header=None, index_col=0) In [798]: df Out[798]: X1 X0 id8141 id1594 id1849 id1230 id1948 360.242940 444.953632 364.136849 413.836124 502.953953 X2 149.910199 166.985655 183.628767 184.375703 173.237159 X3 11950.7 11788.4 11806.2 11916.8 12468.3
Note how the parser automatically picks column names X.<column number> when header=None argument is specied. Alternatively, you can supply just the column widths for contiguous columns:
#Widths are a list of integers In [799]: widths = [6, 14, 13, 10] In [800]: df = read_fwf(bar.csv, widths=widths, header=None) In [801]: df Out[801]: X0 X1 0 id8141 360.242940 1 id1594 444.953632 2 id1849 364.136849 3 id1230 413.836124 4 id1948 502.953953
The parser will take care of extra white spaces around the columns so its ok to have extra separation between the columns in the le.
In this special case, read_csv assumes that the rst column is to be used as the index of the DataFrame:
257
Note that the dates werent automatically parsed. In that case you would need to do as before:
In [804]: df = read_csv(foo.csv, parse_dates=True) In [805]: df.index Out[805]: <class pandas.tseries.index.DatetimeIndex> [2009-01-01 00:00:00, ..., 2009-01-03 00:00:00] Length: 3, Freq: None, Timezone: None
The index_col argument to read_csv and read_table can take a list of column numbers to turn multiple columns into a MultiIndex:
In [807]: df = read_csv("data/mindex_ex.csv", index_col=[0,1]) In [808]: df Out[808]: zit year indiv 1977 A B C 1978 A B C D E 1979 C 1.20 1.50 1.70 0.20 0.70 0.80 0.90 1.40 0.20 xit 0.60 0.50 0.80 0.06 0.20 0.30 0.50 0.90 0.15
258
D E F G H I
In [809]: df.ix[1978] Out[809]: zit xit indiv A 0.2 0.06 B 0.7 0.20 C 0.8 0.30 D 0.9 0.50 E 1.4 0.90
259
In [812]: print open(tmp.sv).read() |0|1|2|3 0|0.46911229990718628|-0.28286334432866328|-1.5090585031735124|-1.1356323710171934 1|1.2121120250208506|-0.17321464905330858|0.11920871129693428|-1.0442359662799567 2|-0.86184896334779992|-2.1045692188948086|-0.49492927406878129|1.0718038070373379 3|0.72155516224436689|-0.70677113363008448|-1.0395749851146963|0.27185988554282986 4|-0.42497232978883753|0.567020349793672|0.27623201927771873|-1.0874006912859915 5|-0.67368970808837059|0.1136484096888855|-1.4784265524372235|0.52498766711470468 6|0.40470521868023651|0.57704598592048362|-1.7150020161146375|-1.0392684835147725 7|-0.37064685823644639|-1.1578922506419993|-1.3443118127316671|0.84488514142488413 8|1.0757697837155533|-0.10904997528022223|1.6435630703622064|-1.4693879595399115 9|0.35702056413309086|-0.67460010372998824|-1.7769037169718671|-0.96891381244734975 In [813]: table = read_table(tmp.sv, sep=|) In [814]: table Out[814]: Unnamed: 0 0 0 0 0.469112 1 1 1.212112 2 2 -0.861849 3 3 0.721555 4 4 -0.424972 5 5 -0.673690 6 6 0.404705 7 7 -0.370647 8 8 1.075770 9 9 0.357021
1 -0.282863 -0.173215 -2.104569 -0.706771 0.567020 0.113648 0.577046 -1.157892 -0.109050 -0.674600
2 -1.509059 0.119209 -0.494929 -1.039575 0.276232 -1.478427 -1.715002 -1.344312 1.643563 -1.776904
3 -1.135632 -1.044236 1.071804 0.271860 -1.087401 0.524988 -1.039268 0.844885 -1.469388 -0.968914
By speciying a chunksize to read_csv or read_table, the return value will be an iterable object of type TextParser:
In [815]: reader = read_table(tmp.sv, sep=|, chunksize=4) In [816]: reader Out[816]: <pandas.io.parsers.TextFileReader at 0xaff33d0> In [817]: for .....: .....: Unnamed: 0 0 0 1 1 2 2 3 3 Unnamed: 0 0 4 1 5 2 6 3 7 Unnamed: 0 0 8 1 9 chunk in reader: print chunk 0 0.469112 1.212112 -0.861849 0.721555 0 -0.424972 -0.673690 0.404705 -0.370647 0 1.075770 0.357021 1 2 3 -0.282863 -1.509059 -1.135632 -0.173215 0.119209 -1.044236 -2.104569 -0.494929 1.071804 -0.706771 -1.039575 0.271860 1 2 3 0.567020 0.276232 -1.087401 0.113648 -1.478427 0.524988 0.577046 -1.715002 -1.039268 -1.157892 -1.344312 0.844885 1 2 3 -0.10905 1.643563 -1.469388 -0.67460 -1.776904 -0.968914
260
Out[819]: Unnamed: 0 1 2 3 4
0 0 1 2 3 0 0.469112 -0.282863 -1.509059 -1.135632 1 1.212112 -0.173215 0.119209 -1.044236 2 -0.861849 -2.104569 -0.494929 1.071804 3 0.721555 -0.706771 -1.039575 0.271860 4 -0.424972 0.567020 0.276232 -1.087401
261
The Series object also has a to_string method, but with only the buf, na_rep, float_format arguments. There is also a length argument which, if set to True, will additionally output the length of the Series.
Then use the parse instance method with a sheetname, then use the same additional arguments as the parsers above:
xls.parse(Sheet1, index_col=None, na_values=[NA])
To read sheets from an Excel 2007 le, you can pass a lename with a .xlsx extension, in which case the openpyxl module will be used to read the le. It is often the case that users will insert columns to do temporary computations in Excel and you may not want to read in those columns. ExcelFile.parse takes a parse_cols keyword to allow you to specify a subset of columns to parse. If parse_cols is an integer, then it is assumed to indicate the last column to be parsed.
xls.parse(Sheet1, parse_cols=2, index_col=None, na_values=[NA])
To write a DataFrame object to a sheet of an Excel le, you can use the to_excel instance method. The arguments are largely the same as to_csv described above, the rst argument being the name of the excel le, and the optional second argument the name of the sheet to which the DataFrame should be written. For example:
df.to_excel(path_to_file.xlsx, sheet_name=sheet1)
Files with a .xls extension will be written using xlwt and those with a .xlsx extension will be written using openpyxl. The Panel class also has a to_excel instance method, which writes each DataFrame in the Panel to a separate sheet. In order to write separate DataFrames to separate sheets in a single Excel le, one can use the ExcelWriter class, as in the following example:
writer = ExcelWriter(path_to_file.xlsx) df1.to_excel(writer, sheet_name=sheet1) df2.to_excel(writer, sheet_name=sheet2) writer.save()
262
In [820]: store = HDFStore(store.h5) In [821]: print store <class pandas.io.pytables.HDFStore> File path: store.h5 Empty
Objects can be written to the le just like adding key-value pairs to a dict:
In [822]: index = date_range(1/1/2000, periods=8) In [823]: s = Series(randn(5), index=[a, b, c, d, e]) In [824]: df = DataFrame(randn(8, 3), index=index, .....: columns=[A, B, C]) .....: In [825]: wp = Panel(randn(2, 5, 4), items=[Item1, Item2], .....: major_axis=date_range(1/1/2000, periods=5), .....: minor_axis=[A, B, C, D]) .....: In [826]: store[s] = s In [827]: store[df] = df In [828]: store[wp] = wp In [829]: store Out[829]: <class pandas.io.pytables.HDFStore> File path: store.h5 df DataFrame s Series wp Panel
263
264
CHAPTER
SIXTEEN
The to_sparse method takes a kind argument (for the sparse index, see below) and a fill_value. So if we had a mostly zero Series, we could convert it to sparse with fill_value=0:
In [1056]: ts.fillna(0).to_sparse(fill_value=0) Out[1056]: 0 0.469112 1 -0.282863 2 0.000000 3 0.000000 4 0.000000 5 0.000000 6 0.000000 7 0.000000 8 -0.861849 9 -2.104569 BlockIndex
265
Block locations: array([0, 8], dtype=int32) Block lengths: array([2, 2], dtype=int32)
The sparse objects exist for memory efciency reasons. Suppose you had a large, mostly NA DataFrame:
In [1057]: df = DataFrame(randn(10000, 4)) In [1058]: df.ix[:9998] = np.nan In [1059]: sdf = df.to_sparse() In [1060]: sdf Out[1060]: <class pandas.sparse.frame.SparseDataFrame> Int64Index: 10000 entries, 0 to 9999 Columns: 4 entries, 0 to 3 dtypes: float64(4) In [1061]: sdf.density Out[1061]: 0.0001
As you can see, the density (% of values that have not been compressed) is extremely low. This sparse object takes up much less memory on disk (pickled) and in the Python interpreter. Functionally, their behavior should be nearly identical to their dense counterparts. Any sparse object can be converted back to the standard dense form by calling to_dense:
In [1062]: sts.to_dense() Out[1062]: 0 0.469112 1 -0.282863 2 NaN 3 NaN 4 NaN 5 NaN 6 NaN 7 NaN 8 -0.861849 9 -2.104569
16.1 SparseArray
SparseArray is the base layer for all of the sparse indexed data structures. It is a 1-dimensional ndarray-like object storing only values distinct from the fill_value:
In [1063]: arr = np.random.randn(10) In [1064]: arr[2:5] = np.nan; arr[7:8] = np.nan In [1065]: sparr = SparseArray(arr) In [1066]: sparr Out[1066]: SparseArray([-1.9557, -1.6589, nan, nan, nan, 0.606 , 1.3342]) IntIndex Indices: array([0, 1, 5, 6, 8, 9], dtype=int32)
nan,
1.1589,
0.1453,
266
Like the indexed objects (SparseSeries, SparseDataFrame, SparsePanel), a SparseArray can be converted back to a regular ndarray by calling to_dense:
In [1067]: sparr.to_dense() Out[1067]: array([-1.9557, -1.6589, nan, nan, 0.606 , 1.3342])
nan,
nan,
1.1589,
0.1453,
16.2 SparseList
SparseList is a list-like data structure for managing a dynamic collection of SparseArrays. To create one, simply call the SparseList constructor with a fill_value (defaulting to NaN):
In [1068]: spl = SparseList() In [1069]: spl Out[1069]: <pandas.sparse.list.SparseList object at 0xb2c70d0>
The two important methods are append and to_array. append can accept scalar values or any 1-dimensional sequence:
In [1070]: spl.append(np.array([1., nan, nan, 2., 3.])) In [1071]: spl.append(5) In [1072]: spl.append(sparr) In [1073]: spl Out[1073]: <pandas.sparse.list.SparseList object at 0xb2c70d0> SparseArray([ 1., nan, nan, 2., 3.]) IntIndex Indices: array([0, 3, 4], dtype=int32) SparseArray([ 5.]) IntIndex Indices: array([0], dtype=int32) SparseArray([-1.9557, -1.6589, nan, nan, nan, 0.606 , 1.3342]) IntIndex Indices: array([0, 1, 5, 6, 8, 9], dtype=int32)
nan,
1.1589,
0.1453,
As you can see, all of the contents are stored internally as a list of memory-efcient SparseArray objects. Once youve accumulated all of the data, you can call to_array to get a single SparseArray with all the data:
In [1074]: spl.to_array() Out[1074]: SparseArray([ 1. , nan, nan, 2. , 3. , 5. , -1.9557, -1.6589, nan, nan, nan, 1.1589, 0.1453, nan, 0.606 , 1.3342]) IntIndex Indices: array([ 0, 3, 4, 5, 6, 7, 11, 12, 14, 15], dtype=int32)
16.2. SparseList
267
268
CHAPTER
SEVENTEEN
269
c f u
3 NaN NaN
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.
While this may seem like a heavy trade-off, in practice I have found very few cases where this is an issue in practice. Some explanation for the motivation here in the next section.
The R language, by contrast, only has a handful of built-in data types: integer, numeric (oating-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 oat and object arrays to denote NA, and promoting integer arrays to oating when NAs must be introduced.
270
This deliberate decision was made to prevent ambiguities and subtle bugs (many users reported nding bugs when the API change was made to stop falling back on position-based indexing).
However, if you only had c and e, determining the next element in the index can be somewhat complicated. For example, the following does not work:
s.ix[c:e+1]
A very common use case is to limit a time series to start and end at two specic dates. To enable this, we made the design design to make label-based slicing include both endpoints:
In [433]: s.ix[c:e] Out[433]: c 1.331458
271
d e
-0.571329 -0.026671
This is most denitely a practicality beats purity sort of thing, but it is something to watch out for if you expect label-based slicing to behave exactly in the way that standard Python integer slicing works.
This is, of course, completely equivalent in this case to using th reindex method:
In [437]: df.reindex([b, c, e]) Out[437]: one two three four b 0.112572 -1.495309 0.898435 -0.148217 c -1.596070 0.159653 0.262136 0.036220 e 0.294633 -1.165787 0.846974 -0.685597
Some might conclude that ix and reindex are 100% equivalent based on this. This is indeed true except in the case of integer indexing. For example, the above operation could alternately have been expressed as:
In [438]: df.ix[[1, 2, Out[438]: one two b 0.112572 -1.495309 c -1.596070 0.159653 e 0.294633 -1.165787 4]] three four 0.898435 -0.148217 0.262136 0.036220 0.846974 -0.685597
If you pass [1, 2, 4] to reindex you will get another thing entirely:
In [439]: df.reindex([1, 2, 4]) Out[439]:
272
1 2 4
So its important to remember that reindex is strict label indexing only. This can lead to some potentially surprising results in pathological cases where an index contains, say, both integers and strings:
In [440]: s = Series([1, 2, 3], index=[a, 0, 1]) In [441]: s Out[441]: a 1 0 2 1 3 In [442]: s.ix[[0, 1]] Out[442]: 0 2 1 3 In [443]: s.reindex([0, 1]) Out[443]: 0 2 1 3
Because the index in this case does not contain solely integers, ix falls back on integer indexing. By contrast, reindex only looks for the values passed in the index, thus nding the integers 0 and 1. While it would be possible to insert some logic to check whether a passed sequence is all contained in the index, that logic would exact a very high cost in large data sets.
If you need to represent time series data outside the nanosecond timespan, use PeriodIndex:
In [448]: span = period_range(1215-01-01, 1381-01-01, freq=D) In [449]: span Out[449]: <class pandas.tseries.period.PeriodIndex> freq: D
273
274
CHAPTER
EIGHTEEN
RPY2 / R INTERFACE
Note: This is all highly experimental. I would like to get more people involved with building a nice RPy2 interface for pandas If your computer has R and rpy2 (> 2.2) installed (which will be left to the reader), you will be able to leverage the below functionality. On Windows, doing this is quite an ordeal at the moment, but users on Unix-like systems should nd it quite easy. rpy2 evolves in time and the current interface is designed for the 2.2.x series, and we recommend to use over other series unless you are prepared to x parts of the code. Released packages are available in PyPi, but should the latest code in the 2.2.x series be wanted it can be obtained with:
# if installing for the first time hg clone http://bitbucket.org/lgautier/rpy2 cd rpy2 hg pull hg update version_2.2.x sudo python setup.py install
Note: To use R packages with this interface, you will need to install them inside R yourself. At the moment it cannot install them for you. Once you have done installed R and rpy2, you should be able to import pandas.rpy.common without a hitch.
induced 1 1 2
case 1 1 1
spontaneous 2 0 0
stratum 1 2 3
pooled.stratum 3 1 4
275
4 5
0-5yrs 6-11yrs
34 35
4 3
2 1
1 1
0 1
4 5
2 32
The DataFrames index is stored as the rownames attribute of the data.frame instance. You can also use convert_to_r_matrix to obtain a Matrix instance, but bear in mind that it will only work with homogeneously-typed DataFrames (as R matrices bear no information on the data type):
In [998]: r_matrix = com.convert_to_r_matrix(df) In [999]: print type(r_matrix) <class rpy2.robjects.vectors.Matrix> In [1000]: print r_matrix A B C one 1 4 7 two 2 5 8 three 3 6 9
18.3 Calling R functions with pandas objects 18.4 High-level interface to R estimators
276
CHAPTER
NINETEEN
19.2 statsmodels
The main statistics and econometrics library for Python. pandas has become a dependency of this library.
19.3 scikits.timeseries
scikits.timeseries provides a data structure for xed frequency time series data based on the numpy.MaskedArray class. For time series data, it provides some of the same functionality to the pandas Series class. It has many more functions for time series-specic manipulation. Also, it has support for many more frequencies, though less customizable by the user (so 5-minutely data is easier to do with pandas for example). We are aiming to merge these libraries together in the near future. Progress: It has a collection of moving window statistics implemented in Bottleneck Outstanding issues
277
Summarising, Pandas offers superior functionality due to its combination with the pandas.DataFrame. An introduction for former users of scikits.timeseries is provided in the migration guide.
278
CHAPTER
TWENTY
20.1 data.frame 20.2 zoo 20.3 xts 20.4 plyr 20.5 reshape / reshape2
279
280
CHAPTER
TWENTYONE
API REFERENCE
21.1 General functions
21.1.1 Data manipulations
pivot_table(data[, values, rows, cols, ...]) Create a spreadsheet-style pivot table as a DataFrame. The levels in the
pandas.tools.pivot.pivot_table pandas.tools.pivot.pivot_table(data, values=None, rows=None, cols=None, aggfunc=mean, ll_value=None, margins=False) Create a spreadsheet-style pivot table as a DataFrame. The levels in the pivot table will be stored in MultiIndex objects (hierarchical indexes) on the index and columns of the result DataFrame Parameters data : DataFrame values : column to aggregate, optional rows : list of column names or arrays to group on Keys to group on the x-axis of the pivot table cols : list of column names or arrays to group on Keys to group on the y-axis of the pivot table aggfunc : function, default numpy.mean, or list of functions If list of functions passed, the resulting pivot table will have hierarchical columns whose top level are the function names (inferred from the function objects themselves) ll_value : scalar, default None Value to replace missing values with margins : boolean, default False Add all row / columns (e.g. for subtotal / grand totals) Returns table : DataFrame
281
Examples >>> df A 0 foo 1 foo 2 foo 3 foo 4 foo 5 bar 6 bar 7 bar 8 bar
D 1 2 2 3 3 4 5 6 7
>>> table = pivot_table(df, values=D, rows=[A, B], ... cols=[C], aggfunc=np.sum) >>> table small large foo one 1 4 two 6 NaN bar one 5 4 two 6 7
merge(left, right[, how, on, left_on, ...]) concat(objs[, axis, join, join_axes, ...])
Merge DataFrame objects by performing a database-style join operation by Concatenate pandas objects along a particular axis with optional set logic along the other a
pandas.tools.merge.merge pandas.tools.merge.merge(left, right, how=inner, on=None, left_on=None, right_on=None, left_index=False, right_index=False, sort=True, sufxes=(_x, _y), copy=True) Merge DataFrame objects by performing a database-style join operation by columns or indexes. If joining columns on columns, the DataFrame indexes will be ignored. Otherwise if joining indexes on indexes or indexes on a column or columns, the index will be passed on. Parameters left : DataFrame right : DataFrame how : {left, right, outer, inner}, default inner left: use only keys from left frame (SQL: left outer join) right: use only keys from right frame (SQL: right outer join) outer: use union of keys from both frames (SQL: full outer join) inner: use intersection of keys from both frames (SQL: inner join) on : label or list Field names to join on. Must be found in both DataFrames. If on is None and not merging on indexes, then it merges on the intersection of the columns by default. left_on : label or list, or array-like Field names to join on in left DataFrame. Can be a vector or list of vectors of the length of the DataFrame to use a particular vector as the join key instead of columns
282
right_on : label or list, or array-like Field names to join on in right DataFrame or vector/list of vectors per left_on docs left_index : boolean, default False Use the index from the left DataFrame as the join key(s). If it is a MultiIndex, the number of keys in the other DataFrame (either the index or a number of columns) must match the number of levels right_index : boolean, default False Use the index from the right DataFrame as the join key. Same caveats as left_index sort : boolean, default True Sort the join keys lexicographically in the result DataFrame sufxes : 2-length sequence (tuple, list, ...) Sufx to apply to overlapping column names in the left and right side, respectively copy : boolean, default True If False, do not copy data unnecessarily Returns merged : DataFrame
Examples >>> A lkey 0 foo 1 bar 2 baz 3 foo >>> B rkey 0 foo 1 bar 2 qux 3 bar
value 1 2 3 4
value 5 6 7 8
>>> merge(A, B, left_on=lkey, right_on=rkey, how=outer) lkey value_x rkey value_y 0 bar 2 bar 6 1 bar 2 bar 8 2 baz 3 NaN NaN 3 foo 1 foo 5 4 foo 4 foo 5 5 NaN NaN qux 7
pandas.tools.merge.concat pandas.tools.merge.concat(objs, axis=0, join=outer, join_axes=None, ignore_index=False, keys=None, levels=None, names=None, verify_integrity=False) Concatenate pandas objects along a particular axis with optional set logic along the other axes. Can also add a layer of hierarchical indexing on the concatenation axis, which may be useful if the labels are the same (or overlapping) on the passed axis number Parameters objs : list or dict of Series, DataFrame, or Panel objects If a dict is passed, the sorted keys will be used as the keys argument, unless it is passed, in which case the values will be selected (see below). Any None objects will be dropped silently unless they are all None in which case an Exception will be raised axis : {0, 1, ...}, default 0 21.1. General functions 283
The axis to concatenate along join : {inner, outer}, default outer How to handle indexes on other axis(es) join_axes : list of Index objects Specic indexes to use for the other n - 1 axes instead of performing inner/outer set logic verify_integrity : boolean, default False Check whether the new concatenated axis contains duplicates. This can be very expensive relative to the actual data concatenation keys : sequence, default None If multiple levels passed, should contain tuples. Construct hierarchical index using the passed keys as the outermost level levels : list of sequences, default None Specic levels (unique values) to use for constructing a MultiIndex. Otherwise they will be inferred from the keys names : list, default None Names for the levels in the resulting hierarchical index ignore_index : boolean, default False If True, do not use the index values along the concatenation axis. The resulting axis will be labeled 0, ..., n - 1. This is useful if you are concatenating objects where the concatenation axis does not have meaningful indexing information. Note the the index values on the other axes are still respected in the join. Returns concatenated : type of objects
Notes
21.1.2 Pickling
load(path) save(obj, path) Load pickled pandas object (or any other pickled object) from the specied Pickle (serialize) object to input le path
pandas.core.common.load pandas.core.common.load(path) Load pickled pandas object (or any other pickled object) from the specied le path Parameters path : string File path Returns unpickled : type of object stored in le
284
pandas.core.common.save pandas.core.common.save(obj, path) Pickle (serialize) object to input le path Parameters obj : any object path : string File path
21.1.3 File IO
read_table(lepath_or_buffer[, sep, ...]) read_csv(lepath_or_buffer[, sep, dialect, ...]) ExcelFile.parse(sheetname[, header, ...]) Read general delimited le into DataFrame Read CSV (comma-separated) le into DataFrame Read Excel table into DataFrame
pandas.io.parsers.read_table pandas.io.parsers.read_table(lepath_or_buffer, sep=\t, dialect=None, compression=None, doublequote=True, escapechar=None, quotechar=, quoting=0, skipinitialspace=False, header=0, index_col=None, names=None, skiprows=None, skipfooter=None, skip_footer=0, na_values=None, delimiter=None, converters=None, dtype=None, usecols=None, engine=c, delim_whitespace=False, as_recarray=False, na_lter=True, compact_ints=False, use_unsigned=False, low_memory=True, buffer_lines=None, warn_bad_lines=True, error_bad_lines=True, keep_default_na=True, thousands=None, comment=None, parse_dates=False, keep_date_col=False, dayrst=False, date_parser=None, memory_map=False, nrows=None, iterator=False, chunksize=None, verbose=False, encoding=None, squeeze=False) Read general delimited le into DataFrame Also supports optionally iterating or breaking of the le into chunks. Parameters lepath_or_buffer : string or le handle / StringIO. The string could be a URL. Valid URL schemes include http, ftp, and le. For le URLs, a host is expected. For instance, a local le could be le ://localhost/path/to/table.csv sep : string, default t (tab-stop) Delimiter to use. Regular expressions are accepted. compression : {gzip, bz2, None}, default None For on-the-y decompression of on-disk data dialect : string or csv.Dialect instance, default None If None defaults to Excel dialect. Ignored if sep longer than 1 char See csv.Dialect documentation for more details header : int, default 0 Row to use for the column labels of the parsed DataFrame. Specify None if there is no header row. 21.1. General functions 285
skiprows : list-like or integer Row numbers to skip (0-indexed) or number of rows to skip (int) at the start of the le index_col : int or sequence, default None Column to use as the row labels of the DataFrame. If a sequence is given, a MultiIndex is used. names : array-like List of column names to use. If header=True, replace existing header, otherwise this ignores the header=0 default na_values : list-like or dict, default None Additional strings to recognize as NA/NaN. If dict passed, specic per-column NA values keep_default_na : bool, default True If na_values are specied and keep_default_na is False the default NaN values are overridden, otherwise theyre appended to parse_dates : boolean, list of ints or names, list of lists, or dict If True -> try parsing the index. If [1, 2, 3] -> try parsing columns 1, 2, 3 each as a separate date column. If [[1, 3]] -> combine columns 1 and 3 and parse as a single date column. {foo : [1, 3]} -> parse columns 1, 3 as date and call result foo keep_date_col : boolean, default False If True and parse_dates species combining multiple columns then keep the original columns. date_parser : function Function to use for converting dates to strings. Defaults to dateutil.parser dayrst : boolean, default False DD/MM format dates, international and European format thousands : str, default None Thousands separator comment : str, default None Indicates remainder of line should not be parsed Does not support line commenting (will return empty line) nrows : int, default None Number of rows of le to read. Useful for reading pieces of large les iterator : boolean, default False Return TextParser object chunksize : int, default None Return TextParser object for iteration skipfooter : int, default 0 Number of line at bottom of le to skip converters : dict. optional
286
Dict of functions for converting values in certain columns. Keys can either be integers or column labels verbose : boolean, default False Indicate number of NA values placed in non-numeric columns delimiter : string, default None Alternative argument name for sep. Regular expressions are accepted. encoding : string, default None Encoding to use for UTF when reading/writing (ex. utf-8) squeeze : boolean, default False If the parsed data only contains one column then return a Series Returns result : DataFrame or TextParser pandas.io.parsers.read_csv pandas.io.parsers.read_csv(lepath_or_buffer, sep=, , dialect=None, compression=None, doublequote=True, escapechar=None, quotechar=, quoting=0, skipinitialspace=False, header=0, index_col=None, names=None, skiprows=None, skipfooter=None, skip_footer=0, na_values=None, delimiter=None, converters=None, dtype=None, usecols=None, engine=c, delim_whitespace=False, as_recarray=False, na_lter=True, compact_ints=False, use_unsigned=False, low_memory=True, buffer_lines=None, warn_bad_lines=True, error_bad_lines=True, keep_default_na=True, thousands=None, comment=None, parse_dates=False, keep_date_col=False, dayrst=False, date_parser=None, memory_map=False, nrows=None, iterator=False, chunksize=None, verbose=False, encoding=None, squeeze=False) Read CSV (comma-separated) le into DataFrame Also supports optionally iterating or breaking of the le into chunks. Parameters lepath_or_buffer : string or le handle / StringIO. The string could be a URL. Valid URL schemes include http, ftp, and le. For le URLs, a host is expected. For instance, a local le could be le ://localhost/path/to/table.csv sep : string, default , Delimiter to use. If sep is None, will try to automatically determine this. Regular expressions are accepted. compression : {gzip, bz2, None}, default None For on-the-y decompression of on-disk data dialect : string or csv.Dialect instance, default None If None defaults to Excel dialect. Ignored if sep longer than 1 char See csv.Dialect documentation for more details header : int, default 0 Row to use for the column labels of the parsed DataFrame. Specify None if there is no header row.
287
skiprows : list-like or integer Row numbers to skip (0-indexed) or number of rows to skip (int) at the start of the le index_col : int or sequence, default None Column to use as the row labels of the DataFrame. If a sequence is given, a MultiIndex is used. names : array-like List of column names to use. If header=True, replace existing header, otherwise this ignores the header=0 default na_values : list-like or dict, default None Additional strings to recognize as NA/NaN. If dict passed, specic per-column NA values keep_default_na : bool, default True If na_values are specied and keep_default_na is False the default NaN values are overridden, otherwise theyre appended to parse_dates : boolean, list of ints or names, list of lists, or dict If True -> try parsing the index. If [1, 2, 3] -> try parsing columns 1, 2, 3 each as a separate date column. If [[1, 3]] -> combine columns 1 and 3 and parse as a single date column. {foo : [1, 3]} -> parse columns 1, 3 as date and call result foo keep_date_col : boolean, default False If True and parse_dates species combining multiple columns then keep the original columns. date_parser : function Function to use for converting dates to strings. Defaults to dateutil.parser dayrst : boolean, default False DD/MM format dates, international and European format thousands : str, default None Thousands separator comment : str, default None Indicates remainder of line should not be parsed Does not support line commenting (will return empty line) nrows : int, default None Number of rows of le to read. Useful for reading pieces of large les iterator : boolean, default False Return TextParser object chunksize : int, default None Return TextParser object for iteration skipfooter : int, default 0 Number of line at bottom of le to skip converters : dict. optional
288
Dict of functions for converting values in certain columns. Keys can either be integers or column labels verbose : boolean, default False Indicate number of NA values placed in non-numeric columns delimiter : string, default None Alternative argument name for sep. Regular expressions are accepted. encoding : string, default None Encoding to use for UTF when reading/writing (ex. utf-8) squeeze : boolean, default False If the parsed data only contains one column then return a Series Returns result : DataFrame or TextParser pandas.io.parsers.ExcelFile.parse ExcelFile.parse(sheetname, header=0, skiprows=None, skip_footer=0, index_col=None, parse_cols=None, parse_dates=False, date_parser=None, na_values=None, thousands=None, chunksize=None, **kwds) Read Excel table into DataFrame Parameters sheetname : string Name of Excel sheet header : int, default 0 Row to use for the column labels of the parsed DataFrame skiprows : list-like Rows to skip at the beginning (0-indexed) skip_footer : int, default 0 Rows at the end to skip (0-indexed) index_col : int, default None Column to use as the row labels of the DataFrame. Pass None if there is no such column parse_cols : int or list, default None If None then parse all columns, If int then indicates last column to be parsed If list of ints then indicates list of column numbers to be parsed If string then indicates comma separated list of column names and column ranges (e.g. A:E or A,C,E:F) na_values : list-like, default None List of additional strings to recognize as NA/NaN Returns parsed : DataFrame
289
pandas.io.pytables.HDFStore.put HDFStore.put(key, value, table=False, append=False, compression=None) Store object in HDFStore Parameters key : object value : {Series, DataFrame, Panel} table : boolean, default False Write as a PyTables Table structure which may perform worse but allow more exible operations like searching / selecting subsets of the data append : boolean, default False For table data structures, append the input data to the existing table compression : {None, blosc, lzo, zlib}, default None Use a compression algorithm to compress the data If None, the compression settings specied in the ctor will be used. pandas.io.pytables.HDFStore.get HDFStore.get(key) Retrieve pandas object stored in le Parameters key : object Returns obj : type of object stored in le
pandas.stats.moments.rolling_count pandas.stats.moments.rolling_count(arg, window, freq=None, time_rule=None) Rolling count of number of non-NaN observations inside provided window. 290 Chapter 21. API Reference
Parameters arg : DataFrame or numpy ndarray-like window : Number of observations used for calculating statistic freq : None or string alias / date offset object, default=None Frequency to conform to before computing statistic Returns rolling_count : type of caller pandas.stats.moments.rolling_sum pandas.stats.moments.rolling_sum(arg, window, min_periods=None, time_rule=None, **kwargs) Moving sum Parameters arg : Series, DataFrame window : Number of observations used for calculating statistic min_periods : int Minimum number of observations in window required to have a value freq : None or string alias / date offset object, default=None Frequency to conform to before computing statistic time_rule is a legacy alias for freq Returns y : type of input argument pandas.stats.moments.rolling_mean pandas.stats.moments.rolling_mean(arg, window, min_periods=None, time_rule=None, **kwargs) Moving mean Parameters arg : Series, DataFrame window : Number of observations used for calculating statistic min_periods : int Minimum number of observations in window required to have a value freq : None or string alias / date offset object, default=None Frequency to conform to before computing statistic time_rule is a legacy alias for freq Returns y : type of input argument pandas.stats.moments.rolling_median pandas.stats.moments.rolling_median(arg, window, min_periods=None, time_rule=None, **kwargs) O(N log(window)) implementation using skip list Moving median Parameters arg : Series, DataFrame window : Number of observations used for calculating statistic min_periods : int freq=None, freq=None, freq=None,
291
Minimum number of observations in window required to have a value freq : None or string alias / date offset object, default=None Frequency to conform to before computing statistic time_rule is a legacy alias for freq Returns y : type of input argument pandas.stats.moments.rolling_var pandas.stats.moments.rolling_var(arg, window, min_periods=None, time_rule=None, **kwargs) Unbiased moving variance Parameters arg : Series, DataFrame window : Number of observations used for calculating statistic min_periods : int Minimum number of observations in window required to have a value freq : None or string alias / date offset object, default=None Frequency to conform to before computing statistic time_rule is a legacy alias for freq Returns y : type of input argument pandas.stats.moments.rolling_std pandas.stats.moments.rolling_std(arg, window, min_periods=None, time_rule=None, **kwargs) Unbiased moving standard deviation Parameters arg : Series, DataFrame window : Number of observations used for calculating statistic min_periods : int Minimum number of observations in window required to have a value freq : None or string alias / date offset object, default=None Frequency to conform to before computing statistic time_rule is a legacy alias for freq Returns y : type of input argument pandas.stats.moments.rolling_corr pandas.stats.moments.rolling_corr(arg1, arg2, window, min_periods=None, time_rule=None) Moving sample correlation Parameters arg1 : Series, DataFrame, or ndarray arg2 : Series, DataFrame, or ndarray window : Number of observations used for calculating statistic min_periods : int Minimum number of observations in window required to have a value freq : None or string alias / date offset object, default=None 292 Chapter 21. API Reference freq=None, freq=None,
Frequency to conform to before computing statistic time_rule is a legacy alias for freq Returns y : type depends on inputs DataFrame / DataFrame -> DataFrame (matches on columns) DataFrame / Series -> Computes result for each column Series / Series -> Series pandas.stats.moments.rolling_cov pandas.stats.moments.rolling_cov(arg1, arg2, window, min_periods=None, time_rule=None) Unbiased moving covariance Parameters arg1 : Series, DataFrame, or ndarray arg2 : Series, DataFrame, or ndarray window : Number of observations used for calculating statistic min_periods : int Minimum number of observations in window required to have a value freq : None or string alias / date offset object, default=None Frequency to conform to before computing statistic time_rule is a legacy alias for freq Returns y : type depends on inputs DataFrame / DataFrame -> DataFrame (matches on columns) DataFrame / Series -> Computes result for each column Series / Series -> Series pandas.stats.moments.rolling_skew pandas.stats.moments.rolling_skew(arg, window, min_periods=None, time_rule=None, **kwargs) Unbiased moving skewness Parameters arg : Series, DataFrame window : Number of observations used for calculating statistic min_periods : int Minimum number of observations in window required to have a value freq : None or string alias / date offset object, default=None Frequency to conform to before computing statistic time_rule is a legacy alias for freq Returns y : type of input argument pandas.stats.moments.rolling_kurt pandas.stats.moments.rolling_kurt(arg, window, min_periods=None, time_rule=None, **kwargs) Unbiased moving kurtosis Parameters arg : Series, DataFrame window : Number of observations used for calculating statistic min_periods : int freq=None, freq=None,
293
Minimum number of observations in window required to have a value freq : None or string alias / date offset object, default=None Frequency to conform to before computing statistic time_rule is a legacy alias for freq Returns y : type of input argument pandas.stats.moments.rolling_apply pandas.stats.moments.rolling_apply(arg, window, func, min_periods=None, freq=None, time_rule=None) Generic moving function application Parameters arg : Series, DataFrame window : Number of observations used for calculating statistic func : function Must produce a single value from an ndarray input min_periods : int Minimum number of observations in window required to have a value freq : None or string alias / date offset object, default=None Frequency to conform to before computing statistic Returns y : type of input argument pandas.stats.moments.rolling_quantile pandas.stats.moments.rolling_quantile(arg, window, quantile, freq=None, time_rule=None) Moving quantile Parameters arg : Series, DataFrame window : Number of observations used for calculating statistic quantile : 0 <= quantile <= 1 min_periods : int Minimum number of observations in window required to have a value freq : None or string alias / date offset object, default=None Frequency to conform to before computing statistic Returns y : type of input argument min_periods=None,
294
Table 21.7 continued from previous page expanding_var(arg[, min_periods, freq, ...]) Unbiased expanding variance expanding_std(arg[, min_periods, freq, ...]) Unbiased expanding standard deviation expanding_corr(arg1, arg2[, min_periods, ...]) Expanding sample correlation expanding_cov(arg1, arg2[, min_periods, ...]) Unbiased expanding covariance expanding_skew(arg[, min_periods, freq, ...]) Unbiased expanding skewness expanding_kurt(arg[, min_periods, freq, ...]) Unbiased expanding kurtosis expanding_apply(arg, func[, min_periods, ...]) Generic expanding function application expanding_quantile(arg, quantile[, ...]) Expanding quantile
pandas.stats.moments.expanding_count pandas.stats.moments.expanding_count(arg, freq=None, time_rule=None) Expanding count of number of non-NaN observations. Parameters arg : DataFrame or numpy ndarray-like freq : None or string alias / date offset object, default=None Frequency to conform to before computing statistic Returns expanding_count : type of caller pandas.stats.moments.expanding_sum pandas.stats.moments.expanding_sum(arg, min_periods=1, **kwargs) Expanding sum Parameters arg : Series, DataFrame min_periods : int Minimum number of observations in window required to have a value freq : None or string alias / date offset object, default=None Frequency to conform to before computing statistic Returns y : type of input argument pandas.stats.moments.expanding_mean pandas.stats.moments.expanding_mean(arg, min_periods=1, **kwargs) Expanding mean Parameters arg : Series, DataFrame min_periods : int Minimum number of observations in window required to have a value freq : None or string alias / date offset object, default=None Frequency to conform to before computing statistic Returns y : type of input argument freq=None, time_rule=None, freq=None, time_rule=None,
295
pandas.stats.moments.expanding_median pandas.stats.moments.expanding_median(arg, min_periods=1, freq=None, time_rule=None, **kwargs) O(N log(window)) implementation using skip list Expanding median Parameters arg : Series, DataFrame min_periods : int Minimum number of observations in window required to have a value freq : None or string alias / date offset object, default=None Frequency to conform to before computing statistic Returns y : type of input argument pandas.stats.moments.expanding_var pandas.stats.moments.expanding_var(arg, min_periods=1, **kwargs) Unbiased expanding variance Parameters arg : Series, DataFrame min_periods : int Minimum number of observations in window required to have a value freq : None or string alias / date offset object, default=None Frequency to conform to before computing statistic Returns y : type of input argument pandas.stats.moments.expanding_std pandas.stats.moments.expanding_std(arg, min_periods=1, **kwargs) Unbiased expanding standard deviation Parameters arg : Series, DataFrame min_periods : int Minimum number of observations in window required to have a value freq : None or string alias / date offset object, default=None Frequency to conform to before computing statistic Returns y : type of input argument pandas.stats.moments.expanding_corr pandas.stats.moments.expanding_corr(arg1, arg2, min_periods=1, time_rule=None) Expanding sample correlation freq=None, time_rule=None, freq=None, time_rule=None,
296
Parameters arg1 : Series, DataFrame, or ndarray arg2 : Series, DataFrame, or ndarray min_periods : int Minimum number of observations in window required to have a value freq : None or string alias / date offset object, default=None Frequency to conform to before computing statistic Returns y : type depends on inputs DataFrame / DataFrame -> DataFrame (matches on columns) DataFrame / Series -> Computes result for each column Series / Series -> Series pandas.stats.moments.expanding_cov pandas.stats.moments.expanding_cov(arg1, arg2, min_periods=1, time_rule=None) Unbiased expanding covariance Parameters arg1 : Series, DataFrame, or ndarray arg2 : Series, DataFrame, or ndarray min_periods : int Minimum number of observations in window required to have a value freq : None or string alias / date offset object, default=None Frequency to conform to before computing statistic Returns y : type depends on inputs DataFrame / DataFrame -> DataFrame (matches on columns) DataFrame / Series -> Computes result for each column Series / Series -> Series pandas.stats.moments.expanding_skew pandas.stats.moments.expanding_skew(arg, min_periods=1, **kwargs) Unbiased expanding skewness Parameters arg : Series, DataFrame min_periods : int Minimum number of observations in window required to have a value freq : None or string alias / date offset object, default=None Frequency to conform to before computing statistic Returns y : type of input argument pandas.stats.moments.expanding_kurt pandas.stats.moments.expanding_kurt(arg, min_periods=1, **kwargs) Unbiased expanding kurtosis freq=None, time_rule=None, freq=None, time_rule=None,
297
Parameters arg : Series, DataFrame min_periods : int Minimum number of observations in window required to have a value freq : None or string alias / date offset object, default=None Frequency to conform to before computing statistic Returns y : type of input argument pandas.stats.moments.expanding_apply pandas.stats.moments.expanding_apply(arg, func, time_rule=None) Generic expanding function application Parameters arg : Series, DataFrame func : function Must produce a single value from an ndarray input min_periods : int Minimum number of observations in window required to have a value freq : None or string alias / date offset object, default=None Frequency to conform to before computing statistic Returns y : type of input argument pandas.stats.moments.expanding_quantile pandas.stats.moments.expanding_quantile(arg, quantile, min_periods=1, time_rule=None) Expanding quantile Parameters arg : Series, DataFrame quantile : 0 <= quantile <= 1 min_periods : int Minimum number of observations in window required to have a value freq : None or string alias / date offset object, default=None Frequency to conform to before computing statistic Returns y : type of input argument freq=None, min_periods=1, freq=None,
298
pandas.stats.moments.ewma pandas.stats.moments.ewma(arg, com=None, span=None, time_rule=None, adjust=True) Exponentially-weighted moving average Parameters arg : Series, DataFrame com : oat. optional Center of mass: alpha = com / (1 + com), span : oat, optional Specify decay in terms of span, alpha = 2 / (span + 1) min_periods : int, default 0 Number of observations in sample to require (only affects beginning) freq : None or string alias / date offset object, default=None Frequency to conform to before computing statistic time_rule is a legacy alias for freq adjust : boolean, default True Divide by decaying adjustment factor in beginning periods to account for imbalance in relative weightings (viewing EWMA as a moving average) Returns y : type of input argument
Notes
min_periods=0,
freq=None,
Either center of mass or span must be specied EWMA is sometimes specied using a span parameter s, we have have that the decay parameter alpha is related to the span as = 1 2/(s + 1) = c/(1 + c) where c is the center of mass. Given a span, the associated center of mass is c = (s 1)/2 So a 20-day EWMA would have center 9.5. pandas.stats.moments.ewmstd pandas.stats.moments.ewmstd(arg, com=None, time_rule=None) Exponentially-weighted moving std Parameters arg : Series, DataFrame com : oat. optional Center of mass: alpha = com / (1 + com), span : oat, optional Specify decay in terms of span, alpha = 2 / (span + 1) min_periods : int, default 0 Number of observations in sample to require (only affects beginning) freq : None or string alias / date offset object, default=None Frequency to conform to before computing statistic time_rule is a legacy alias for freq 21.1. General functions 299 span=None, min_periods=0, bias=False,
adjust : boolean, default True Divide by decaying adjustment factor in beginning periods to account for imbalance in relative weightings (viewing EWMA as a moving average) bias : boolean, default False Use a standard estimation bias correction Returns y : type of input argument
Notes
Either center of mass or span must be specied EWMA is sometimes specied using a span parameter s, we have have that the decay parameter alpha is related to the span as = 1 2/(s + 1) = c/(1 + c) where c is the center of mass. Given a span, the associated center of mass is c = (s 1)/2 So a 20-day EWMA would have center 9.5. pandas.stats.moments.ewmvar pandas.stats.moments.ewmvar(arg, com=None, span=None, freq=None, time_rule=None) Exponentially-weighted moving variance Parameters arg : Series, DataFrame com : oat. optional Center of mass: alpha = com / (1 + com), span : oat, optional Specify decay in terms of span, alpha = 2 / (span + 1) min_periods : int, default 0 Number of observations in sample to require (only affects beginning) freq : None or string alias / date offset object, default=None Frequency to conform to before computing statistic time_rule is a legacy alias for freq adjust : boolean, default True Divide by decaying adjustment factor in beginning periods to account for imbalance in relative weightings (viewing EWMA as a moving average) bias : boolean, default False Use a standard estimation bias correction Returns y : type of input argument
Notes
min_periods=0,
bias=False,
Either center of mass or span must be specied EWMA is sometimes specied using a span parameter s, we have have that the decay parameter alpha is related to the span as = 1 2/(s + 1) = c/(1 + c) 300 Chapter 21. API Reference
where c is the center of mass. Given a span, the associated center of mass is c = (s 1)/2 So a 20-day EWMA would have center 9.5. pandas.stats.moments.ewmcorr pandas.stats.moments.ewmcorr(arg1, arg2, com=None, span=None, min_periods=0, freq=None, time_rule=None) Exponentially-weighted moving correlation Parameters arg1 : Series, DataFrame, or ndarray arg2 : Series, DataFrame, or ndarray com : oat. optional Center of mass: alpha = com / (1 + com), span : oat, optional Specify decay in terms of span, alpha = 2 / (span + 1) min_periods : int, default 0 Number of observations in sample to require (only affects beginning) freq : None or string alias / date offset object, default=None Frequency to conform to before computing statistic time_rule is a legacy alias for freq adjust : boolean, default True Divide by decaying adjustment factor in beginning periods to account for imbalance in relative weightings (viewing EWMA as a moving average) Returns y : type of input argument
Notes
Either center of mass or span must be specied EWMA is sometimes specied using a span parameter s, we have have that the decay parameter alpha is related to the span as = 1 2/(s + 1) = c/(1 + c) where c is the center of mass. Given a span, the associated center of mass is c = (s 1)/2 So a 20-day EWMA would have center 9.5. pandas.stats.moments.ewmcov pandas.stats.moments.ewmcov(arg1, arg2, com=None, span=None, min_periods=0, bias=False, freq=None, time_rule=None) Exponentially-weighted moving covariance Parameters arg1 : Series, DataFrame, or ndarray arg2 : Series, DataFrame, or ndarray com : oat. optional Center of mass: alpha = com / (1 + com), span : oat, optional
301
Specify decay in terms of span, alpha = 2 / (span + 1) min_periods : int, default 0 Number of observations in sample to require (only affects beginning) freq : None or string alias / date offset object, default=None Frequency to conform to before computing statistic time_rule is a legacy alias for freq adjust : boolean, default True Divide by decaying adjustment factor in beginning periods to account for imbalance in relative weightings (viewing EWMA as a moving average) Returns y : type of input argument
Notes
Either center of mass or span must be specied EWMA is sometimes specied using a span parameter s, we have have that the decay parameter alpha is related to the span as = 1 2/(s + 1) = c/(1 + c) where c is the center of mass. Given a span, the associated center of mass is c = (s 1)/2 So a 20-day EWMA would have center 9.5.
21.2 Series
21.2.1 Attributes and underlying data
Axes index: axis labels Series.values Series.dtype Series.isnull(obj) Series.notnull(obj) Return Series as ndarray Data-type of the arrays elements. Replacement for numpy.isnan / -numpy.isnite which is suitable for use on object arrays. Replacement for numpy.isnite / -numpy.isnan which is suitable for use on object arrays.
pandas.Series.values Series.values Return Series as ndarray Returns arr : numpy.ndarray pandas.Series.dtype Series.dtype Data-type of the arrays elements. Parameters None : Returns d : numpy dtype object
302
pandas.Series.isnull Series.isnull(obj) Replacement for numpy.isnan / -numpy.isnite which is suitable for use on object arrays. Parameters arr: ndarray or object value : Returns boolean ndarray or boolean : pandas.Series.notnull Series.notnull(obj) Replacement for numpy.isnite / -numpy.isnan which is suitable for use on object arrays. Parameters arr: ndarray or object value : Returns boolean ndarray or boolean :
pandas.Series.__init__ Series.__init__(data=None, index=None, dtype=None, name=None, copy=False) One-dimensional ndarray with axis labels (including time series). Labels need not be unique but must be any hashable type. The object supports both integer- and label-based indexing and provides a host of methods for performing operations involving the index. Statistical methods from ndarray have been overridden to automatically exclude missing data (currently represented as NaN) Operations between Series (+, -, /, , *) align values based on their associated index values they need not be the same length. The result index will be the sorted union of the two indexes. Parameters data : array-like, dict, or scalar value Contains data stored in Series index : array-like or Index (1d) 21.2. Series 303
Values must be unique and hashable, same length as data. Index object (or other iterable of same length as data) Will default to np.arange(len(data)) if not provided. If both a dict and index sequence are used, the index will override the keys found in the dict. dtype : numpy.dtype or None If None, dtype will be inferred copy : boolean, default False Copy input data copy : boolean, default False pandas.Series.astype Series.astype(dtype) See numpy.ndarray.astype pandas.Series.copy Series.copy(order=C) Return new Series with copy of underlying values Returns cp : Series
pandas.Series.get Series.get(label, default=None) Returns value occupying requested label, default to specied missing value if not present. Analogous to dict.get Parameters label : object Label value looking for default : object, optional Value to return if label not in index Returns y : scalar pandas.Series.ix Series.ix pandas.Series.__iter__ Series.__iter__()
304
pandas.Series.add Series.add(other, level=None, ll_value=None) Binary operator add with support to substitute a ll_value for missing data in one of the inputs Parameters other: Series or scalar value : ll_value : None or oat value, default None (NaN) Fill missing (NaN) values with this value. If both Series are missing, the result will be missing level : int or name Broadcast across a level, matching Index values on the passed MultiIndex level Returns result : Series pandas.Series.div Series.div(other, level=None, ll_value=None) Binary operator divide with support to substitute a ll_value for missing data in one of the inputs Parameters other: Series or scalar value : ll_value : None or oat value, default None (NaN) Fill missing (NaN) values with this value. If both Series are missing, the result will be missing level : int or name Broadcast across a level, matching Index values on the passed MultiIndex level Returns result : Series pandas.Series.mul Series.mul(other, level=None, ll_value=None) Binary operator multiply with support to substitute a ll_value for missing data in one of the inputs
21.2. Series
305
Parameters other: Series or scalar value : ll_value : None or oat value, default None (NaN) Fill missing (NaN) values with this value. If both Series are missing, the result will be missing level : int or name Broadcast across a level, matching Index values on the passed MultiIndex level Returns result : Series pandas.Series.sub Series.sub(other, level=None, ll_value=None) Binary operator subtract with support to substitute a ll_value for missing data in one of the inputs Parameters other: Series or scalar value : ll_value : None or oat value, default None (NaN) Fill missing (NaN) values with this value. If both Series are missing, the result will be missing level : int or name Broadcast across a level, matching Index values on the passed MultiIndex level Returns result : Series pandas.Series.combine Series.combine(other, func, ll_value=nan) Perform elementwise binary operation on two Series using given function with optional ll value when an index is missing from one Series or the other Parameters other : Series or scalar value func : function ll_value : scalar value Returns result : Series pandas.Series.combine_rst Series.combine_first(other) Combine Series values, choosing the calling Seriess values rst. Result index will be the union of the two indexes Parameters other : Series Returns y : Series
306
pandas.Series.round Series.round(decimals=0, out=None) Return a with each element rounded to the given number of decimals. Refer to numpy.around for full documentation. See Also: numpy.around equivalent function
pandas.Series.apply Series.apply(func, convert_dtype=True, args=(), **kwds) Invoke function on values of Series. Can be ufunc, a Python function that applies to the entire Series, or a Python function that only works on single values Parameters func : function convert_dtype : boolean, default True Try to nd better dtype for elementwise function results. If False, leave as dtype=object Returns y : Series See Also: Series.map For element-wise operations
Notes
func is applied to the entire Series at once rst. If an exception is raised, then apply to each value. pandas.Series.map Series.map(arg, na_action=None) Map values of Series using input correspondence (which can be a dict, Series, or function) Parameters arg : function, dict, or Series na_action : {None, ignore} If ignore, propagate NA values Returns y : Series same index as caller
21.2. Series
307
Examples >>> x one 1 two 2 three 3 >>> y 1 foo 2 bar 3 baz >>> x.map(y) one foo two bar three baz
pandas.Series.groupby Series.groupby(by=None, axis=0, level=None, as_index=True, sort=True, group_keys=True) Group series using mapper (dict or key function, apply given function to group, return result as series) or by a series of columns Parameters by : mapping function / list of functions, dict, Series, or tuple / list of column names. Called on each element of the object index to determine the groups. If a dict or Series is passed, the Series or dict VALUES will be used to determine the groups axis : int, default 0 level : int, level name, or sequence of such, default None If the axis is a MultiIndex (hierarchical), group by a particular level or levels as_index : boolean, default True For aggregated output, return object with group labels as the index. Only relevant for DataFrame input. as_index=False is effectively SQL-style grouped output sort : boolean, default True Sort group keys. Get better performance by turning this off group_keys : boolean, default True When calling apply, add group keys to index to identify pieces Returns GroupBy object :
Examples
# DataFrame result >>> data.groupby(func, axis=0).mean() # DataFrame result >>> data.groupby([col1, col2])[col3].mean() # DataFrame with hierarchical index >>> data.groupby([col1, col2]).mean()
308
21.2. Series
309
Series.abs() Series.any([axis, out]) Series.autocorr() Series.between(left, right[, inclusive]) Series.clip([lower, upper, out]) Series.clip_lower(threshold) Series.clip_upper(threshold) Series.corr(other[, method]) Series.count([level]) Series.cov(other) Series.cummax([axis, dtype, out, skipna]) Series.cummin([axis, dtype, out, skipna]) Series.cumprod([axis, dtype, out, skipna]) Series.cumsum([axis, dtype, out, skipna]) Series.describe([percentile_width]) Series.diff([periods]) Series.kurt([skipna, level]) Series.mad([skipna, level]) Series.max([axis, out, skipna, level]) Series.mean([axis, dtype, out, skipna, level]) Series.median([axis, dtype, out, skipna, level]) Series.min([axis, out, skipna, level]) Series.nunique() Series.pct_change([periods, ll_method, ...]) Series.prod([axis, dtype, out, skipna, level]) Series.quantile([q]) Series.rank([method, na_option, ascending]) Series.skew([skipna, level]) Series.std([axis, dtype, out, ddof, skipna, ...]) Series.sum([axis, dtype, out, skipna, level]) Series.unique() Series.var([axis, dtype, out, ddof, skipna, ...]) Series.value_counts()
Return an object with absolute value taken. Returns True if any of the elements of a evaluate to True. Lag-1 autocorrelation Return boolean Series equivalent to left <= series <= right. NA values Trim values at input threshold(s) Return copy of series with values below given value truncated Return copy of series with values above given value truncated Compute correlation two Series, excluding missing values Return number of non-NA/null observations in the Series Compute covariance with Series, excluding missing values Cumulative max of values. Cumulative min of values. Cumulative product of values. Cumulative sum of values. Generate various summary statistics of Series, excluding NaN 1st discrete difference of object Return unbiased kurtosis of values Return mean absolute deviation of values Return maximum of values Return mean of values Return median of values Return minimum of values Return count of unique elements in the Series Percent change over given number of periods Return product of values Return value at the given quantile, a la scoreatpercentile in Compute data ranks (1 through n). Return unbiased skewness of values Return standard deviation of values Return sum of values Return array of unique values in the Series. Signicantly faster than Return variance of values Returns Series containing counts of unique values. The resulting Series
pandas.Series.abs Series.abs() Return an object with absolute value taken. Only applicable to objects that are all numeric Returns abs: type of caller : pandas.Series.any Series.any(axis=None, out=None) Returns True if any of the elements of a evaluate to True. Refer to numpy.any for full documentation. See Also: numpy.any equivalent function
310
pandas.Series.autocorr Series.autocorr() Lag-1 autocorrelation Returns autocorr : oat pandas.Series.between Series.between(left, right, inclusive=True) Return boolean Series equivalent to left <= series <= right. NA values will be treated as False Parameters left : scalar Left boundary right : scalar Right boundary Returns is_between : Series pandas.Series.clip Series.clip(lower=None, upper=None, out=None) Trim values at input threshold(s) Parameters lower : oat, default None upper : oat, default None Returns clipped : Series pandas.Series.clip_lower Series.clip_lower(threshold) Return copy of series with values below given value truncated Returns clipped : Series See Also: clip pandas.Series.clip_upper Series.clip_upper(threshold) Return copy of series with values above given value truncated Returns clipped : Series See Also: clip
21.2. Series
311
pandas.Series.corr Series.corr(other, method=pearson) Compute correlation two Series, excluding missing values Parameters other : Series method : {pearson, kendall, spearman} pearson : standard correlation coefcient kendall : Kendall Tau correlation coefcient spearman : Spearman rank correlation Returns correlation : oat pandas.Series.count Series.count(level=None) Return number of non-NA/null observations in the Series Parameters level : int, default None If the axis is a MultiIndex (hierarchical), count along a particular level, collapsing into a smaller Series Returns nobs : int or Series (if level specied) pandas.Series.cov Series.cov(other) Compute covariance with Series, excluding missing values Parameters other : Series Returns covariance : oat Normalized by N-1 (unbiased estimator). : pandas.Series.cummax Series.cummax(axis=0, dtype=None, out=None, skipna=True) Cumulative max of values. Preserves locations of NaN values Extra parameters are to preserve ndarray interface. Parameters skipna : boolean, default True Exclude NA/null values Returns cummax : Series pandas.Series.cummin Series.cummin(axis=0, dtype=None, out=None, skipna=True) Cumulative min of values. Preserves locations of NaN values Extra parameters are to preserve ndarray interface. Parameters skipna : boolean, default True Exclude NA/null values 312 Chapter 21. API Reference
Returns cummin : Series pandas.Series.cumprod Series.cumprod(axis=0, dtype=None, out=None, skipna=True) Cumulative product of values. Preserves locations of NaN values Extra parameters are to preserve ndarray interface. Parameters skipna : boolean, default True Exclude NA/null values Returns cumprod : Series pandas.Series.cumsum Series.cumsum(axis=0, dtype=None, out=None, skipna=True) Cumulative sum of values. Preserves locations of NaN values Extra parameters are to preserve ndarray interface. Parameters skipna : boolean, default True Exclude NA/null values Returns cumsum : Series pandas.Series.describe Series.describe(percentile_width=50) Generate various summary statistics of Series, excluding NaN values. These include: count, mean, std, min, max, and lower%/50%/upper% percentiles Parameters percentile_width : oat, optional width of the desired uncertainty interval, default is 50, which corresponds to lower=25, upper=75 Returns desc : Series pandas.Series.diff Series.diff(periods=1) 1st discrete difference of object Parameters periods : int, default 1 Periods to shift for forming difference Returns diffed : Series pandas.Series.kurt Series.kurt(skipna=True, level=None) Return unbiased kurtosis of values NA/null values are excluded Parameters skipna : boolean, default True 21.2. Series 313
Exclude NA/null values level : int, default None If the axis is a MultiIndex (hierarchical), count along a particular level, collapsing into a smaller Series Returns kurt : oat (or Series if level specied) pandas.Series.mad Series.mad(skipna=True, level=None) Return mean absolute deviation of values NA/null values are excluded Parameters skipna : boolean, default True Exclude NA/null values level : int, default None If the axis is a MultiIndex (hierarchical), count along a particular level, collapsing into a smaller Series Returns mad : oat (or Series if level specied) pandas.Series.max Series.max(axis=None, out=None, skipna=True, level=None) Return maximum of values NA/null values are excluded Parameters skipna : boolean, default True Exclude NA/null values level : int, default None If the axis is a MultiIndex (hierarchical), count along a particular level, collapsing into a smaller Series Returns max : oat (or Series if level specied) pandas.Series.mean Series.mean(axis=0, dtype=None, out=None, skipna=True, level=None) Return mean of values NA/null values are excluded Parameters skipna : boolean, default True Exclude NA/null values level : int, default None If the axis is a MultiIndex (hierarchical), count along a particular level, collapsing into a smaller Series Extra parameters are to preserve ndarrayinterface. : Returns mean : oat (or Series if level specied)
314
pandas.Series.median Series.median(axis=0, dtype=None, out=None, skipna=True, level=None) Return median of values NA/null values are excluded Parameters skipna : boolean, default True Exclude NA/null values level : int, default None If the axis is a MultiIndex (hierarchical), count along a particular level, collapsing into a smaller Series Returns median : oat (or Series if level specied) pandas.Series.min Series.min(axis=None, out=None, skipna=True, level=None) Return minimum of values NA/null values are excluded Parameters skipna : boolean, default True Exclude NA/null values level : int, default None If the axis is a MultiIndex (hierarchical), count along a particular level, collapsing into a smaller Series Returns min : oat (or Series if level specied) pandas.Series.nunique Series.nunique() Return count of unique elements in the Series Returns nunique : int pandas.Series.pct_change Series.pct_change(periods=1, ll_method=pad, limit=None, freq=None, **kwds) Percent change over given number of periods Parameters periods : int, default 1 Periods to shift for forming percent change ll_method : str, default pad How to handle NAs before computing percent changes limit : int, default None The number of consecutive NAs to ll before stopping freq : DateOffset, timedelta, or offset alias string, optional Increment to use from time series API (e.g. M or BDay()) Returns chg : Series or DataFrame
21.2. Series
315
pandas.Series.prod Series.prod(axis=0, dtype=None, out=None, skipna=True, level=None) Return product of values NA/null values are excluded Parameters skipna : boolean, default True Exclude NA/null values level : int, default None If the axis is a MultiIndex (hierarchical), count along a particular level, collapsing into a smaller Series Returns prod : oat (or Series if level specied) pandas.Series.quantile Series.quantile(q=0.5) Return value at the given quantile, a la scoreatpercentile in scipy.stats Parameters q : quantile 0 <= q <= 1 Returns quantile : oat pandas.Series.rank Series.rank(method=average, na_option=keep, ascending=True) Compute data ranks (1 through n). Equal values are assigned a rank that is the average of the ranks of those values Parameters method : {average, min, max, rst} average: average rank of group min: lowest rank in group max: highest rank in group rst: ranks assigned in order they appear in the array na_option : {keep} keep: leave NA values where they are ascending : boolean, default True False for ranks by high (1) to low (N) Returns ranks : Series pandas.Series.skew Series.skew(skipna=True, level=None) Return unbiased skewness of values NA/null values are excluded Parameters skipna : boolean, default True Exclude NA/null values level : int, default None If the axis is a MultiIndex (hierarchical), count along a particular level, collapsing into a smaller Series
316
Returns skew : oat (or Series if level specied) pandas.Series.std Series.std(axis=None, dtype=None, out=None, ddof=1, skipna=True, level=None) Return standard deviation of values NA/null values are excluded Parameters skipna : boolean, default True Exclude NA/null values level : int, default None If the axis is a MultiIndex (hierarchical), count along a particular level, collapsing into a smaller Series Returns stdev : oat (or Series if level specied) Normalized by N-1 (unbiased estimator). pandas.Series.sum Series.sum(axis=0, dtype=None, out=None, skipna=True, level=None) Return sum of values NA/null values are excluded Parameters skipna : boolean, default True Exclude NA/null values level : int, default None If the axis is a MultiIndex (hierarchical), count along a particular level, collapsing into a smaller Series Extra parameters are to preserve ndarrayinterface. : Returns sum : oat (or Series if level specied) pandas.Series.unique Series.unique() Return array of unique values in the Series. Signicantly faster than numpy.unique Returns uniques : ndarray pandas.Series.var Series.var(axis=None, dtype=None, out=None, ddof=1, skipna=True, level=None) Return variance of values NA/null values are excluded Parameters skipna : boolean, default True Exclude NA/null values level : int, default None If the axis is a MultiIndex (hierarchical), count along a particular level, collapsing into a smaller Series Returns var : oat (or Series if level specied)
21.2. Series
317
Normalized by N-1 (unbiased estimator). pandas.Series.value_counts Series.value_counts() Returns Series containing counts of unique values. The resulting Series will be in descending order so that the rst element is the most frequently-occurring element. Excludes NA values Returns counts : Series
Aligned Series pandas.Series.drop Series.drop(labels, axis=0, level=None) Return new object with labels in requested axis removed Parameters labels : array-like axis : int level : int or name, default None For MultiIndex Returns dropped : type of caller pandas.Series.rst Series.first(offset) Convenience method for subsetting initial periods of time series data based on a date offset Parameters offset : string, DateOffset, dateutil.relativedelta Returns subset : type of caller
Examples
ts.last(10D) -> First 10 days pandas.Series.head Series.head(n=5) Returns rst n rows of Series pandas.Series.idxmax Series.idxmax(axis=None, out=None, skipna=True) Index of rst occurrence of maximum of values. Parameters skipna : boolean, default True Exclude NA/null values Returns idxmax : Index of minimum of values pandas.Series.idxmin Series.idxmin(axis=None, out=None, skipna=True) Index of rst occurrence of minimum of values. Parameters skipna : boolean, default True Exclude NA/null values Returns idxmin : Index of minimum of values
21.2. Series
319
pandas.Series.isin Series.isin(values) Return boolean vector showing whether each element in the Series is exactly contained in the passed sequence of values Parameters values : sequence Returns isin : Series (boolean dtype) pandas.Series.last Series.last(offset) Convenience method for subsetting nal periods of time series data based on a date offset Parameters offset : string, DateOffset, dateutil.relativedelta Returns subset : type of caller
Examples
ts.last(5M) -> Last 5 months pandas.Series.reindex Series.reindex(index=None, method=None, level=None, ll_value=nan, limit=None, copy=True) Conform Series to new index with optional lling logic, placing NA/NaN in locations having no value in the previous index. A new object is produced unless the new index is equivalent to the current one and copy=False Parameters index : array-like or Index New labels / index to conform to. Preferably an Index object to avoid duplicating data method : {backll, bll, pad, fll, None} Method to use for lling holes in reindexed Series pad / fll: propagate LAST valid observation forward to next valid backll / bll: use NEXT valid observation to ll gap copy : boolean, default True Return a new object, even if the passed indexes are the same level : int or name Broadcast across a level, matching Index values on the passed MultiIndex level ll_value : scalar, default np.NaN Value to use for missing values. Defaults to NaN, but can be any compatible value limit : int, default None Maximum size gap to forward or backward ll Returns reindexed : Series
320
pandas.Series.reindex_like Series.reindex_like(other, method=None, limit=None) Reindex Series to match index of another Series, optionally with lling logic Parameters other : Series method : string or None See Series.reindex docstring limit : int, default None Maximum size gap to forward or backward ll Returns reindexed : Series
Notes
Like calling s.reindex(other.index, method=...) pandas.Series.rename Series.rename(mapper, inplace=False) Alter Series index using dict or function Parameters mapper : dict-like or function Transformation to apply to each index Returns renamed : Series (new object)
Notes
21.2. Series
321
pandas.Series.reset_index Series.reset_index(level=None, drop=False, name=None, inplace=False) Analogous to the DataFrame.reset_index function, see docstring there. Parameters level : int, str, tuple, or list, default None Only remove the given levels from the index. Removes all levels by default drop : boolean, default False Do not try to insert index into dataframe columns name : object, default None The name of the column corresponding to the Series values inplace : boolean, default False Modify the Series in place (do not create a new object) Returns resetted : DataFrame, or Series if drop == True pandas.Series.select Series.select(crit, axis=0) Return data corresponding to axis labels matching criteria Parameters crit : function To be called on each index (label). Should return True or False axis : int Returns selection : type of caller pandas.Series.take Series.take(indices, axis=0) Analogous to ndarray.take, return Series corresponding to requested indices Parameters indices : list / array of ints Returns taken : Series pandas.Series.tail Series.tail(n=5) Returns last n rows of Series pandas.Series.truncate Series.truncate(before=None, after=None, copy=True) Function truncate a sorted DataFrame / Series before and/or after some particular dates. Parameters before : date Truncate before date after : date
322
pandas.Series.dropna Series.dropna() Return Series without null values Returns valid : Series pandas.Series.llna Series.fillna(value=None, method=pad, inplace=False, limit=None) Fill NA/NaN values using the specied method Parameters value : any kind (should be same type as array) Value to use to ll holes (e.g. 0) method : {backll, bll, pad, fll, None}, default pad Method to use for lling holes in reindexed Series pad / fll: propagate last valid observation forward to next valid backll / bll: use NEXT valid observation to ll gap inplace : boolean, default False If True, ll the Series in place. Note: this will modify any other views on this Series, for example a column in a DataFrame. Returns a reference to the lled object, which is self if inplace=True limit : int, default None Maximum size gap to forward or backward ll Returns lled : Series See Also: reindex, asfreq pandas.Series.interpolate Series.interpolate(method=linear) Interpolate missing values (after the rst valid value) Parameters method : {linear, time, values} Interpolation method. time interpolation works on daily and higher resolution data to interpolate given length of interval values using the actual index numeric values Returns interpolated : Series
21.2. Series
323
pandas.Series.argsort Series.argsort(axis=0, kind=quicksort, order=None) Overrides ndarray.argsort. Argsorts the value, omitting NA/null values, and places the result in the same locations as the non-NA values Parameters axis : int (can only be zero) kind : {mergesort, quicksort, heapsort}, default quicksort Choice of sorting algorithm. See np.sort for more information. mergesort is the only stable algorithm order : ignored Returns argsorted : Series pandas.Series.order Series.order(na_last=True, ascending=True, kind=mergesort) Sorts Series object, by value, maintaining index-value link Parameters na_last : boolean (optional, default=True) Put NaNs at beginning or end ascending : boolean, default True Sort ascending. Passing False sorts descending kind : {mergesort, quicksort, heapsort}, default mergesort Choice of sorting algorithm. See np.sort for more information. mergesort is the only stable algorithm Returns y : Series pandas.Series.reorder_levels Series.reorder_levels(order) Rearrange index levels using input order. May not drop or duplicate levels Parameters order: list of int representing new level order. : (reference level by number not by key) axis: where to reorder levels :
324
Returns type of caller (new object) : pandas.Series.sort Series.sort(axis=0, kind=quicksort, order=None) Sort values and index labels by value, in place. For compatibility with ndarray API. No return value Parameters axis : int (can only be zero) kind : {mergesort, quicksort, heapsort}, default quicksort Choice of sorting algorithm. See np.sort for more information. mergesort is the only stable algorithm order : ignored pandas.Series.sort_index Series.sort_index(ascending=True) Sort object by labels (along an axis) Parameters ascending : boolean or list, default True Sort ascending vs. descending. Specify list for multiple sort orders Returns sorted_obj : Series
Examples >>> result1 = s.sort_index(ascending=False) >>> result2 = s.sort_index(ascending=[1, 0])
pandas.Series.sortlevel Series.sortlevel(level=0, ascending=True) Sort Series with MultiIndex by chosen level. Data will be lexicographically sorted by the chosen level followed by the other levels (in order) Parameters level : int ascending : bool, default True Returns sorted : Series pandas.Series.swaplevel Series.swaplevel(i, j, copy=True) Swap levels i and j in a MultiIndex Returns swapped : Series
21.2. Series
325
pandas.Series.unstack Series.unstack(level=-1) Unstack, a.k.a. pivot, Series with MultiIndex to produce DataFrame Parameters level : int, string, or list of these, default last level Level(s) to unstack, can pass level name Returns unstacked : DataFrame
Examples >>> s one a one b two a two b
1. 2. 3. 4.
pandas.Series.append Series.append(to_append, verify_integrity=False) Concatenate two or more Series. The indexes must not overlap Parameters to_append : Series or list/tuple of Series verify_integrity : boolean, default False If True, raise Exception on creating index with duplicates Returns appended : Series pandas.Series.replace Series.replace(to_replace, value=None, method=pad, inplace=False, limit=None) Replace arbitrary values in a Series Parameters to_replace : list or dict
326
list of values to be replaced or dict of replacement values value : anything if to_replace is a list then value is the replacement value method : {backll, bll, pad, fll, None}, default pad Method to use for lling holes in reindexed Series pad / fll: propagate last valid observation forward to next valid backll / bll: use NEXT valid observation to ll gap inplace : boolean, default False If True, ll the Series in place. Note: this will modify any other views on this Series, for example a column in a DataFrame. Returns a reference to the lled object, which is self if inplace=True limit : int, default None Maximum size gap to forward or backward ll Returns replaced : Series See Also: fillna, reindex, asfreq
Notes
replace does not distinguish between NaN and None pandas.Series.update Series.update(other) Modify Series in place using non-NA values from passed Series. Aligns on index Parameters other : Series
Convenience method for frequency conversion and resampling of regular time-series dat Convert TimeSeries to target time zone Localize tz-naive TimeSeries to target time zone
pandas.Series.asfreq Series.asfreq(freq, method=None, how=None) Convert all TimeSeries inside to specied frequency using DateOffset objects. Optionally provide ll method to pad/backll missing values.
21.2. Series
327
Parameters freq : DateOffset object, or string method : {backll, bll, pad, fll, None} Method to use for lling holes in reindexed Series pad / fll: propagate last valid observation forward to next valid backll / bll: use NEXT valid observation to ll methdo how : {start, end}, default end For PeriodIndex only, see PeriodIndex.asfreq Returns converted : type of caller pandas.Series.asof Series.asof(where) Return last good (non-NaN) value in TimeSeries if value is NaN for requested date. If there is no good value, NaN is returned. Parameters where : date or array of dates Returns value or NaN :
Notes
Dates are assumed to be sorted pandas.Series.shift Series.shift(periods=1, freq=None, copy=True, **kwds) Shift the index of the Series by desired number of periods with an optional time offset Parameters periods : int Number of periods to move, can be positive or negative freq : DateOffset, timedelta, or offset alias string, optional Increment to use from datetools module or time rule (e.g. EOM) Returns shifted : Series pandas.Series.rst_valid_index Series.first_valid_index() Return label for rst non-NA/null value pandas.Series.last_valid_index Series.last_valid_index() Return label for last non-NA/null value pandas.Series.weekday Series.weekday
328
pandas.Series.resample Series.resample(rule, how=None, axis=0, ll_method=None, closed=right, label=right, convention=start, kind=None, loffset=None, limit=None, base=0) Convenience method for frequency conversion and resampling of regular time-series data. Parameters rule : the offset string or object representing target conversion how : string, method for down- or re-sampling, default to mean for downsampling ll_method : string, ll_method for upsampling, default None axis : int, optional, default 0 closed : {right, left}, default right Which side of bin interval is closed label : {right, left}, default right Which bin edge label to label bucket with convention : {start, end, s, e} loffset : timedelta Adjust the resampled time labels base : int, default 0 For frequencies that evenly subdivide 1 day, the origin of the aggregated intervals. For example, for 5min frequency, base could range from 0 through 4. Defaults to 0 pandas.Series.tz_convert Series.tz_convert(tz, copy=True) Convert TimeSeries to target time zone Parameters tz : string or pytz.timezone object copy : boolean, default True Also make a copy of the underlying data Returns converted : TimeSeries pandas.Series.tz_localize Series.tz_localize(tz, copy=True) Localize tz-naive TimeSeries to target time zone Parameters tz : string or pytz.timezone object copy : boolean, default True Also make a copy of the underlying data Returns localized : TimeSeries
21.2. Series
329
21.2.12 Plotting
330
Draw histogram of the input series using matplotlib Plot the input series with the index on the x-axis using matplotlib
pandas.Series.hist Series.hist(by=None, ax=None, grid=True, xlabelsize=None, xrot=None, ylabelsize=None, yrot=None, **kwds) Draw histogram of the input series using matplotlib Parameters by : object, optional If passed, then used to form histograms for separate groups ax : matplotlib axis object If not passed, uses gca() grid : boolean, default True Whether to show axis grid lines xlabelsize : int, default None If specied changes the x-axis label size xrot : oat, default None rotation of x axis labels ylabelsize : int, default None If specied changes the y-axis label size yrot : oat, default None rotation of y axis labels kwds : keywords To be passed to the actual plotting function
Notes
See matplotlib documentation online for more on this pandas.Series.plot Series.plot(series, label=None, kind=line, use_index=True, rot=None, xticks=None, yticks=None, xlim=None, ylim=None, ax=None, style=None, grid=None, legend=False, logy=False, secondary_y=False, **kwds) Plot the input series with the index on the x-axis using matplotlib Parameters label : label argument to provide to plot kind : {line, bar, barh} bar : vertical bar plot barh : horizontal bar plot rot : int, default 30 Rotation for tick labels
21.2. Series
331
use_index : boolean, default True Plot index as axis tick labels ax : matplotlib axis object If not passed, uses gca() style : string, default matplotlib default matplotlib line style to use ax : matplotlib axis object If not passed, uses gca() logy : boolean, default False For line plots, use log scaling on y axis xticks : sequence Values to use for the xticks yticks : sequence Values to use for the yticks xlim : 2-tuple/list ylim : 2-tuple/list rot : int, default None Rotation for ticks kwds : keywords Options to pass to matplotlib plotting method
Notes
Write Series to a comma-separated values (csv) le Convert Series to {label -> value} dict Convert Series to SparseSeries Render a string representation of the Series
sep : string, default , Field delimiter parse_dates : boolean, default True Parse dates. Different default from read_table header : int, default 0 Row to use at header (skip prior rows) index_col : int or sequence, default 0 Column to use for index. If a sequence is given, a MultiIndex is used. Different default from read_table encoding : string, optional a string representing the encoding to use if the contents are non-ascii, for python versions prior to 3 Returns y : Series pandas.Series.load classmethod Series.load(path) pandas.Series.save Series.save(path) pandas.Series.to_csv Series.to_csv(path, index=True, sep=, , na_rep=, oat_format=None, header=False, index_label=None, mode=w, nanRep=None, encoding=None) Write Series to a comma-separated values (csv) le Parameters path : string le path or le handle / StringIO na_rep : string, default Missing data representation oat_format : string, default None Format string for oating point numbers header : boolean, default False Write out series name index : boolean, default True Write row names (index) index_label : string or sequence, default None Column label for index column(s) if desired. If None is given, and header and index are True, then the index names are used. A sequence should be given if the DataFrame uses MultiIndex.
21.2. Series
333
mode : Python write mode, default w sep : character, default , Field delimiter for the output le. encoding : string, optional a string representing the encoding to use if the contents are non-ascii, for python versions prior to 3 pandas.Series.to_dict Series.to_dict() Convert Series to {label -> value} dict Returns value_dict : dict pandas.Series.to_sparse Series.to_sparse(kind=block, ll_value=None) Convert Series to SparseSeries Parameters kind : {block, integer} ll_value : oat, defaults to NaN (missing) Returns sp : SparseSeries pandas.Series.to_string Series.to_string(buf=None, na_rep=NaN, oat_format=None, name=False) Render a string representation of the Series Parameters buf : StringIO-like, optional buffer to write to na_rep : string, optional string representation of NAN to use, default NaN oat_format : one-parameter function, optional formatter function to apply to columns elements if they are oats default None length : boolean, default False Add the Series length name : boolean, default False Add the Series name (which may be None) Returns formatted : string (if not buffer passed) nanRep=None, length=False,
334
21.3 DataFrame
21.3.1 Attributes and underlying data
Axes index: row labels columns: column labels DataFrame.as_matrix([columns]) DataFrame.dtypes DataFrame.get_dtype_counts() DataFrame.values DataFrame.axes DataFrame.ndim DataFrame.shape Convert the frame to its Numpy-array matrix representation. Columns
pandas.DataFrame.as_matrix DataFrame.as_matrix(columns=None) Convert the frame to its Numpy-array matrix representation. Columns are presented in sorted order unless a specic list of columns is provided. Parameters columns : array-like Specic column order Returns values : ndarray If the DataFrame is heterogeneous and contains booleans or objects, the result will be of dtype=object pandas.DataFrame.dtypes DataFrame.dtypes pandas.DataFrame.get_dtype_counts DataFrame.get_dtype_counts() pandas.DataFrame.values DataFrame.values Convert the frame to its Numpy-array matrix representation. Columns are presented in sorted order unless a specic list of columns is provided. Parameters columns : array-like Specic column order Returns values : ndarray If the DataFrame is heterogeneous and contains booleans or objects, the result will be of dtype=object 21.3. DataFrame 335
Two-dimensional size-mutable, potentially heterogeneous tabular data structu Cast object to input numpy.dtype Attempt to infer better dtype for object columns Make a copy of this object
pandas.DataFrame.__init__ DataFrame.__init__(data=None, index=None, columns=None, dtype=None, copy=False) Two-dimensional size-mutable, potentially heterogeneous tabular data structure with labeled axes (rows and columns). Arithmetic operations align on both row and column labels. Can be thought of as a dict-like container for Series objects. The primary pandas data structure Parameters data : numpy ndarray (structured or homogeneous), dict, or DataFrame Dict can contain Series, arrays, constants, or list-like objects index : Index or array-like Index to use for resulting frame. Will default to np.arange(n) if no indexing information part of input data and no index provided columns : Index or array-like Will default to np.arange(n) if not column labels provided dtype : dtype, default None Data type to force, otherwise infer copy : boolean, default False Copy data from inputs. Only affects DataFrame / 2d ndarray input See Also: DataFrame.from_records constructor from tuples, also record arrays DataFrame.from_dict from dicts of Series, arrays, or dicts DataFrame.from_csv from CSV les DataFrame.from_items from sequence of (key, value) pairs
336
read_csv
Examples >>> >>> >>> >>> ... d = {col1: ts1, col2: ts2} df = DataFrame(data=d, index=index) df2 = DataFrame(np.random.randn(10, 5)) df3 = DataFrame(np.random.randn(10, 5), columns=[a, b, c, d, e])
pandas.DataFrame.astype DataFrame.astype(dtype) Cast object to input numpy.dtype Parameters dtype : numpy.dtype or Python type Returns casted : type of caller pandas.DataFrame.convert_objects DataFrame.convert_objects() Attempt to infer better dtype for object columns Returns converted : DataFrame pandas.DataFrame.copy DataFrame.copy(deep=True) Make a copy of this object Parameters deep : boolean, default True Make a deep copy, i.e. also copy data Returns copy : type of caller
21.3. DataFrame
337
pandas.DataFrame.head DataFrame.head(n=5) Returns rst n rows of DataFrame pandas.DataFrame.ix DataFrame.ix pandas.DataFrame.insert DataFrame.insert(loc, column, value) Insert column into DataFrame at specied location. Raises Exception if column is already contained in the DataFrame Parameters loc : int Must have 0 <= loc <= len(columns) column : object value : int, Series, or array-like pandas.DataFrame.__iter__ DataFrame.__iter__() Iterate over columns of the frame. pandas.DataFrame.iteritems DataFrame.iteritems() Iterator over (column, series) pairs pandas.DataFrame.iterrows DataFrame.iterrows() Iterate over rows of DataFrame as (index, Series) pairs pandas.DataFrame.itertuples DataFrame.itertuples(index=True) Iterate over rows of DataFrame as tuples, with index value as rst element of the tuple pandas.DataFrame.lookup DataFrame.lookup(row_labels, col_labels) Label-based fancy indexing function for DataFrame. Given equal-length arrays of row and column labels, return an array of the values corresponding to each (row, col) pair. Parameters row_labels : sequence col_labels : sequence
338
Notes
Akin to result = [] for row, col in zip(row_labels, col_labels): result.append(df.get_value(row, col)) pandas.DataFrame.pop DataFrame.pop(item) Return column and drop from frame. Raise KeyError if not found. Returns column : Series pandas.DataFrame.tail DataFrame.tail(n=5) Returns last n rows of DataFrame pandas.DataFrame.xs DataFrame.xs(key, axis=0, level=None, copy=True) Returns a cross-section (row(s) or column(s)) from the DataFrame. Defaults to cross-section on the rows (axis=0). Parameters key : object Some label contained in the index, or partially in a MultiIndex axis : int, default 0 Axis to retrieve cross-section on level : object, defaults to rst n levels (n=1 or len(key)) In case of a key partially contained in a MultiIndex, indicate which levels are used. Levels can be referred by label or position. copy : boolean, default True Whether to make a copy of the data Returns xs : Series or DataFrame
Examples >>> df A B C a 4 5 2 b 4 0 9 c 9 7 3 >>> df.xs(a) A 4 B 5 C 2 Name: a
21.3. DataFrame
339
>>> df.xs(C, axis=1) a 2 b 9 c 3 Name: C >>> s = df.xs(a, copy=False) >>> s[A] = 100 >>> df A B C a 100 5 2 b 4 0 9 c 9 7 3 >>> df A B C D first second third bar one 1 4 1 8 9 two 1 7 5 5 0 baz one 1 6 6 8 0 three 2 5 3 5 3 >>> df.xs((baz, three)) A B C D third 2 5 3 5 3 >>> df.xs(one, level=1) A B C D first third bar 1 4 1 8 9 baz 1 6 6 8 0 >>> df.xs((baz, 2), level=[0, third]) A B C D second three 5 3 5 3
Binary operator add with support to substitute a ll_value for missing data in Binary operator divide with support to substitute a ll_value for missing data in Binary operator multiply with support to substitute a ll_value for missing data Binary operator subtract with support to substitute a ll_value for missing data i Binary operator radd with support to substitute a ll_value for missing data in Binary operator rdivide with support to substitute a ll_value for missing data in Binary operator rmultiply with support to substitute a ll_value for missing data Binary operator rsubtract with support to substitute a ll_value for missing data Add two DataFrame objects and do not propagate NaN values, so if for a Add two DataFrame objects and do not propagate Combine two DataFrame objects and default to non-null values in frame Multiply two DataFrame objects and do not propagate NaN values, so if
pandas.DataFrame.add DataFrame.add(other, axis=columns, level=None, ll_value=None) Binary operator add with support to substitute a ll_value for missing data in one of the inputs
340
Parameters other : Series, DataFrame, or constant axis : {0, 1, index, columns} For Series input, axis to match Series index on ll_value : None or oat value, default None Fill missing (NaN) values with this value. If both DataFrame locations are missing, the result will be missing level : int or name Broadcast across a level, matching Index values on the passed MultiIndex level Returns result : DataFrame
Notes
Mismatched indices will be unioned together pandas.DataFrame.div DataFrame.div(other, axis=columns, level=None, ll_value=None) Binary operator divide with support to substitute a ll_value for missing data in one of the inputs Parameters other : Series, DataFrame, or constant axis : {0, 1, index, columns} For Series input, axis to match Series index on ll_value : None or oat value, default None Fill missing (NaN) values with this value. If both DataFrame locations are missing, the result will be missing level : int or name Broadcast across a level, matching Index values on the passed MultiIndex level Returns result : DataFrame
Notes
Mismatched indices will be unioned together pandas.DataFrame.mul DataFrame.mul(other, axis=columns, level=None, ll_value=None) Binary operator multiply with support to substitute a ll_value for missing data in one of the inputs Parameters other : Series, DataFrame, or constant axis : {0, 1, index, columns} For Series input, axis to match Series index on ll_value : None or oat value, default None
21.3. DataFrame
341
Fill missing (NaN) values with this value. If both DataFrame locations are missing, the result will be missing level : int or name Broadcast across a level, matching Index values on the passed MultiIndex level Returns result : DataFrame
Notes
Mismatched indices will be unioned together pandas.DataFrame.sub DataFrame.sub(other, axis=columns, level=None, ll_value=None) Binary operator subtract with support to substitute a ll_value for missing data in one of the inputs Parameters other : Series, DataFrame, or constant axis : {0, 1, index, columns} For Series input, axis to match Series index on ll_value : None or oat value, default None Fill missing (NaN) values with this value. If both DataFrame locations are missing, the result will be missing level : int or name Broadcast across a level, matching Index values on the passed MultiIndex level Returns result : DataFrame
Notes
Mismatched indices will be unioned together pandas.DataFrame.radd DataFrame.radd(other, axis=columns, level=None, ll_value=None) Binary operator radd with support to substitute a ll_value for missing data in one of the inputs Parameters other : Series, DataFrame, or constant axis : {0, 1, index, columns} For Series input, axis to match Series index on ll_value : None or oat value, default None Fill missing (NaN) values with this value. If both DataFrame locations are missing, the result will be missing level : int or name Broadcast across a level, matching Index values on the passed MultiIndex level Returns result : DataFrame
342
Notes
Mismatched indices will be unioned together pandas.DataFrame.rdiv DataFrame.rdiv(other, axis=columns, level=None, ll_value=None) Binary operator rdivide with support to substitute a ll_value for missing data in one of the inputs Parameters other : Series, DataFrame, or constant axis : {0, 1, index, columns} For Series input, axis to match Series index on ll_value : None or oat value, default None Fill missing (NaN) values with this value. If both DataFrame locations are missing, the result will be missing level : int or name Broadcast across a level, matching Index values on the passed MultiIndex level Returns result : DataFrame
Notes
Mismatched indices will be unioned together pandas.DataFrame.rmul DataFrame.rmul(other, axis=columns, level=None, ll_value=None) Binary operator rmultiply with support to substitute a ll_value for missing data in one of the inputs Parameters other : Series, DataFrame, or constant axis : {0, 1, index, columns} For Series input, axis to match Series index on ll_value : None or oat value, default None Fill missing (NaN) values with this value. If both DataFrame locations are missing, the result will be missing level : int or name Broadcast across a level, matching Index values on the passed MultiIndex level Returns result : DataFrame
Notes
21.3. DataFrame
343
pandas.DataFrame.rsub DataFrame.rsub(other, axis=columns, level=None, ll_value=None) Binary operator rsubtract with support to substitute a ll_value for missing data in one of the inputs Parameters other : Series, DataFrame, or constant axis : {0, 1, index, columns} For Series input, axis to match Series index on ll_value : None or oat value, default None Fill missing (NaN) values with this value. If both DataFrame locations are missing, the result will be missing level : int or name Broadcast across a level, matching Index values on the passed MultiIndex level Returns result : DataFrame
Notes
Mismatched indices will be unioned together pandas.DataFrame.combine DataFrame.combine(other, func, ll_value=None) Add two DataFrame objects and do not propagate NaN values, so if for a (column, time) one frame is missing a value, it will default to the other frames value (which might be NaN as well) Parameters other : DataFrame func : function ll_value : scalar value Returns result : DataFrame pandas.DataFrame.combineAdd DataFrame.combineAdd(other) Add two DataFrame objects and do not propagate NaN values, so if for a (column, time) one frame is missing a value, it will default to the other frames value (which might be NaN as well) Parameters other : DataFrame Returns DataFrame : pandas.DataFrame.combine_rst DataFrame.combine_first(other) Combine two DataFrame objects and default to non-null values in frame calling the method. Result index will be the union of the two indexes Parameters other : DataFrame Returns combined : DataFrame 344 Chapter 21. API Reference
Examples >>> a.combine_first(b) as values prioritized, use values from b to fill holes
pandas.DataFrame.combineMult DataFrame.combineMult(other) Multiply two DataFrame objects and do not propagate NaN values, so if for a (column, time) one frame is missing a value, it will default to the other frames value (which might be NaN as well) Parameters other : DataFrame Returns DataFrame :
pandas.DataFrame.apply DataFrame.apply(func, axis=0, broadcast=False, raw=False, args=(), **kwds) Applies function along input axis of DataFrame. Objects passed to functions are Series objects having index either the DataFrames index (axis=0) or the columns (axis=1). Return type depends on whether passed function aggregates Parameters func : function Function to apply to each column axis : {0, 1} 0 : apply function to each column 1 : apply function to each row broadcast : bool, default False For aggregation functions, return object of same size with values propagated raw : boolean, default False If False, convert each row or column into a Series. If raw=True the passed function will receive ndarray objects instead. If you are just applying a NumPy reduction function this will achieve much better performance args : tuple Positional arguments to pass to function in addition to the array/series Additional keyword arguments will be passed as keywords to the function : Returns applied : Series or DataFrame
21.3. DataFrame
345
Notes
pandas.DataFrame.applymap DataFrame.applymap(func) Apply a function to a DataFrame that is intended to operate elementwise, i.e. like doing map(func, series) for each series in the DataFrame Parameters func : function Python function, returns a single value from a single value Returns applied : DataFrame pandas.DataFrame.groupby DataFrame.groupby(by=None, axis=0, level=None, as_index=True, sort=True, group_keys=True) Group series using mapper (dict or key function, apply given function to group, return result as series) or by a series of columns Parameters by : mapping function / list of functions, dict, Series, or tuple / list of column names. Called on each element of the object index to determine the groups. If a dict or Series is passed, the Series or dict VALUES will be used to determine the groups axis : int, default 0 level : int, level name, or sequence of such, default None If the axis is a MultiIndex (hierarchical), group by a particular level or levels as_index : boolean, default True For aggregated output, return object with group labels as the index. Only relevant for DataFrame input. as_index=False is effectively SQL-style grouped output sort : boolean, default True Sort group keys. Get better performance by turning this off group_keys : boolean, default True When calling apply, add group keys to index to identify pieces Returns GroupBy object :
346
Examples
# DataFrame result >>> data.groupby(func, axis=0).mean() # DataFrame result >>> data.groupby([col1, col2])[col3].mean() # DataFrame with hierarchical index >>> data.groupby([col1, col2]).mean()
Return an object with absolute value taken. Return whether any element is True over requested axis. Trim values at input threshold(s) Trim values below threshold Trim values above threshold Compute pairwise correlation of columns, excluding NA/null values Compute pairwise correlation between rows or columns of two DataFrame Return Series with number of non-NA/null observations over requested Compute pairwise covariance of columns, excluding NA/null values Return DataFrame of cumulative max over requested axis. Return DataFrame of cumulative min over requested axis. Return cumulative product over requested axis as DataFrame Return DataFrame of cumulative sums over requested axis. Generate various summary statistics of each column, excluding 1st discrete difference of object Return unbiased kurtosis over requested axis. Return mean absolute deviation over requested axis. Return maximum over requested axis. Return mean over requested axis. Return median over requested axis. Return minimum over requested axis. Percent change over given number of periods Return product over requested axis. Return values at the given quantile over requested axis, a la Compute numerical data ranks (1 through n) along axis. Return unbiased skewness over requested axis. Return sum over requested axis. Return standard deviation over requested axis. Return variance over requested axis.
pandas.DataFrame.abs DataFrame.abs() Return an object with absolute value taken. Only applicable to objects that are all numeric Returns abs: type of caller : pandas.DataFrame.any DataFrame.any(axis=0, bool_only=None, skipna=True, level=None) Return whether any element is True over requested axis. %(na_action)s
21.3. DataFrame
347
Parameters axis : {0, 1} 0 for row-wise, 1 for column-wise skipna : boolean, default True Exclude NA/null values. If an entire row/column is NA, the result will be NA level : int, default None If the axis is a MultiIndex (hierarchical), count along a particular level, collapsing into a DataFrame bool_only : boolean, default None Only include boolean data. Returns any : Series (or DataFrame if level specied) pandas.DataFrame.clip DataFrame.clip(upper=None, lower=None) Trim values at input threshold(s) Parameters lower : oat, default None upper : oat, default None Returns clipped : DataFrame pandas.DataFrame.clip_lower DataFrame.clip_lower(threshold) Trim values below threshold Returns clipped : DataFrame pandas.DataFrame.clip_upper DataFrame.clip_upper(threshold) Trim values above threshold Returns clipped : DataFrame pandas.DataFrame.corr DataFrame.corr(method=pearson) Compute pairwise correlation of columns, excluding NA/null values Parameters method : {pearson, kendall, spearman} pearson : standard correlation coefcient kendall : Kendall Tau correlation coefcient spearman : Spearman rank correlation Returns y : DataFrame
348
pandas.DataFrame.corrwith DataFrame.corrwith(other, axis=0, drop=False) Compute pairwise correlation between rows or columns of two DataFrame objects. Parameters other : DataFrame axis : {0, 1} 0 to compute column-wise, 1 for row-wise drop : boolean, default False Drop missing indices from result, default returns union of all Returns correls : Series pandas.DataFrame.count DataFrame.count(axis=0, level=None, numeric_only=False) Return Series with number of non-NA/null observations over requested axis. Works with non-oating point data as well (detects NaN and None) Parameters axis : {0, 1} 0 for row-wise, 1 for column-wise level : int, default None If the axis is a MultiIndex (hierarchical), count along a particular level, collapsing into a DataFrame numeric_only : boolean, default False Include only oat, int, boolean data Returns count : Series (or DataFrame if level specied) pandas.DataFrame.cov DataFrame.cov() Compute pairwise covariance of columns, excluding NA/null values Returns y : DataFrame y contains the covariance matrix of the DataFrames time series. : The covariance is normalized by N-1 (unbiased estimator). : pandas.DataFrame.cummax DataFrame.cummax(axis=None, skipna=True) Return DataFrame of cumulative max over requested axis. Parameters axis : {0, 1} 0 for row-wise, 1 for column-wise skipna : boolean, default True Exclude NA/null values. If an entire row/column is NA, the result will be NA
21.3. DataFrame
349
Returns y : DataFrame pandas.DataFrame.cummin DataFrame.cummin(axis=None, skipna=True) Return DataFrame of cumulative min over requested axis. Parameters axis : {0, 1} 0 for row-wise, 1 for column-wise skipna : boolean, default True Exclude NA/null values. If an entire row/column is NA, the result will be NA Returns y : DataFrame pandas.DataFrame.cumprod DataFrame.cumprod(axis=None, skipna=True) Return cumulative product over requested axis as DataFrame Parameters axis : {0, 1} 0 for row-wise, 1 for column-wise skipna : boolean, default True Exclude NA/null values. If an entire row/column is NA, the result will be NA Returns y : DataFrame pandas.DataFrame.cumsum DataFrame.cumsum(axis=None, skipna=True) Return DataFrame of cumulative sums over requested axis. Parameters axis : {0, 1} 0 for row-wise, 1 for column-wise skipna : boolean, default True Exclude NA/null values. If an entire row/column is NA, the result will be NA Returns y : DataFrame pandas.DataFrame.describe DataFrame.describe(percentile_width=50) Generate various summary statistics of each column, excluding NaN values. These include: count, mean, std, min, max, and lower%/50%/upper% percentiles Parameters percentile_width : oat, optional width of the desired uncertainty interval, default is 50, which corresponds to lower=25, upper=75 Returns DataFrame of summary statistics :
350
pandas.DataFrame.diff DataFrame.diff(periods=1) 1st discrete difference of object Parameters periods : int, default 1 Periods to shift for forming difference Returns diffed : DataFrame pandas.DataFrame.kurt DataFrame.kurt(axis=0, skipna=True, level=None) Return unbiased kurtosis over requested axis. NA/null values are excluded Parameters axis : {0, 1} 0 for row-wise, 1 for column-wise skipna : boolean, default True Exclude NA/null values. If an entire row/column is NA, the result will be NA level : int, default None If the axis is a MultiIndex (hierarchical), count along a particular level, collapsing into a DataFrame Returns kurt : Series (or DataFrame if level specied) pandas.DataFrame.mad DataFrame.mad(axis=0, skipna=True, level=None) Return mean absolute deviation over requested axis. NA/null values are excluded Parameters axis : {0, 1} 0 for row-wise, 1 for column-wise skipna : boolean, default True Exclude NA/null values. If an entire row/column is NA, the result will be NA level : int, default None If the axis is a MultiIndex (hierarchical), count along a particular level, collapsing into a DataFrame Returns mad : Series (or DataFrame if level specied) pandas.DataFrame.max DataFrame.max(axis=0, skipna=True, level=None) Return maximum over requested axis. NA/null values are excluded Parameters axis : {0, 1} 0 for row-wise, 1 for column-wise skipna : boolean, default True
21.3. DataFrame
351
Exclude NA/null values. If an entire row/column is NA, the result will be NA level : int, default None If the axis is a MultiIndex (hierarchical), count along a particular level, collapsing into a DataFrame Returns max : Series (or DataFrame if level specied) pandas.DataFrame.mean DataFrame.mean(axis=0, skipna=True, level=None) Return mean over requested axis. NA/null values are excluded Parameters axis : {0, 1} 0 for row-wise, 1 for column-wise skipna : boolean, default True Exclude NA/null values. If an entire row/column is NA, the result will be NA level : int, default None If the axis is a MultiIndex (hierarchical), count along a particular level, collapsing into a DataFrame Returns mean : Series (or DataFrame if level specied) pandas.DataFrame.median DataFrame.median(axis=0, skipna=True, level=None) Return median over requested axis. NA/null values are excluded Parameters axis : {0, 1} 0 for row-wise, 1 for column-wise skipna : boolean, default True Exclude NA/null values. If an entire row/column is NA, the result will be NA level : int, default None If the axis is a MultiIndex (hierarchical), count along a particular level, collapsing into a DataFrame Returns median : Series (or DataFrame if level specied) pandas.DataFrame.min DataFrame.min(axis=0, skipna=True, level=None) Return minimum over requested axis. NA/null values are excluded Parameters axis : {0, 1} 0 for row-wise, 1 for column-wise skipna : boolean, default True Exclude NA/null values. If an entire row/column is NA, the result will be NA level : int, default None 352 Chapter 21. API Reference
If the axis is a MultiIndex (hierarchical), count along a particular level, collapsing into a DataFrame Returns min : Series (or DataFrame if level specied) pandas.DataFrame.pct_change DataFrame.pct_change(periods=1, ll_method=pad, limit=None, freq=None, **kwds) Percent change over given number of periods Parameters periods : int, default 1 Periods to shift for forming percent change ll_method : str, default pad How to handle NAs before computing percent changes limit : int, default None The number of consecutive NAs to ll before stopping freq : DateOffset, timedelta, or offset alias string, optional Increment to use from time series API (e.g. M or BDay()) Returns chg : Series or DataFrame pandas.DataFrame.prod DataFrame.prod(axis=0, skipna=True, level=None) Return product over requested axis. NA/null values are treated as 1 Parameters axis : {0, 1} 0 for row-wise, 1 for column-wise skipna : boolean, default True Exclude NA/null values. If an entire row/column is NA, the result will be NA level : int, default None If the axis is a MultiIndex (hierarchical), count along a particular level, collapsing into a DataFrame Returns product : Series (or DataFrame if level specied) pandas.DataFrame.quantile DataFrame.quantile(q=0.5, axis=0) Return values at the given quantile over requested axis, a la scoreatpercentile in scipy.stats Parameters q : quantile, default 0.5 (50% quantile) 0 <= q <= 1 axis : {0, 1} 0 for row-wise, 1 for column-wise Returns quantiles : Series
21.3. DataFrame
353
pandas.DataFrame.rank DataFrame.rank(axis=0, numeric_only=None, method=average, na_option=keep, ascending=True) Compute numerical data ranks (1 through n) along axis. Equal values are assigned a rank that is the average of the ranks of those values Parameters axis : {0, 1}, default 0 Ranks over columns (0) or rows (1) numeric_only : boolean, default None Include only oat, int, boolean data method : {average, min, max, rst} average: average rank of group min: lowest rank in group max: highest rank in group rst: ranks assigned in order they appear in the array na_option : {keep, top, bottom} keep: leave NA values where they are top: smallest rank if ascending bottom: smallest rank if descending ascending : boolean, default True False for ranks by high (1) to low (N) Returns ranks : DataFrame pandas.DataFrame.skew DataFrame.skew(axis=0, skipna=True, level=None) Return unbiased skewness over requested axis. NA/null values are excluded Parameters axis : {0, 1} 0 for row-wise, 1 for column-wise skipna : boolean, default True Exclude NA/null values. If an entire row/column is NA, the result will be NA level : int, default None If the axis is a MultiIndex (hierarchical), count along a particular level, collapsing into a DataFrame Returns skew : Series (or DataFrame if level specied) pandas.DataFrame.sum DataFrame.sum(axis=0, numeric_only=None, skipna=True, level=None) Return sum over requested axis. NA/null values are excluded Parameters axis : {0, 1} 0 for row-wise, 1 for column-wise skipna : boolean, default True Exclude NA/null values. If an entire row/column is NA, the result will be NA level : int, default None
354
If the axis is a MultiIndex (hierarchical), count along a particular level, collapsing into a DataFrame numeric_only : boolean, default None Include only oat, int, boolean data. If None, will attempt to use everything, then use only numeric data Returns sum : Series (or DataFrame if level specied) pandas.DataFrame.std DataFrame.std(axis=0, skipna=True, level=None, ddof=1) Return standard deviation over requested axis. NA/null values are excluded Parameters axis : {0, 1} 0 for row-wise, 1 for column-wise skipna : boolean, default True Exclude NA/null values. If an entire row/column is NA, the result will be NA level : int, default None If the axis is a MultiIndex (hierarchical), count along a particular level, collapsing into a DataFrame Returns std : Series (or DataFrame if level specied) Normalized by N-1 (unbiased estimator). pandas.DataFrame.var DataFrame.var(axis=0, skipna=True, level=None, ddof=1) Return variance over requested axis. NA/null values are excluded Parameters axis : {0, 1} 0 for row-wise, 1 for column-wise skipna : boolean, default True Exclude NA/null values. If an entire row/column is NA, the result will be NA level : int, default None If the axis is a MultiIndex (hierarchical), count along a particular level, collapsing into a DataFrame Returns var : Series (or DataFrame if level specied) Normalized by N-1 (unbiased estimator).
21.3. DataFrame
355
Table 21.28 continued from previous page DataFrame.drop(labels[, axis, level]) Return new object with labels in requested axis removed DataFrame.drop_duplicates([cols, take_last, ...]) Return DataFrame with duplicate rows removed, optionally only DataFrame.duplicated([cols, take_last]) Return boolean Series denoting duplicate rows, optionally only DataFrame.filter([items, like, regex]) Restrict frames columns to set of items or wildcard DataFrame.first(offset) Convenience method for subsetting initial periods of time series data DataFrame.head([n]) Returns rst n rows of DataFrame DataFrame.idxmax([axis, skipna]) Return index of rst occurrence of maximum over requested axis. DataFrame.idxmin([axis, skipna]) Return index of rst occurrence of minimum over requested axis. DataFrame.last(offset) Convenience method for subsetting nal periods of time series data DataFrame.reindex([index, columns, method, ...]) Conform DataFrame to new index with optional lling logic, placing DataFrame.reindex_axis(labels[, axis, ...]) Conform DataFrame to new index with optional lling logic, placing DataFrame.reindex_like(other[, method, ...]) Reindex DataFrame to match indices of another DataFrame, optionally DataFrame.rename([index, columns, copy, inplace]) Alter index and / or columns using input function or functions. DataFrame.reset_index([level, drop, ...]) For DataFrame with multi-level index, return new DataFrame with DataFrame.select(crit[, axis]) Return data corresponding to axis labels matching criteria DataFrame.set_index(keys[, drop, append, ...]) Set the DataFrame index (row labels) using one or more existing DataFrame.tail([n]) Returns last n rows of DataFrame DataFrame.take(indices[, axis]) Analogous to ndarray.take, return DataFrame corresponding to requested DataFrame.truncate([before, after, copy]) Function truncate a sorted DataFrame / Series before and/or after
pandas.DataFrame.add_prex DataFrame.add_prefix(prex) Concatenate prex string with panel items names. Parameters prex : string Returns with_prex : type of caller pandas.DataFrame.add_sufx DataFrame.add_suffix(sufx) Concatenate sufx string with panel items names Parameters sufx : string Returns with_sufx : type of caller pandas.DataFrame.align DataFrame.align(other, join=outer, axis=None, level=None, copy=True, ll_value=nan, method=None, limit=None, ll_axis=0) Align two DataFrame object on their index and columns with the specied join method for each axis Index Parameters other : DataFrame or Series join : {outer, inner, left, right}, default outer axis : {0, 1, None}, default None Align on index (0), columns (1), or both (None) level : int or name Broadcast across a level, matching Index values on the passed MultiIndex level
356
copy : boolean, default True Always returns new objects. If copy=False and no reindexing is required then original objects are returned. ll_value : scalar, default np.NaN Value to use for missing values. Defaults to NaN, but can be any compatible value method : str, default None limit : int, default None ll_axis : {0, 1}, default 0 Filling axis, method and limit Returns (left, right) : (DataFrame, type of other) Aligned objects pandas.DataFrame.drop DataFrame.drop(labels, axis=0, level=None) Return new object with labels in requested axis removed Parameters labels : array-like axis : int level : int or name, default None For MultiIndex Returns dropped : type of caller pandas.DataFrame.drop_duplicates DataFrame.drop_duplicates(cols=None, take_last=False, inplace=False) Return DataFrame with duplicate rows removed, optionally only considering certain columns Parameters cols : column label or sequence of labels, optional Only consider certain columns for identifying duplicates, by default use all of the columns take_last : boolean, default False Take the last observed row in a row. Defaults to the rst row inplace : boolean, default False Whether to drop duplicates in place or to return a copy Returns deduplicated : DataFrame pandas.DataFrame.duplicated DataFrame.duplicated(cols=None, take_last=False) Return boolean Series denoting duplicate rows, optionally only considering certain columns Parameters cols : column label or sequence of labels, optional
21.3. DataFrame
357
Only consider certain columns for identifying duplicates, by default use all of the columns take_last : boolean, default False Take the last observed row in a row. Defaults to the rst row Returns duplicated : Series pandas.DataFrame.lter DataFrame.filter(items=None, like=None, regex=None) Restrict frames columns to set of items or wildcard Parameters items : list-like List of columns to restrict to (must not all be present) like : string Keep columns where arg in col == True regex : string (regular expression) Keep columns with re.search(regex, col) == True Returns DataFrame with ltered columns :
Notes
Arguments are mutually exclusive, but this is not checked for pandas.DataFrame.rst DataFrame.first(offset) Convenience method for subsetting initial periods of time series data based on a date offset Parameters offset : string, DateOffset, dateutil.relativedelta Returns subset : type of caller
Examples
ts.last(10D) -> First 10 days pandas.DataFrame.head DataFrame.head(n=5) Returns rst n rows of DataFrame
358
pandas.DataFrame.idxmax DataFrame.idxmax(axis=0, skipna=True) Return index of rst occurrence of maximum over requested axis. NA/null values are excluded. Parameters axis : {0, 1} 0 for row-wise, 1 for column-wise skipna : boolean, default True Exclude NA/null values. If an entire row/column is NA, the result will be rst index. Returns idxmax : Series pandas.DataFrame.idxmin DataFrame.idxmin(axis=0, skipna=True) Return index of rst occurrence of minimum over requested axis. NA/null values are excluded. Parameters axis : {0, 1} 0 for row-wise, 1 for column-wise skipna : boolean, default True Exclude NA/null values. If an entire row/column is NA, the result will be NA Returns idxmin : Series pandas.DataFrame.last DataFrame.last(offset) Convenience method for subsetting nal periods of time series data based on a date offset Parameters offset : string, DateOffset, dateutil.relativedelta Returns subset : type of caller
Examples
ts.last(5M) -> Last 5 months pandas.DataFrame.reindex DataFrame.reindex(index=None, columns=None, method=None, level=None, ll_value=nan, limit=None, copy=True) Conform DataFrame to new index with optional lling logic, placing NA/NaN in locations having no value in the previous index. A new object is produced unless the new index is equivalent to the current one and copy=False Parameters index : array-like, optional New labels / index to conform to. Preferably an Index object to avoid duplicating data columns : array-like, optional Same usage as index argument method : {backll, bll, pad, fll, None}, default None 21.3. DataFrame 359
Method to use for lling holes in reindexed DataFrame pad / fll: propagate last valid observation forward to next valid backll / bll: use NEXT valid observation to ll gap copy : boolean, default True Return a new object, even if the passed indexes are the same level : int or name Broadcast across a level, matching Index values on the passed MultiIndex level ll_value : scalar, default np.NaN Value to use for missing values. Defaults to NaN, but can be any compatible value limit : int, default None Maximum size gap to forward or backward ll Returns reindexed : same type as calling instance
Examples >>> df.reindex(index=[date1, date2, date3], columns=[A, B, C])
pandas.DataFrame.reindex_axis DataFrame.reindex_axis(labels, axis=0, method=None, level=None, copy=True, limit=None, ll_value=nan) Conform DataFrame to new index with optional lling logic, placing NA/NaN in locations having no value in the previous index. A new object is produced unless the new index is equivalent to the current one and copy=False Parameters index : array-like, optional New labels / index to conform to. Preferably an Index object to avoid duplicating data axis : {0, 1} 0 -> index (rows) 1 -> columns method : {backll, bll, pad, fll, None}, default None Method to use for lling holes in reindexed DataFrame pad / fll: propagate last valid observation forward to next valid backll / bll: use NEXT valid observation to ll gap copy : boolean, default True Return a new object, even if the passed indexes are the same level : int or name Broadcast across a level, matching Index values on the passed MultiIndex level limit : int, default None Maximum size gap to forward or backward ll Returns reindexed : same type as calling instance See Also: DataFrame.reindex, DataFrame.reindex_like
360
pandas.DataFrame.reindex_like DataFrame.reindex_like(other, method=None, copy=True, limit=None) Reindex DataFrame to match indices of another DataFrame, optionally with lling logic Parameters other : DataFrame method : string or None copy : boolean, default True limit : int, default None Maximum size gap to forward or backward ll Returns reindexed : DataFrame
Notes
Like calling s.reindex(index=other.index, columns=other.columns, method=...) pandas.DataFrame.rename DataFrame.rename(index=None, columns=None, copy=True, inplace=False) Alter index and / or columns using input function or functions. Function / dict values must be unique (1-to-1). Labels not contained in a dict / Series will be left as-is. Parameters index : dict-like or function, optional Transformation to apply to index values columns : dict-like or function, optional Transformation to apply to column values copy : boolean, default True Also copy underlying data inplace : boolean, default False Whether to return a new DataFrame. If True then value of copy is ignored. Returns renamed : DataFrame (new object) See Also: Series.rename pandas.DataFrame.reset_index DataFrame.reset_index(level=None, drop=False, inplace=False, col_level=0, col_ll=) For DataFrame with multi-level index, return new DataFrame with labeling information in the columns under the index names, defaulting to level_0, level_1, etc. if any are None. For a standard index, the index name will be used (if set), otherwise a default index or level_0 (if index is already taken) will be used. 21.3. DataFrame 361
Parameters level : int, str, tuple, or list, default None Only remove the given levels from the index. Removes all levels by default drop : boolean, default False Do not try to insert index into dataframe columns. This resets the index to the default integer index. inplace : boolean, default False Modify the DataFrame in place (do not create a new object) col_level : int or str, default 0 If the columns have multiple levels, determines which level the labels are inserted into. By default it is inserted into the rst level. col_ll : object, default If the columns have multiple levels, determines how the other levels are named. If None then the index name is repeated. Returns resetted : DataFrame pandas.DataFrame.select DataFrame.select(crit, axis=0) Return data corresponding to axis labels matching criteria Parameters crit : function To be called on each index (label). Should return True or False axis : int Returns selection : type of caller pandas.DataFrame.set_index DataFrame.set_index(keys, drop=True, append=False, inplace=False, verify_integrity=False) Set the DataFrame index (row labels) using one or more existing columns. By default yields a new object. Parameters keys : column label or list of column labels / arrays drop : boolean, default True Delete columns to be used as the new index append : boolean, default False Whether to append columns to existing index inplace : boolean, default False Modify the DataFrame in place (do not create a new object) verify_integrity : boolean, default False Check the new index for duplicates. Otherwise defer the check until necessary. Setting to False will improve the performance of this method Returns dataframe : DataFrame
362
Examples
indexed_df = df.set_index([A, B]) indexed_df2 = df.set_index([A, [0, 1, 2, 0, 1, 2]]) indexed_df3 = df.set_index([[0, 1, 2, 0, 1, 2]]) pandas.DataFrame.tail DataFrame.tail(n=5) Returns last n rows of DataFrame pandas.DataFrame.take DataFrame.take(indices, axis=0) Analogous to ndarray.take, return DataFrame corresponding to requested indices along an axis Parameters indices : list / array of ints axis : {0, 1} Returns taken : DataFrame pandas.DataFrame.truncate DataFrame.truncate(before=None, after=None, copy=True) Function truncate a sorted DataFrame / Series before and/or after some particular dates. Parameters before : date Truncate before date after : date Truncate after date Returns truncated : type of caller
int value : require that many non-NA values subset : array-like Labels along other axis to consider, e.g. if you are dropping rows these would be a list of columns to include Returns dropped : DataFrame pandas.DataFrame.llna DataFrame.fillna(value=None, method=pad, axis=0, inplace=False, limit=None) Fill NA/NaN values using the specied method Parameters method : {backll, bll, pad, fll, None}, default pad Method to use for lling holes in reindexed Series pad / fll: propagate last valid observation forward to next valid backll / bll: use NEXT valid observation to ll gap value : scalar or dict Value to use to ll holes (e.g. 0), alternately a dict of values specifying which value to use for each column (columns not in the dict will not be lled) axis : {0, 1}, default 0 0: ll column-by-column 1: ll row-by-row inplace : boolean, default False If True, ll the DataFrame in place. Note: this will modify any other views on this DataFrame, like if you took a no-copy slice of an existing DataFrame, for example a column in a DataFrame. Returns a reference to the lled object, which is self if inplace=True limit : int, default None Maximum size gap to forward or backward ll Returns lled : DataFrame See Also: reindex, asfreq
Reshape data (produce a pivot table) based on column values. Rearrange index levels using input order. Sort DataFrame either by labels (along either axis) or by the values in Sort DataFrame either by labels (along either axis) or by the values in Sort multilevel index by chosen axis and primary level. Swap levels i and j in a MultiIndex on a particular axis Pivot a level of the (possibly hierarchical) column labels, returning a Pivot a level of the (necessarily hierarchical) index labels, returning Returns a DataFrame with the rows/columns switched. If the DataFrame is Transform long (stacked) format (DataFrame) into wide (3D, Panel) Continued on next page
364
DataFrame.transpose()
Table 21.30 continued from previous page Returns a DataFrame with the rows/columns switched. If the DataFrame is
pandas.DataFrame.delevel DataFrame.delevel(*args, **kwargs) pandas.DataFrame.pivot DataFrame.pivot(index=None, columns=None, values=None) Reshape data (produce a pivot table) based on column values. Uses unique values from index / columns to form axes and return either DataFrame or Panel, depending on whether you request a single value column (DataFrame) or all columns (Panel) Parameters index : string or object Column name to use to make new frames index columns : string or object Column name to use to make new frames columns values : string or object, optional Column name to use for populating new frames values Returns pivoted : DataFrame If no values column specied, will have hierarchically indexed columns
Notes
For ner-tuned control, see hierarchical indexing documentation along with the related stack/unstack methods
Examples >>> df foo 0 one 1 one 2 one 3 two 4 two 5 two
bar A B C A B C
baz 1. 2. 3. 4. 5. 6.
>>> df.pivot(foo, bar, baz) A B C one 1 2 3 two 4 5 6 >>> df.pivot(foo, bar)[baz] A B C one 1 2 3 two 4 5 6
21.3. DataFrame
365
pandas.DataFrame.reorder_levels DataFrame.reorder_levels(order, axis=0) Rearrange index levels using input order. May not drop or duplicate levels Parameters order: list of int representing new level order. : (reference level by number not by key) axis: where to reorder levels : Returns type of caller (new object) : pandas.DataFrame.sort DataFrame.sort(columns=None, column=None, axis=0, ascending=True, inplace=False) Sort DataFrame either by labels (along either axis) or by the values in column(s) Parameters columns : object Column name(s) in frame. Accepts a column name or a list or tuple for a nested sort. ascending : boolean or list, default True Sort ascending vs. descending. Specify list for multiple sort orders axis : {0, 1} Sort index/rows versus columns inplace : boolean, default False Sort the DataFrame without creating a new instance Returns sorted : DataFrame
Examples >>> result = df.sort([A, B], ascending=[1, 0])
pandas.DataFrame.sort_index DataFrame.sort_index(axis=0, by=None, ascending=True, inplace=False) Sort DataFrame either by labels (along either axis) or by the values in a column Parameters axis : {0, 1} Sort index/rows versus columns by : object Column name(s) in frame. Accepts a column name or a list or tuple for a nested sort. ascending : boolean or list, default True Sort ascending vs. descending. Specify list for multiple sort orders inplace : boolean, default False Sort the DataFrame without creating a new instance Returns sorted : DataFrame
366
pandas.DataFrame.sortlevel DataFrame.sortlevel(level=0, axis=0, ascending=True, inplace=False) Sort multilevel index by chosen axis and primary level. Data will be lexicographically sorted by the chosen level followed by the other levels (in order) Parameters level : int axis : {0, 1} ascending : bool, default True inplace : boolean, default False Sort the DataFrame without creating a new instance Returns sorted : DataFrame pandas.DataFrame.swaplevel DataFrame.swaplevel(i, j, axis=0) Swap levels i and j in a MultiIndex on a particular axis Returns swapped : type of caller (new object) pandas.DataFrame.stack DataFrame.stack(level=-1, dropna=True) Pivot a level of the (possibly hierarchical) column labels, returning a DataFrame (or Series in the case of an object with a single level of column labels) having a hierarchical index with a new inner-most level of row labels. Parameters level : int, string, or list of these, default last level Level(s) to stack, can pass level name dropna : boolean, default True Whether to drop rows in the resulting Frame/Series with no valid values Returns stacked : DataFrame or Series
Examples >>> s one two a 1. 3. b 2. 4.
21.3. DataFrame
367
pandas.DataFrame.unstack DataFrame.unstack(level=-1) Pivot a level of the (necessarily hierarchical) index labels, returning a DataFrame having a new level of column labels whose inner-most level consists of the pivoted index labels. If the index is not a MultiIndex, the output will be a Series (the analogue of stack when the columns are not a MultiIndex) Parameters level : int, string, or list of these, default last level Level(s) of index to unstack, can pass level name Returns unstacked : DataFrame or Series
Examples >>> s one a one b two a two b
1. 2. 3. 4.
>>> s.unstack(level=-1) a b one 1. 2. two 3. 4. >>> df = s.unstack(level=0) >>> df one two a 1. 2. b 3. 4. >>> df.unstack() one a 1. b 3. two a 2. b 4.
pandas.DataFrame.T DataFrame.T Returns a DataFrame with the rows/columns switched. If the DataFrame is homogeneously-typed, the data is not copied pandas.DataFrame.to_panel DataFrame.to_panel() Transform long (stacked) format (DataFrame) into wide (3D, Panel) format. 368 Chapter 21. API Reference
Currently the index of the DataFrame must be a 2-level MultiIndex. This may be generalized later Returns panel : Panel pandas.DataFrame.transpose DataFrame.transpose() Returns a DataFrame with the rows/columns switched. If the DataFrame is homogeneously-typed, the data is not copied
pandas.DataFrame.append DataFrame.append(other, ignore_index=False, verify_integrity=False) Append columns of other to end of this frames columns and index, returning a new object. Columns not in this frame are added as new columns. Parameters other : DataFrame or list of Series/dict-like objects ignore_index : boolean, default False If True do not use the index labels. Useful for gluing together record arrays verify_integrity : boolean, default False If True, raise Exception on creating index with duplicates Returns appended : DataFrame
Notes
If a list of dict is passed and the keys are all contained in the DataFrames index, the order of the columns in the resulting DataFrame will be unchanged pandas.DataFrame.join DataFrame.join(other, on=None, how=left, lsufx=, rsufx=, sort=False) Join columns with other DataFrame either on index or on a key column. Efciently Join multiple DataFrame objects by index at once by passing a list. Parameters other : DataFrame, Series with name eld set, or list of DataFrame Index should be similar to one of the columns in this one. If a Series is passed, its name attribute must be set, and that will be used as the column name in the resulting joined DataFrame on : column name, tuple/list of column names, or array-like
21.3. DataFrame
369
Column(s) to use for joining, otherwise join on index. If multiples columns given, the passed DataFrame must have a MultiIndex. Can pass an array as the join key if not already contained in the calling DataFrame. Like an Excel VLOOKUP operation how : {left, right, outer, inner} How to handle indexes of the two objects. Default: left for joining on index, None otherwise * left: use calling frames index * right: use input frames index * outer: form union of indexes * inner: use intersection of indexes lsufx : string Sufx to use from left frames overlapping columns rsufx : string Sufx to use from right frames overlapping columns sort : boolean, default False Order result DataFrame lexicographically by the join key. If False, preserves the index order of the calling (left) DataFrame Returns joined : DataFrame
Notes
on, lsufx, and rsufx options are not supported when passing a list of DataFrame objects pandas.DataFrame.merge DataFrame.merge(right, how=inner, on=None, left_on=None, right_on=None, left_index=False, right_index=False, sort=True, sufxes=(_x, _y), copy=True) Merge DataFrame objects by performing a database-style join operation by columns or indexes. If joining columns on columns, the DataFrame indexes will be ignored. Otherwise if joining indexes on indexes or indexes on a column or columns, the index will be passed on. Parameters right : DataFrame how : {left, right, outer, inner}, default inner left: use only keys from left frame (SQL: left outer join) right: use only keys from right frame (SQL: right outer join) outer: use union of keys from both frames (SQL: full outer join) inner: use intersection of keys from both frames (SQL: inner join) on : label or list Field names to join on. Must be found in both DataFrames. If on is None and not merging on indexes, then it merges on the intersection of the columns by default. left_on : label or list, or array-like Field names to join on in left DataFrame. Can be a vector or list of vectors of the length of the DataFrame to use a particular vector as the join key instead of columns right_on : label or list, or array-like Field names to join on in right DataFrame or vector/list of vectors per left_on docs
370
left_index : boolean, default False Use the index from the left DataFrame as the join key(s). If it is a MultiIndex, the number of keys in the other DataFrame (either the index or a number of columns) must match the number of levels right_index : boolean, default False Use the index from the right DataFrame as the join key. Same caveats as left_index sort : boolean, default True Sort the join keys lexicographically in the result DataFrame sufxes : 2-length sequence (tuple, list, ...) Sufx to apply to overlapping column names in the left and right side, respectively copy : boolean, default True If False, do not copy data unnecessarily Returns merged : DataFrame
Examples >>> A lkey 0 foo 1 bar 2 baz 3 foo >>> B rkey 0 foo 1 bar 2 qux 3 bar
value 1 2 3 4
value 5 6 7 8
>>> merge(A, B, left_on=lkey, right_on=rkey, how=outer) lkey value_x rkey value_y 0 bar 2 bar 6 1 bar 2 bar 8 2 baz 3 NaN NaN 3 foo 1 foo 5 4 foo 4 foo 5 5 NaN NaN qux 7
pandas.DataFrame.replace DataFrame.replace(to_replace, value=None, method=pad, axis=0, inplace=False, limit=None) Replace values given in to_replace with value or using method Parameters value : scalar or dict, default None Value to use to ll holes (e.g. 0), alternately a dict of values specifying which value to use for each column (columns not in the dict will not be lled) method : {backll, bll, pad, fll, None}, default pad Method to use for lling holes in reindexed Series pad / fll: propagate last valid observation forward to next valid backll / bll: use NEXT valid observation to ll gap axis : {0, 1}, default 0 0: ll column-by-column 1: ll row-by-row inplace : boolean, default False 21.3. DataFrame 371
If True, ll the DataFrame in place. Note: this will modify any other views on this DataFrame, like if you took a no-copy slice of an existing DataFrame, for example a column in a DataFrame. Returns a reference to the lled object, which is self if inplace=True limit : int, default None Maximum size gap to forward or backward ll Returns lled : DataFrame See Also: reindex, asfreq pandas.DataFrame.update DataFrame.update(other, join=left, overwrite=True, lter_func=None, raise_conict=False) Modify DataFrame in place using non-NA values from passed DataFrame. Aligns on indices Parameters other : DataFrame, or object coercible into a DataFrame join : {left, right, outer, inner}, default left overwrite : boolean, default True If True then overwrite values for common keys in the calling frame lter_func : callable(1d-array) -> 1d-array<boolean>, default None Can choose to replace values other than NA. Return True for values that should be updated raise_conict : bool If True, will raise an error if the DataFrame and other both contain data in the same place.
Convert all TimeSeries inside to specied frequency using DateOffset Shift the index of the DataFrame by desired number of periods with an Return label for rst non-NA/null value Return label for last non-NA/null value Convenience method for frequency conversion and resampling of regular t Convert DataFrame from DatetimeIndex to PeriodIndex with desired Cast to DatetimeIndex of timestamps, at beginning of period Convert TimeSeries to target time zone. If it is time zone naive, it Localize tz-naive TimeSeries to target time zone
pandas.DataFrame.asfreq DataFrame.asfreq(freq, method=None, how=None) Convert all TimeSeries inside to specied frequency using DateOffset objects. Optionally provide ll method to pad/backll missing values. Parameters freq : DateOffset object, or string
372
method : {backll, bll, pad, fll, None} Method to use for lling holes in reindexed Series pad / fll: propagate last valid observation forward to next valid backll / bll: use NEXT valid observation to ll methdo how : {start, end}, default end For PeriodIndex only, see PeriodIndex.asfreq Returns converted : type of caller pandas.DataFrame.shift DataFrame.shift(periods=1, freq=None, **kwds) Shift the index of the DataFrame by desired number of periods with an optional time freq Parameters periods : int Number of periods to move, can be positive or negative freq : DateOffset, timedelta, or time rule string, optional Increment to use from datetools module or time rule (e.g. EOM) Returns shifted : DataFrame
Notes
If freq is specied then the index values are shifted but the data if not realigned pandas.DataFrame.rst_valid_index DataFrame.first_valid_index() Return label for rst non-NA/null value pandas.DataFrame.last_valid_index DataFrame.last_valid_index() Return label for last non-NA/null value pandas.DataFrame.resample DataFrame.resample(rule, how=None, axis=0, ll_method=None, closed=right, label=right, convention=start, kind=None, loffset=None, limit=None, base=0) Convenience method for frequency conversion and resampling of regular time-series data. Parameters rule : the offset string or object representing target conversion how : string, method for down- or re-sampling, default to mean for downsampling ll_method : string, ll_method for upsampling, default None axis : int, optional, default 0 closed : {right, left}, default right
21.3. DataFrame
373
Which side of bin interval is closed label : {right, left}, default right Which bin edge label to label bucket with convention : {start, end, s, e} loffset : timedelta Adjust the resampled time labels base : int, default 0 For frequencies that evenly subdivide 1 day, the origin of the aggregated intervals. For example, for 5min frequency, base could range from 0 through 4. Defaults to 0 pandas.DataFrame.to_period DataFrame.to_period(freq=None, axis=0, copy=True) Convert DataFrame from DatetimeIndex to PeriodIndex with desired frequency (inferred from index if not passed) Parameters freq : string, default axis : {0, 1}, default 0 The axis to convert (the index by default) copy : boolean, default True If False then underlying input data is not copied Returns ts : TimeSeries with PeriodIndex pandas.DataFrame.to_timestamp DataFrame.to_timestamp(freq=None, how=start, axis=0, copy=True) Cast to DatetimeIndex of timestamps, at beginning of period Parameters freq : string, default frequency of PeriodIndex Desired frequency how : {s, e, start, end} Convention for converting period to timestamp; start of period vs. end axis : {0, 1} default 0 The axis to convert (the index by default) copy : boolean, default True If false then underlying input data is not copied Returns df : DataFrame with DatetimeIndex
374
pandas.DataFrame.tz_convert DataFrame.tz_convert(tz, axis=0, copy=True) Convert TimeSeries to target time zone. If it is time zone naive, it will be localized to the passed time zone. Parameters tz : string or pytz.timezone object copy : boolean, default True Also make a copy of the underlying data pandas.DataFrame.tz_localize DataFrame.tz_localize(tz, axis=0, copy=True) Localize tz-naive TimeSeries to target time zone Parameters tz : string or pytz.timezone object copy : boolean, default True Also make a copy of the underlying data
21.3.12 Plotting
DataFrame.boxplot([column, by, ax, ...]) DataFrame.hist(data[, column, by, grid, ...]) DataFrame.plot([frame, x, y, subplots, ...]) Make a box plot from DataFrame column/columns optionally grouped Draw Histogram the DataFrames series using matplotlib / pylab. Make line or bar plot of DataFrames series with the index on the x-axis
pandas.DataFrame.boxplot DataFrame.boxplot(column=None, by=None, ax=None, fontsize=None, rot=0, grid=True, **kwds) Make a box plot from DataFrame column/columns optionally grouped (stratied) by one or more columns Parameters data : DataFrame column : column names or list of names, or vector Can be any valid input to groupby by : string or sequence Column in the DataFrame to group by fontsize : int or string Returns ax : matplotlib.axes.AxesSubplot pandas.DataFrame.hist DataFrame.hist(data, column=None, by=None, grid=True, xlabelsize=None, xrot=None, ylabelsize=None, yrot=None, ax=None, sharex=False, sharey=False, **kwds) Draw Histogram the DataFrames series using matplotlib / pylab. Parameters grid : boolean, default True Whether to show axis grid lines xlabelsize : int, default None
21.3. DataFrame
375
If specied changes the x-axis label size xrot : oat, default None rotation of x axis labels ylabelsize : int, default None If specied changes the y-axis label size yrot : oat, default None rotation of y axis labels ax : matplotlib axes object, default None sharex : bool, if True, the X axis will be shared amongst all subplots. sharey : bool, if True, the Y axis will be shared amongst all subplots. kwds : other plotting keyword arguments To be passed to hist function pandas.DataFrame.plot DataFrame.plot(frame=None, x=None, y=None, subplots=False, sharex=True, sharey=False, use_index=True, gsize=None, grid=False, legend=True, rot=None, ax=None, style=None, title=None, xlim=None, ylim=None, logy=False, xticks=None, yticks=None, kind=line, sort_columns=False, fontsize=None, secondary_y=False, **kwds) Make line or bar plot of DataFrames series with the index on the x-axis using matplotlib / pylab. Parameters x : label or position, default None y : label or position, default None Allows plotting of one column versus another subplots : boolean, default False Make separate subplots for each time series sharex : boolean, default True In case subplots=True, share x axis sharey : boolean, default False In case subplots=True, share y axis use_index : boolean, default True Use index as ticks for x axis stacked : boolean, default False If True, create stacked bar plot. Only valid for DataFrame input sort_columns: boolean, default False : Sort column names to determine plot ordering title : string Title to use for the plot grid : boolean, default True
376
Axis grid lines legend : boolean, default True Place legend on axis subplots ax : matplotlib axis object, default None style : list or dict matplotlib line style per column kind : {line, bar, barh} bar : vertical bar plot barh : horizontal bar plot logy : boolean, default False For line plots, use log scaling on y axis xticks : sequence Values to use for the xticks yticks : sequence Values to use for the yticks xlim : 2-tuple/list ylim : 2-tuple/list rot : int, default None Rotation for ticks secondary_y : boolean or sequence, default False Whether to plot on the secondary y-axis If dict then can select which columns to plot on secondary y-axis kwds : keywords Options to pass to matplotlib plotting method Returns ax_or_axes : matplotlib.AxesSubplot or list of them
Write DataFrame to a comma-separated values (csv) le Convert DataFrame to dictionary. Write DataFrame to a excel sheet to_html-specic options Convert DataFrame to record array. Index will be put in the Convert to SparseDataFrame Continued on next page
21.3. DataFrame
377
Table 21.34 continued from previous page DataFrame.to_string([buf, columns, ...]) Render a DataFrame to a console-friendly tabular output.
pandas.DataFrame.from_csv classmethod DataFrame.from_csv(path, header=0, sep=, , index_col=0, parse_dates=True, encoding=None) Read delimited le into DataFrame Parameters path : string le path or le handle / StringIO header : int, default 0 Row to use at header (skip prior rows) sep : string, default , Field delimiter index_col : int or sequence, default 0 Column to use for index. If a sequence is given, a MultiIndex is used. Different default from read_table parse_dates : boolean, default True Parse dates. Different default from read_table Returns y : DataFrame
Notes
Preferable to use read_table for most general purposes but from_csv makes for an easy roundtrip to and from le, especially with a DataFrame of time series data pandas.DataFrame.from_dict classmethod DataFrame.from_dict(data, orient=columns, dtype=None) Construct DataFrame from dict of array-like or dicts Parameters data : dict {eld : array-like} or {eld : dict} orient : {columns, index}, default columns The orientation of the data. If the keys of the passed dict should be the columns of the resulting DataFrame, pass columns (default). Otherwise if the keys should be rows, pass index. Returns DataFrame : pandas.DataFrame.from_items classmethod DataFrame.from_items(items, columns=None, orient=columns) Convert (key, value) pairs to DataFrame. The keys will be the axis index (usually the columns, but depends on the specied orientation). The values should be arrays or Series Parameters items : sequence of (key, value) pairs
378
Values should be arrays or Series columns : sequence, optional Must be passed in the orient : {columns, index}, default items The orientation of the data. If the keys of the passed dict should be the items of the result panel, pass items (default). Otherwise if the columns of the values of the passed DataFrame objects should be the items (which in the case of mixed-dtype data you should do), instead pass minor Returns frame : DataFrame pandas.DataFrame.from_records classmethod DataFrame.from_records(data, index=None, exclude=None, erce_oat=False, nrows=None) Convert structured or record ndarray to DataFrame columns=None, co-
Parameters data : ndarray (structured dtype), list of tuples, dict, or DataFrame index : string, list of elds, array-like Field of array to use as the index, alternately a specic set of input labels to use exclude: sequence, default None : Columns or elds to exclude columns : sequence, default None Column names to use. If the passed data do not have named associated with them, this argument provides names for the columns. Otherwise this argument indicates the order of the columns in the result (any names not found in the data will become all-NA columns) coerce_oat : boolean, default False Attempt to convert values to non-string, non-numeric objects (like decimal.Decimal) to oating point, useful for SQL result sets Returns df : DataFrame pandas.DataFrame.info DataFrame.info(verbose=True, buf=None) Concise summary of a DataFrame, used in __repr__ when very large. Parameters verbose : boolean, default True If False, dont print column count summary buf : writable buffer, defaults to sys.stdout pandas.DataFrame.load classmethod DataFrame.load(path)
21.3. DataFrame
379
pandas.DataFrame.save DataFrame.save(path) pandas.DataFrame.to_csv DataFrame.to_csv(path_or_buf, sep=, , na_rep=, oat_format=None, cols=None, header=True, index=True, index_label=None, mode=w, nanRep=None, encoding=None, quoting=None) Write DataFrame to a comma-separated values (csv) le Parameters path_or_buf : string or le handle / StringIO File path na_rep : string, default Missing data representation oat_format : string, default None Format string for oating point numbers cols : sequence, optional Columns to write header : boolean or list of string, default True Write out column names. If a list of string is given it is assumed to be aliases for the column names index : boolean, default True Write row names (index) index_label : string or sequence, or False, default None Column label for index column(s) if desired. If None is given, and header and index are True, then the index names are used. A sequence should be given if the DataFrame uses MultiIndex. If False do not print elds for index names. Use index_label=False for easier importing in R mode : Python write mode, default w sep : character, default , Field delimiter for the output le. encoding : string, optional a string representing the encoding to use if the contents are non-ascii, for python versions prior to 3 pandas.DataFrame.to_dict DataFrame.to_dict(outtype=dict) Convert DataFrame to dictionary. Parameters outtype : str {dict, list, series}
380
Determines the type of the values of the dictionary. The default dict is a nested dictionary {column -> {index -> value}}. list returns {column -> list(values)}. series returns {column -> Series(values)}. Abbreviations are allowed. Returns result : dict like {column -> {index -> value}} pandas.DataFrame.to_excel DataFrame.to_excel(excel_writer, sheet_name=sheet1, na_rep=, oat_format=None, cols=None, header=True, index=True, index_label=None) Write DataFrame to a excel sheet Parameters excel_writer : string or ExcelWriter object File path or existing ExcelWriter sheet_name : string, default sheet1 Name of sheet which will contain DataFrame na_rep : string, default Missing data representation oat_format : string, default None Format string for oating point numbers cols : sequence, optional Columns to write header : boolean or list of string, default True Write out column names. If a list of string is given it is assumed to be aliases for the column names index : boolean, default True Write row names (index) index_label : string or sequence, default None Column label for index column(s) if desired. If None is given, and header and index are True, then the index names are used. A sequence should be given if the DataFrame uses MultiIndex.
Notes
If passing an existing ExcelWriter object, then the sheet will be added to the existing workbook. This can be used to save different DataFrames to one workbook >>> writer = ExcelWriter(output.xlsx) >>> df1.to_excel(writer,sheet1) >>> df2.to_excel(writer,sheet2) >>> writer.save() pandas.DataFrame.to_html DataFrame.to_html(buf=None, columns=None, col_space=None, colSpace=None, header=True, index=True, na_rep=NaN, formatters=None, oat_format=None, sparsify=None, index_names=True, justify=None, force_unicode=False, bold_rows=True, classes=None) to_html-specic options bold_rows : boolean, default True
21.3. DataFrame
381
Make the row labels bold in the output Render a DataFrame to an html table. Parameters frame : DataFrame object to render buf : StringIO-like, optional buffer to write to columns : sequence, optional the subset of columns to write; default None writes all columns col_space : int, optional the width of each columns header : bool, optional whether to print column labels, default True index : bool, optional whether to print index (row) labels, default True na_rep : string, optional string representation of NAN to use, default NaN formatters : list or dict of one-parameter functions, optional formatter functions to apply to columns elements by position or name, default None oat_format : one-parameter function, optional formatter function to apply to columns elements if they are oats default None sparsify : bool, optional Set to False for a DataFrame with a hierarchical index to print every multiindex key at each row, default True justify : {left, right}, default None Left or right-justify the column labels. If None uses the option from the print conguration (controlled by set_printoptions), right out of the box. index_names : bool, optional Prints the names of the indexes, default True force_unicode : bool, default False Always return a unicode result Returns formatted : string (or unicode, depending on data and options) pandas.DataFrame.to_records DataFrame.to_records(index=True) Convert DataFrame to record array. Index will be put in the index eld of the record array if requested Parameters index : boolean, default True Include index in resulting record array, stored in index eld
382
Returns y : recarray pandas.DataFrame.to_sparse DataFrame.to_sparse(ll_value=None, kind=block) Convert to SparseDataFrame Parameters ll_value : oat, default NaN kind : {block, integer} Returns y : SparseDataFrame pandas.DataFrame.to_string DataFrame.to_string(buf=None, columns=None, col_space=None, colSpace=None, header=True, index=True, na_rep=NaN, formatters=None, oat_format=None, sparsify=None, nanRep=None, index_names=True, justify=None, force_unicode=False) Render a DataFrame to a console-friendly tabular output. Parameters frame : DataFrame object to render buf : StringIO-like, optional buffer to write to columns : sequence, optional the subset of columns to write; default None writes all columns col_space : int, optional the width of each columns header : bool, optional whether to print column labels, default True index : bool, optional whether to print index (row) labels, default True na_rep : string, optional string representation of NAN to use, default NaN formatters : list or dict of one-parameter functions, optional formatter functions to apply to columns elements by position or name, default None oat_format : one-parameter function, optional formatter function to apply to columns elements if they are oats default None sparsify : bool, optional Set to False for a DataFrame with a hierarchical index to print every multiindex key at each row, default True justify : {left, right}, default None
21.3. DataFrame
383
Left or right-justify the column labels. If None uses the option from the print conguration (controlled by set_printoptions), right out of the box. index_names : bool, optional Prints the names of the indexes, default True force_unicode : bool, default False Always return a unicode result Returns formatted : string (or unicode, depending on data and options)
21.4 Panel
21.4.1 Computations / Descriptive Stats
384
p
pandas, 1
385
386
p
pandas, 1
387
388
INDEX
Symbols
__init__() (pandas.DataFrame method), 336 __init__() (pandas.Series method), 303 __iter__() (pandas.DataFrame method), 338 __iter__() (pandas.Series method), 304
A
abs() (pandas.DataFrame method), 347 abs() (pandas.Series method), 310 add() (pandas.DataFrame method), 340 add() (pandas.Series method), 305 add_prex() (pandas.DataFrame method), 356 add_sufx() (pandas.DataFrame method), 356 align() (pandas.DataFrame method), 356 align() (pandas.Series method), 318 any() (pandas.DataFrame method), 347 any() (pandas.Series method), 310 append() (pandas.DataFrame method), 369 append() (pandas.Series method), 326 apply() (pandas.DataFrame method), 345 apply() (pandas.Series method), 307 applymap() (pandas.DataFrame method), 346 argsort() (pandas.Series method), 324 as_matrix() (pandas.DataFrame method), 335 asfreq() (pandas.DataFrame method), 372 asfreq() (pandas.Series method), 327 asof() (pandas.Series method), 328 astype() (pandas.DataFrame method), 337 astype() (pandas.Series method), 304 autocorr() (pandas.Series method), 311 axes (pandas.DataFrame attribute), 336
clip_upper() (pandas.DataFrame method), 348 clip_upper() (pandas.Series method), 311 combine() (pandas.DataFrame method), 344 combine() (pandas.Series method), 306 combine_rst() (pandas.DataFrame method), 344 combine_rst() (pandas.Series method), 306 combineAdd() (pandas.DataFrame method), 344 combineMult() (pandas.DataFrame method), 345 concat() (in module pandas.tools.merge), 283 convert_objects() (pandas.DataFrame method), 337 copy() (pandas.DataFrame method), 337 copy() (pandas.Series method), 304 corr() (pandas.DataFrame method), 348 corr() (pandas.Series method), 312 corrwith() (pandas.DataFrame method), 349 count() (pandas.DataFrame method), 349 count() (pandas.Series method), 312 cov() (pandas.DataFrame method), 349 cov() (pandas.Series method), 312 cummax() (pandas.DataFrame method), 349 cummax() (pandas.Series method), 312 cummin() (pandas.DataFrame method), 350 cummin() (pandas.Series method), 312 cumprod() (pandas.DataFrame method), 350 cumprod() (pandas.Series method), 313 cumsum() (pandas.DataFrame method), 350 cumsum() (pandas.Series method), 313
D
delevel() (pandas.DataFrame method), 365 describe() (pandas.DataFrame method), 350 describe() (pandas.Series method), 313 diff() (pandas.DataFrame method), 351 diff() (pandas.Series method), 313 div() (pandas.DataFrame method), 341 div() (pandas.Series method), 305 drop() (pandas.DataFrame method), 357 drop() (pandas.Series method), 319 drop_duplicates() (pandas.DataFrame method), 357 dropna() (pandas.DataFrame method), 363 dropna() (pandas.Series method), 323 dtype (pandas.Series attribute), 302 389
B
between() (pandas.Series method), 311 boxplot() (pandas.DataFrame method), 375
C
clip() (pandas.DataFrame method), 348 clip() (pandas.Series method), 311 clip_lower() (pandas.DataFrame method), 348 clip_lower() (pandas.Series method), 311
E
ewma() (in module pandas.stats.moments), 299 ewmcorr() (in module pandas.stats.moments), 301 ewmcov() (in module pandas.stats.moments), 301 ewmstd() (in module pandas.stats.moments), 299 ewmvar() (in module pandas.stats.moments), 300 expanding_apply() (in module pandas.stats.moments), 298 expanding_corr() (in module pandas.stats.moments), 296 expanding_count() (in module pandas.stats.moments), 295 expanding_cov() (in module pandas.stats.moments), 297 expanding_kurt() (in module pandas.stats.moments), 297 expanding_mean() (in module pandas.stats.moments), 295 expanding_median() (in module pandas.stats.moments), 296 expanding_quantile() (in module pandas.stats.moments), 298 expanding_skew() (in module pandas.stats.moments), 297 expanding_std() (in module pandas.stats.moments), 296 expanding_sum() (in module pandas.stats.moments), 295 expanding_var() (in module pandas.stats.moments), 296
I
idxmax() (pandas.DataFrame method), 359 idxmax() (pandas.Series method), 319 idxmin() (pandas.DataFrame method), 359 idxmin() (pandas.Series method), 319 info() (pandas.DataFrame method), 379 insert() (pandas.DataFrame method), 338 interpolate() (pandas.Series method), 323 isin() (pandas.Series method), 320 isnull() (pandas.Series method), 303 iteritems() (pandas.DataFrame method), 338 iteritems() (pandas.Series method), 305 iterrows() (pandas.DataFrame method), 338 itertuples() (pandas.DataFrame method), 338 ix (pandas.DataFrame attribute), 338 ix (pandas.Series attribute), 304
J
join() (pandas.DataFrame method), 369
K
kurt() (pandas.DataFrame method), 351 kurt() (pandas.Series method), 313
L
last() (pandas.DataFrame method), 359 last() (pandas.Series method), 320 last_valid_index() (pandas.DataFrame method), 373 last_valid_index() (pandas.Series method), 328 load() (in module pandas.core.common), 284 load() (pandas.DataFrame class method), 379 load() (pandas.Series class method), 333 lookup() (pandas.DataFrame method), 338
F
llna() (pandas.DataFrame method), 364 llna() (pandas.Series method), 323 lter() (pandas.DataFrame method), 358 rst() (pandas.DataFrame method), 358 rst() (pandas.Series method), 319 rst_valid_index() (pandas.DataFrame method), 373 rst_valid_index() (pandas.Series method), 328 from_csv() (pandas.DataFrame class method), 378 from_csv() (pandas.Series class method), 332 from_dict() (pandas.DataFrame class method), 378 from_items() (pandas.DataFrame class method), 378 from_records() (pandas.DataFrame class method), 379
M
mad() (pandas.DataFrame method), 351 mad() (pandas.Series method), 314 map() (pandas.Series method), 307 max() (pandas.DataFrame method), 351 max() (pandas.Series method), 314 mean() (pandas.DataFrame method), 352 mean() (pandas.Series method), 314 median() (pandas.DataFrame method), 352 median() (pandas.Series method), 315 merge() (in module pandas.tools.merge), 282 merge() (pandas.DataFrame method), 370 min() (pandas.DataFrame method), 352 min() (pandas.Series method), 315 mul() (pandas.DataFrame method), 341 mul() (pandas.Series method), 305
G
get() (pandas.io.pytables.HDFStore method), 290 get() (pandas.Series method), 304 get_dtype_counts() (pandas.DataFrame method), 335 groupby() (pandas.DataFrame method), 346 groupby() (pandas.Series method), 308
H
head() (pandas.DataFrame method), 338, 358 head() (pandas.Series method), 319 hist() (pandas.DataFrame method), 375
390
Index
N
ndim (pandas.DataFrame attribute), 336 notnull() (pandas.Series method), 303 nunique() (pandas.Series method), 315
O
order() (pandas.Series method), 324
P
pandas (module), 1 parse() (pandas.io.parsers.ExcelFile method), 289 pct_change() (pandas.DataFrame method), 353 pct_change() (pandas.Series method), 315 pivot() (pandas.DataFrame method), 365 pivot_table() (in module pandas.tools.pivot), 281 plot() (pandas.DataFrame method), 376 plot() (pandas.Series method), 331 pop() (pandas.DataFrame method), 339 prod() (pandas.DataFrame method), 353 prod() (pandas.Series method), 316 put() (pandas.io.pytables.HDFStore method), 290
rolling_kurt() (in module pandas.stats.moments), 293 rolling_mean() (in module pandas.stats.moments), 291 rolling_median() (in module pandas.stats.moments), 291 rolling_quantile() (in module pandas.stats.moments), 294 rolling_skew() (in module pandas.stats.moments), 293 rolling_std() (in module pandas.stats.moments), 292 rolling_sum() (in module pandas.stats.moments), 291 rolling_var() (in module pandas.stats.moments), 292 round() (pandas.Series method), 307 rsub() (pandas.DataFrame method), 344
S
save() (in module pandas.core.common), 285 save() (pandas.DataFrame method), 380 save() (pandas.Series method), 333 select() (pandas.DataFrame method), 362 select() (pandas.Series method), 322 set_index() (pandas.DataFrame method), 362 shape (pandas.DataFrame attribute), 336 shift() (pandas.DataFrame method), 373 shift() (pandas.Series method), 328 skew() (pandas.DataFrame method), 354 skew() (pandas.Series method), 316 sort() (pandas.DataFrame method), 366 sort() (pandas.Series method), 325 sort_index() (pandas.DataFrame method), 366 sort_index() (pandas.Series method), 325 sortlevel() (pandas.DataFrame method), 367 sortlevel() (pandas.Series method), 325 stack() (pandas.DataFrame method), 367 std() (pandas.DataFrame method), 355 std() (pandas.Series method), 317 sub() (pandas.DataFrame method), 342 sub() (pandas.Series method), 306 sum() (pandas.DataFrame method), 354 sum() (pandas.Series method), 317 swaplevel() (pandas.DataFrame method), 367 swaplevel() (pandas.Series method), 325
Q
quantile() (pandas.DataFrame method), 353 quantile() (pandas.Series method), 316
R
radd() (pandas.DataFrame method), 342 rank() (pandas.DataFrame method), 354 rank() (pandas.Series method), 316 rdiv() (pandas.DataFrame method), 343 read_csv() (in module pandas.io.parsers), 287 read_table() (in module pandas.io.parsers), 285 reindex() (pandas.DataFrame method), 359 reindex() (pandas.Series method), 320 reindex_axis() (pandas.DataFrame method), 360 reindex_like() (pandas.DataFrame method), 361 reindex_like() (pandas.Series method), 321 rename() (pandas.DataFrame method), 361 rename() (pandas.Series method), 321 reorder_levels() (pandas.DataFrame method), 366 reorder_levels() (pandas.Series method), 324 replace() (pandas.DataFrame method), 371 replace() (pandas.Series method), 326 resample() (pandas.DataFrame method), 373 resample() (pandas.Series method), 329 reset_index() (pandas.DataFrame method), 361 reset_index() (pandas.Series method), 322 rmul() (pandas.DataFrame method), 343 rolling_apply() (in module pandas.stats.moments), 294 rolling_corr() (in module pandas.stats.moments), 292 rolling_count() (in module pandas.stats.moments), 290 rolling_cov() (in module pandas.stats.moments), 293 Index
T
T (pandas.DataFrame attribute), 368 tail() (pandas.DataFrame method), 339, 363 tail() (pandas.Series method), 322 take() (pandas.DataFrame method), 363 take() (pandas.Series method), 322 to_csv() (pandas.DataFrame method), 380 to_csv() (pandas.Series method), 333 to_dict() (pandas.DataFrame method), 380 to_dict() (pandas.Series method), 334 to_excel() (pandas.DataFrame method), 381 to_html() (pandas.DataFrame method), 381 to_panel() (pandas.DataFrame method), 368 to_period() (pandas.DataFrame method), 374 to_records() (pandas.DataFrame method), 382 391
to_sparse() (pandas.DataFrame method), 383 to_sparse() (pandas.Series method), 334 to_string() (pandas.DataFrame method), 383 to_string() (pandas.Series method), 334 to_timestamp() (pandas.DataFrame method), 374 transpose() (pandas.DataFrame method), 369 truncate() (pandas.DataFrame method), 363 truncate() (pandas.Series method), 322 tz_convert() (pandas.DataFrame method), 375 tz_convert() (pandas.Series method), 329 tz_localize() (pandas.DataFrame method), 375 tz_localize() (pandas.Series method), 329
U
unique() (pandas.Series method), 317 unstack() (pandas.DataFrame method), 368 unstack() (pandas.Series method), 326 update() (pandas.DataFrame method), 372 update() (pandas.Series method), 327
V
value_counts() (pandas.Series method), 318 values (pandas.DataFrame attribute), 335 values (pandas.Series attribute), 302 var() (pandas.DataFrame method), 355 var() (pandas.Series method), 317
W
weekday (pandas.Series attribute), 328
X
xs() (pandas.DataFrame method), 339
392
Index