Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import pandas as pd
- from pandas_datareader import data, wb
- import numpy as np
- from datetime import date
- pd.set_option('expand_frame_repr', False)
- df = data.DataReader('GE', 'yahoo', date (2000, 1, 1), date (2000, 2, 1))
- df['x'] = np.where (df['Open'] > df['High'].shift(-2), 1, np.nan)
- print (df.round(2))
- # this section of code works perfectly for an integer based index.......
- ii = df[pd.notnull(df['x'])].index
- dd = np.diff(ii)
- jj = [ii[i] for i in range(1,len(ii)) if dd[i-1] > 2]
- jj = [ii[0]] + jj
- for ci in jj:
- df.loc[ci:ci+2,'x'] = 1.0
- # end of section that works perfectly for an integer based index......
- print (df.round(2))
- Open High Low Close Volume Adj Close x
- Date
- 2000-01-03 153.00 153.69 149.19 150.00 22069800 29.68 1.0
- 2000-01-04 147.25 148.00 144.00 144.00 22121400 28.49 1.0
- 2000-01-05 143.75 147.00 142.56 143.75 27292800 28.44 NaN
- 2000-01-06 143.12 146.94 142.63 145.67 19873200 28.82 NaN
- 2000-01-07 148.00 151.88 147.00 151.31 20141400 29.94 NaN
- 2000-01-10 152.69 154.06 151.12 151.25 15226500 29.93 NaN
- 2000-01-11 151.00 152.69 150.62 151.50 15123000 29.98 NaN
- 2000-01-12 151.06 153.25 150.56 152.00 18342300 30.08 NaN
- 2000-01-13 153.13 154.94 153.00 153.75 14953500 30.42 1.0
- 2000-01-14 153.38 154.63 149.56 151.00 18480300 29.88 1.0
- 2000-01-18 149.62 149.62 146.75 148.00 18296700 29.29 NaN
- 2000-01-19 146.50 150.94 146.25 148.72 14849700 29.43 NaN
- 2000-01-20 149.06 149.75 142.63 145.94 30759000 28.88 1.0
- 2000-01-21 147.94 148.25 143.94 144.13 24005400 28.52 1.0
- 2000-01-24 145.31 145.94 136.44 138.13 27116100 27.33 1.0
- 2000-01-25 138.06 140.38 137.00 138.50 25387500 27.41 NaN
- 2000-01-26 140.50 142.19 138.88 141.44 15856800 27.99 NaN
- 2000-01-27 141.56 141.75 137.06 141.75 19243500 28.05 1.0
- 2000-01-28 140.31 140.50 133.63 134.00 29846700 26.52 1.0
- 2000-01-31 134.00 135.94 133.06 134.00 21782700 26.52 NaN
- 2000-02-01 134.25 137.00 134.00 136.00 27339000 26.91 NaN
- Traceback (most recent call last):
- File "C:stocksquestion4 for stack overflow.py", line 15, in <module>
- jj = [ii[i] for i in range(1,len(ii)) if dd[i-1] > 2]
- File "C:stocksquestion4 for stack overflow.py", line 15, in <listcomp>
- jj = [ii[i] for i in range(1,len(ii)) if dd[i-1] > 2]
- TypeError: Cannot cast ufunc greater input from dtype('<m8[ns]') to dtype('<m8') with casting rule 'same_kind'
- Open High Low Close Volume Adj Close x
- Date
- 2000-01-03 153.00 153.69 149.19 150.00 22069800 29.68 1.0
- 2000-01-04 147.25 148.00 144.00 144.00 22121400 28.49 1.0
- 2000-01-05 143.75 147.00 142.56 143.75 27292800 28.44 1.0
- 2000-01-06 143.12 146.94 142.63 145.67 19873200 28.82 NaN
- 2000-01-07 148.00 151.88 147.00 151.31 20141400 29.94 NaN
- 2000-01-10 152.69 154.06 151.12 151.25 15226500 29.93 NaN
- 2000-01-11 151.00 152.69 150.62 151.50 15123000 29.98 NaN
- 2000-01-12 151.06 153.25 150.56 152.00 18342300 30.08 NaN
- 2000-01-13 153.13 154.94 153.00 153.75 14953500 30.42 1.0
- 2000-01-14 153.38 154.63 149.56 151.00 18480300 29.88 1.0
- 2000-01-18 149.62 149.62 146.75 148.00 18296700 29.29 1.0
- 2000-01-19 146.50 150.94 146.25 148.72 14849700 29.43 NaN
- 2000-01-20 149.06 149.75 142.63 145.94 30759000 28.88 1.0
- 2000-01-21 147.94 148.25 143.94 144.13 24005400 28.52 1.0
- 2000-01-24 145.31 145.94 136.44 138.13 27116100 27.33 1.0
- 2000-01-25 138.06 140.38 137.00 138.50 25387500 27.41 NaN
- 2000-01-26 140.50 142.19 138.88 141.44 15856800 27.99 NaN
- 2000-01-27 141.56 141.75 137.06 141.75 19243500 28.05 1.0
- 2000-01-28 140.31 140.50 133.63 134.00 29846700 26.52 1.0
- 2000-01-31 134.00 135.94 133.06 134.00 21782700 26.52 1.0
- 2000-02-01 134.25 137.00 134.00 136.00 27339000 26.91 NaN
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement