作者 / 來源:林驥(ID:linjiwx)
01
散點(diǎn)圖就像一扇窗,打開它,并仔細(xì)觀察,能讓我們看見更多有價(jià)值的信息。
比如說,假設(shè)表格中有 10000 個(gè)客戶年齡和消費(fèi)金額的數(shù)據(jù):
我們可以計(jì)算每一個(gè)年齡對(duì)應(yīng)的人均消費(fèi)金額,比如說,所有 20 歲客戶的平均消費(fèi)金額約為 1383.69 元,然后我們可以畫出一張散點(diǎn)圖:
從圖中可以看出,客戶的年齡與人均消費(fèi)金額有很強(qiáng)的相關(guān)性,其中應(yīng)用了線性回歸算法,得到一條擬合的直線,并用公式表示出來,R2接近于 1 ,代表算法擬合的效果很好。
接下來,我們看看具體實(shí)現(xiàn)的步驟。
首先,導(dǎo)入所需的庫,并設(shè)置中文字體和定義顏色等。
# 導(dǎo)入所需的庫
import numpy as np
import pandas as pd
import matplotlib as mpl
import matplotlib.pyplot as plt
from sklearn.linear_model import LinearRegression
from sklearn.preprocessing import PolynomialFeatures
from sklearn.pipeline import Pipeline
# 正常顯示中文標(biāo)簽
mpl.rcParams['font.sans-serif'] = ['SimHei']
# 自動(dòng)適應(yīng)布局
mpl.rcParams.update({'figure.autolayout': True})
# 正常顯示負(fù)號(hào)
mpl.rcParams['axes.unicode_minus'] = False
# 禁用科學(xué)計(jì)數(shù)法
pd.set_option('display.float_format', lambda x: '%.2f' % x)
# 定義顏色,主色:藍(lán)色,輔助色:灰色,互補(bǔ)色:橙色
c = {'藍(lán)色':'#00589F', '深藍(lán)色':'#003867', '淺藍(lán)色':'#5D9BCF',
'灰色':'#999999', '深灰色':'#666666', '淺灰色':'#CCCCCC',
'橙色':'#F68F00', '深橙色':'#A05D00', '淺橙色':'#FBC171'}
# 數(shù)據(jù)源路徑
filepath='./data/客戶年齡和消費(fèi)金額.xlsx'
# 讀取 Excel文件
df = pd.read_excel(filepath, index_col='客戶編號(hào)')
# 定義畫圖用的數(shù)據(jù):年齡和人均消費(fèi)金額
df_group = df.groupby('年齡').mean()
x = np.array(df_group.index).reshape(-1, 1)
y = np.array(df_group.values)
# 用管道的方式調(diào)用算法,以便把線性回歸擴(kuò)展為多項(xiàng)式回歸
poly_reg = Pipeline([
('ploy', PolynomialFeatures(degree=1)),
('lin_reg', LinearRegression())
])
# 擬合
poly_reg.fit(x, y)
# 斜率
coef = poly_reg.steps[1][1].coef_
# 截距
intercept = poly_reg.steps[1][1].intercept_
# 評(píng)分
score = poly_reg.score(x, y)
接下來,開始用「面向?qū)ο蟆沟姆椒ㄟM(jìn)行畫圖。
# 使用「面向?qū)ο蟆沟姆椒ó媹D,定義圖片的大小
fig, ax = plt.subplots(figsize=(8, 6))
# 設(shè)置標(biāo)題
ax.set_title('\n客戶每年長(zhǎng)一歲,人均消費(fèi)金額增加' + '%.2f' % coef[0][1] + '元\n', loc='left', size=26, color=c['深灰色'])
# 畫氣泡圖
ax.scatter(x, y, color=c['藍(lán)色'], marker='.', s=100, zorder=1)
# # 繪制預(yù)測(cè)線
y2 = poly_reg.predict(x)
ax.plot(x, y2, '-', c=c['橙色'], zorder=2)
# 隱藏邊框
ax.spines['top'].set_visible(False)
ax.spines['right'].set_visible(False)
ax.spines['bottom'].set_visible(False)
ax.spines['left'].set_visible(False)
# 隱藏刻度線
ax.tick_params(axis='x', which='major', length=0)
ax.tick_params(axis='y', which='major', length=0)
ax.set_ylim(15, 65)
ax.set_ylim(1000, 5000)
# 設(shè)置坐標(biāo)標(biāo)簽字體大小和顏色
ax.tick_params(labelsize=16, colors=c['深灰色'])
ax.text(ax.get_xlim()[0]-6, ax.get_ylim()[1], '人\n均\n消\n費(fèi)\n金\n額', va='top', fontsize=16, color=c['深灰色'])
# 設(shè)置坐標(biāo)軸的標(biāo)題
ax.text(ax.get_xlim()[0]+1, ax.get_ylim()[0]-300, '年齡', ha='left', va='top', fontsize=16, color=c['深灰色'])
# 預(yù)測(cè) 55 歲的人均消費(fèi)金額
predict = poly_reg.predict([[55]])
# 標(biāo)注公式
formula = r'$\mathcal{Y} = ' + '%.2f' % coef[0][1] + '\mathcal{X}' + '%+.2f$' % intercept[0] + '\n' + r'$\mathcal{R}^2 = ' + '%.5f$' % score
ax.annotate(formula, xy=(55, predict), xytext=(55, predict+500), ha='center', fontsize=12, color=c['深灰色'], arrowprops=dict(arrowstyle='->', color=c['橙色']))
plt.show()
下載數(shù)據(jù)文件和完整代碼,可以前往 :
https://github.com/linjiwx/mp
當(dāng)業(yè)務(wù)指標(biāo)很多的時(shí)候,應(yīng)該挑選什么指標(biāo)來進(jìn)行分析,這件事很考驗(yàn)分析者的功力,往往需要對(duì)業(yè)務(wù)有比較深刻的理解。
為什么很多人精通各種工具技術(shù),手上也有很多各種各樣的數(shù)據(jù),卻沒有做出讓領(lǐng)導(dǎo)滿意的圖表?
我們應(yīng)該記住,無論多么漂亮的圖表,如果不能從中獲取有價(jià)值的信息,那么也是一張沒有「靈魂」的圖表。
很多時(shí)候,我們面對(duì)的問題,并不是沒有數(shù)據(jù),而是數(shù)據(jù)太多,卻不知道怎么用。
熟悉數(shù)據(jù)分析的思維,能幫我們找到更重要的數(shù)據(jù),排除過多雜亂數(shù)據(jù)的干擾。
如果把數(shù)據(jù)分析比作醫(yī)生看病的過程,那么可以分為以下 4 個(gè)階段:
聯(lián)系客服