Skip to content

Instantly share code, notes, and snippets.

@hauselin

hauselin/simulate_attract_personality.py

Last active February 23, 2020 03:33
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save hauselin/fcb9e1b10fa2d4f3b3fd16d84739a8e7 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save hauselin/fcb9e1b10fa2d4f3b3fd16d84739a8e7 to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. hauselin renamed this gist Feb 23, 2020. 1 changed file with 0 additions and 0 deletions.
    0 simulate_attract_personality → simulate_attract_personality.py
    View file
    File renamed without changes.
  2. hauselin created this gist Feb 23, 2020.
    21 changes: 21 additions & 0 deletions simulate_attract_personality
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,21 @@
    def simulate(simulations=1000, n=500, quantile=0.80):

    correlations = np.zeros(simulations)

    for i in range(simulations):

    data = {"personality": np.random.randn(
    n), "attract": np.random.randn(n)}
    df = pd.DataFrame(data)

    df['personality_attract'] = df['personality'] + df['attract']
    cutoff = df['personality_attract'].quantile(quantile)

    df['outcome'] = np.where(
    df['personality_attract'] > cutoff, 'dated', 'rejected')

    dated = df.query('outcome == "dated"')

    correlations[i] = dated.corr()['personality'][1]

    return correlations