数模竞赛代码整理----画图

seasns yyds！！！

相关矩阵图

import seaborn as sns
import numpy as np
import matplotlib.pyplot as plt

f, ax= plt.subplots(figsize = (14, 10))
sns.heatmap(data,cmap='RdBu', linewidths = 0.05, annot=True)

# 设置Axes的标题
ax.set_title('Correlation between features')
plt.show()
plt.close()
f.savefig('sns_style_origin.jpg', dpi=100, bbox_inches='tight')

效果

折线图

### 画图
import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns; sns.set()
%matplotlib inline

sns.set_style(style='white')
f, ax= plt.subplots(figsize = (14, 5))

dd = [model_out_pd['prediction'], real_pd['real']]

ax = sns.lineplot(data=dd)
plt.show()

AUC

from sklearn import metrics

def plot_AUC(model,X_test,y_test):
    probs = model.predict_proba(X_test)
    preds = probs[:,1]
    fpr, tpr, threshold = metrics.roc_curve(y_test, preds)
    roc_auc = metrics.auc(fpr, tpr)

    plt.title('Receiver Operating Characteristic')
    plt.plot(fpr, tpr, 'b', label = 'AUC = %0.2f' % roc_auc)
    plt.legend(loc = 'lower right')
    plt.plot([0, 1], [0, 1],'r--')
    plt.xlim([0, 1])
    plt.ylim([0, 1])
    plt.ylabel('True Positive Rate')
    plt.xlabel('False Positive Rate')
    plt.show()

条形图

import matplotlib.pyplot as plt
import seaborn as sns

f, ax= plt.subplots(figsize = (14, 10))
plt.xticks([])
sns.barplot(y=result.index.to_list(),x=result.iloc[:,0])

箱型图

def plot_box(df):
    title = "Box plot of {}".format(df.columns[0])
    df.plot.box(title= title)
    plt.grid(linestyle="--", alpha=0.3)
    plt.show()