Skip to content

Instantly share code, notes, and snippets.

@quantra-go-algo
Created December 22, 2021 04:12
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 quantra-go-algo/c2416193e9ab6bebada455f46ddace1b to your computer and use it in GitHub Desktop.
Save quantra-go-algo/c2416193e9ab6bebada455f46ddace1b to your computer and use it in GitHub Desktop.
#Convert data into numerical values with mean
Infected = [2, 4, 5, 6, 4, 3]
Predictor = ['Delta', 'Lambda', 'Omicron’, ’Lambda’, ’Delta’, ’Omicron’]
Infected_df = pd.DataFrame(data={'Infected':Infected, 'Predictor':Predictor})
means = Infected_df.groupby('Predictor')['Infected'].mean()
Infected_df['Predictor_encoded'] = Infected_df['predictor'].map(means)
Infected_df
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment