Vectorized look-up of values in Pandas dataframe

by

Use our friend lookup, designed precisely for this purpose:

In [17]: prices
Out[17]: 
              AAPL    GOOG     IBM    XOM
2011-01-10  339.44  614.21  142.78  71.57
2011-01-13  342.64  616.69  143.92  73.08
2011-01-26  340.82  616.50  155.74  75.89
2011-02-02  341.29  612.00  157.93  79.46
2011-02-10  351.42  616.44  159.32  79.68
2011-03-03  356.40  609.56  158.73  82.19
2011-05-03  345.14  533.89  167.84  82.00
2011-06-03  340.42  523.08  160.97  78.19
2011-06-10  323.03  509.51  159.14  76.84
2011-08-01  393.26  606.77  176.28  76.67
2011-12-20  392.46  630.37  184.14  79.97

In [18]: orders
Out[18]: 
                  Date direction  size ticker  prices
0  2011-01-10 00:00:00       Buy  1500   AAPL  339.44
1  2011-01-13 00:00:00      Sell  1500   AAPL  342.64
2  2011-01-13 00:00:00       Buy  4000    IBM  143.92
3  2011-01-26 00:00:00       Buy  1000   GOOG  616.50
4  2011-02-02 00:00:00      Sell  4000    XOM   79.46
5  2011-02-10 00:00:00       Buy  4000    XOM   79.68
6  2011-03-03 00:00:00      Sell  1000   GOOG  609.56
7  2011-03-03 00:00:00      Sell  2200    IBM  158.73
8  2011-06-03 00:00:00      Sell  3300    IBM  160.97
9  2011-05-03 00:00:00       Buy  1500    IBM  167.84
10 2011-06-10 00:00:00       Buy  1200   AAPL  323.03
11 2011-08-01 00:00:00       Buy    55   GOOG  606.77
12 2011-08-01 00:00:00      Sell    55   GOOG  606.77
13 2011-12-20 00:00:00      Sell  1200   AAPL  392.46

In [19]: prices.lookup(orders.Date, orders.ticker)
Out[19]: 
array([ 339.44,  342.64,  143.92,  616.5 ,   79.46,   79.68,  609.56,
        158.73,  160.97,  167.84,  323.03,  606.77,  606.77,  392.46])

More Related Contents:

  1. Vectorizing Haversine distance calculation in Python
  2. vectorize conditional assignment in pandas dataframe
  3. NumPy version of “Exponential weighted moving average”, equivalent to pandas.ewm().mean()
  4. Still getting ModuleNotFoundError: No module named ‘pandas’ ‘ despite having installed Pandas, anaconda, [closed]
  5. Binning a column with Python Pandas
  6. Best way to join / merge by range in pandas
  7. How to take column-slices of dataframe in pandas
  8. Creating a Pandas DataFrame from a Numpy array: How do I specify the index column and column headers?
  9. float64 with pandas to_csv
  10. How to check if a value is in the list in selection from pandas data frame?
  11. shuffling/permutating a DataFrame in pandas
  12. Pandas finding local max and min
  13. SQL-like window functions in PANDAS: Row Numbering in Python Pandas Dataframe
  14. Recursive definitions in Pandas
  15. extracting days from a numpy.timedelta64 value
  16. Set values on the diagonal of pandas.DataFrame
  17. What techniques can be used to measure performance of pandas/numpy solutions
  18. Custom transformer for sklearn Pipeline that alters both X and y
  19. Python Numpy TypeError: ufunc ‘isfinite’ not supported for the input types
  20. Pass percentiles to pandas agg function
  21. Subtracting numpy arrays of different shape efficiently
  22. Why is np.where faster than pd.apply
  23. How to determine whether a column/variable is numeric or not in Pandas/NumPy?
  24. pandas.read_csv from string or package data
  25. How to automatically annotate maximum value in pyplot
  26. What’s the best way to sum all values in a Pandas dataframe? [duplicate]
  27. Split Python sequence (time series/array) into subsequences with overlap
  28. Vectorizing `numpy.random.choice` for given 2D array of probabilities along an axis
  29. Add density curve on the histogram
  30. Numpy.dtype has the wrong size, try recompiling

Leave a Comment