Skip to content

Instantly share code, notes, and snippets.

@karamanbk

karamanbk/g9_twa_dataset.py

Created August 23, 2019 12:56
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save karamanbk/ddf0a83d1ea9d6ce9ba8ff784e727c93 to your computer and use it in GitHub Desktop.
Save karamanbk/ddf0a83d1ea9d6ce9ba8ff784e727c93 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
g9_twa_dataset.py
#create hv segment
df_hv = pd.DataFrame()
df_hv['customer_id'] = np.array([count for count in range(20000)])
df_hv['segment'] = np.array(['high-value' for _ in range(20000)])
df_hv['group'] = 'control'
df_hv.loc[df_hv.index<10000,'group'] = 'test'
df_hv.loc[df_hv.group == 'control', 'purchase_count'] = np.random.poisson(0.6, 10000)
df_hv.loc[df_hv.group == 'test', 'purchase_count'] = np.random.poisson(0.8, 10000)
df_lv = pd.DataFrame()
df_lv['customer_id'] = np.array([count for count in range(20000,100000)])
df_lv['segment'] = np.array(['low-value' for _ in range(80000)])
df_lv['group'] = 'control'
df_lv.loc[df_lv.index<40000,'group'] = 'test'
df_lv.loc[df_lv.group == 'control', 'purchase_count'] = np.random.poisson(0.2, 40000)
df_lv.loc[df_lv.group == 'test', 'purchase_count'] = np.random.poisson(0.3, 40000)
df_customers = pd.concat([df_hv,df_lv],axis=0)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment