数据分析习题Python

本次作业是emu193教程的课后作业，来源自：

https://nbviewer.jupyter.org/github/schmit/cme193-ipython-notebooks-lecture/blob/master/Exercises.ipynb

Part 1

For each of the four datasets...

• Compute the mean and variance of both x and y
• Compute the correlation coefficient between x and y
• Compute the linear regression line:  (hint: use statsmodels and look at the Statsmodels notebook)

第一部分主要是要求我们对读入的数据进行分析，求出均值，方差，相关系数，线性拟合等等。代码如下

# coding: utf-8
import random

import numpy as np
import scipy as sp
import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns

import statsmodels.api as sm
import statsmodels.formula.api as smf

sns.set_context("talk")

anascombe = pd.read_csv('anscombe.csv')
anascombe.head()

print("\nThe mean of x and y:")
print(anascombe.groupby(['dataset']).mean())


print("\nThe varience of x and y:")
print(anascombe.groupby(['dataset']).var())


print("\nThe correlation coefficient of x and y:")
print(anascombe.groupby(["dataset"]).corr())

keys = ['I', 'II', 'III', 'IV']  
for k in keys:  
    lin_model = smf.ols('y ~ x', anascombe[anascombe['dataset'] == k]).fit()  
    print('\nThe linear model for dataset', k)  
    print(lin_model.summary())

运行结果如下：

The mean of x and y:
           x         y
dataset
I        9.0  7.500909
II       9.0  7.500909
III      9.0  7.500000
IV       9.0  7.500909

The varience of x and y:
            x         y
dataset
I        11.0  4.127269
II       11.0  4.127629
III      11.0  4.122620
IV       11.0  4.123249

The correlation coefficient of x and y:
                  x         y
dataset
I       x  1.000000  0.816421
        y  0.816421  1.000000
II      x  1.000000  0.816237
        y  0.816237  1.000000
III     x  1.000000  0.816287
        y  0.816287  1.000000
IV      x