我的方法不使用map
,但函数stack, groupby, pivot_table, merge and cumsum:
df.sort_values(by='date', axis=0, inplace = True)
#set helper column for counting cumsum
df['one'] = 1
print df
# date league home away hscore ascore one
#5 2015-01-03 03:02:00 MLB Cle Tex 9 6 1
#0 2015-05-10 03:03:00 MLB Bos Cle 6 7 1
#1 2015-10-15 03:00:00 MLB Tex Bos 5 2 1
#2 2015-10-15 03:30:00 MLB Tex Bos 1 6 1
#3 2015-10-16 00:00:00 MLB Tex Bos 4 4 1
#4 2015-10-17 03:30:00 MLB Bos Tex 2 8 1
#6 2015-10-18 00:00:00 MLB Tex Bos 9 10 1
#7 2015-10-20 00:00:00 MLB Bos Tex 2 3 1
#8 2015-10-21 00:00:00 MLB Tex Bos 5 1 1
#9 2015-10-22 03:00:00 MLB Tex Bos 5 3 1
#10 2015-10-23 00:00:00 MLB Bos Tex 3 4 1
#11 2015-10-25 23:00:00 MLB Bos Tex 6 6 1
#12 2015-10-25 23:00:00 MLB Bos Tex 5 1 1
#13 2015-10-26 00:00:00 MLB Tex Bos 9 6 1
#14 2015-10-27 01:30:00 MLB Bos Tex 10 5 1
#15 2015-10-28 01:00:00 MLB Tex Bos NaN NaN 1
#16 2015-11-20 03:01:00 MLB Cle Bos NaN NaN 1
#set columns home and away to one columns for cumsum
df2 = df[['date', 'home', 'away', 'one']].set_index(['date', 'one'])
df2 = df2.stack().reset_index(name="both")
df2['new'] = df2.groupby(['both'])['one'].cumsum() - 1
#print df2
#get back to original index
df1 = pd.pivot_table(df2, index=['date'], columns=['level_2'], values='new').reset_index()
#print df1
#merge with original df
df1 = pd.merge(df, df1, on=['date'], suffixes=('', '_new'))
#rename and casr float columns to integers columns
df1 = df1.rename(columns={'away_new':'aag', 'home_new':'hag',})
df1['aag'] = df1['aag'].astype(int)
df1['hag'] = df1['hag'].astype(int)
#count aaag and hahg
df1['aaag'] = df1.groupby(['away'])['one'].cumsum() - 1
df1['hahg'] = df1.groupby(['home'])['one'].cumsum() - 1
#drop helper column one and set index
df1 = df1.drop(['one'], axis=1 ).set_index('date')
#reorder columns
df1 = df1[['league', 'home', 'away', 'hscore', 'ascore', 'hag', 'aag', 'hahg', 'aaag']]
print df1
league home away hscore ascore hag aag hahg aaag
date
2015-01-03 03:02:00 MLB Cle Tex 9 6 0 0 0 0
2015-05-10 03:03:00 MLB Bos Cle 6 7 0 1 0 0
2015-10-15 03:00:00 MLB Tex Bos 5 2 1 1 0 0
2015-10-15 03:30:00 MLB Tex Bos 1 6 2 2 1 1
2015-10-16 00:00:00 MLB Tex Bos 4 4 3 3 2 2
2015-10-17 03:30:00 MLB Bos Tex 2 8 4 4 1 1
2015-10-18 00:00:00 MLB Tex Bos 9 10 5 5 3 3
2015-10-20 00:00:00 MLB Bos Tex 2 3 6 6 2 2
2015-10-21 00:00:00 MLB Tex Bos 5 1 7 7 4 4
2015-10-22 03:00:00 MLB Tex Bos 5 3 8 8 5 5
2015-10-23 00:00:00 MLB Bos Tex 3 4 9 9 3 3
2015-10-25 23:00:00 MLB Bos Tex 6 6 10 10 4 4
2015-10-25 23:00:00 MLB Bos Tex 5 1 10 10 5 5
2015-10-26 00:00:00 MLB Tex Bos 9 6 12 12 6 6
2015-10-27 01:30:00 MLB Bos Tex 10 5 13 13 6 6
2015-10-28 01:00:00 MLB Tex Bos NaN NaN 14 14 7 7
2015-11-20 03:01:00 MLB Cle Bos NaN NaN 2 15 1 8