初识数据之特征选择

第1关：初识数据之特征选择

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.image as mpimg

def test():

    data_url = "/data/workspace/myshixun/step1/train.csv"
    df = pd.read_csv(data_url)
    S=df[df.Survived==1]
    D=df[df.Survived==0]
    #绘制根据S、D的前9列数据绘制直方图
    ########## Begin ##########
    plt.hist(S.iloc[:,9])
    plt.hist(D.iloc[:,9])

    ########## Begin ##########
    plt.savefig('/data/workspace/myshixun/step1/picture2/直方图.png')
    
    #sex = df.groupby('Sex')['Survived'