小狗疗法协方差分析

import pandas as pd
import numpy as np

url = 'https://liangdaima.com/static/data/ds/puppy_love.csv'
df = pd.read_csv(url)
df['dose_code'] = df['dose'].map({'No puppies': 0, '15 mins': 1, '30 mins': 2})
r_xy = df['happiness'].corr(df['puppy_love'])
r_xz = df['happiness'].corr(df['dose_code'])
r_yz = df['puppy_love'].corr(df['dose_code'])
partial_r = (r_xy - r_xz * r_yz) / np.sqrt((1 - r_xz**2) * (1 - r_yz**2))
print(f'{partial_r:.4f}')