前言

怎么用图表对比前后变化？用python试试吧。本次对比了喷火龙和超梦mega进化的前后对比。

例16

import matplotlib.lines as mlines
# Import Data
L_val=df['Defense']
R_val=df['Defense.1']


def newline(p1, p2, color='black'):
    ax = plt.gca()
    l = mlines.Line2D([p1[0],p2[0]], [p1[1],p2[1]], color='red' if p1[1]-p2[1] >= 0 else 'green', marker='o', markersize=6)
    ax.add_line(l)
    return l

fig, ax = plt.subplots(1,1,figsize=(10.5,7.5), dpi= 80)

# Vertical Lines
for i in range(4):
    if i==2:
        continue
    ax.vlines(x=i, ymin=30, ymax=150, color='black', alpha=0.7, linewidth=1, linestyles='dotted')

# Points
ax.scatter(y=L_val, x=np.repeat(1, df.shape[0]), s=10, color='black', alpha=0.7)
ax.scatter(y=R_val, x=np.repeat(3, df.shape[0]), s=10, color='black', alpha=0.7)

# Line Segmentsand Annotation
for p1, p2, l,r  in zip(L_val, R_val,df['Name'],df['Name.1']):
    newline([1,p1], [3,p2])
    ax.text(1-0.05, p1, l + ', ' + str(round(p1)), horizontalalignment='right', verticalalignment='center', fontdict={'size':14},rotation=0)
    ax.text(3+0.05, p2, r + ', ' + str(round(p2)), horizontalalignment='left', verticalalignment='center', fontdict={'size':14},rotation=0)
# 'Before' and 'After' Annotations
ax.text(1-0.05, 150, 'BEFORE', horizontalalignment='right', verticalalignment='center', fontdict={'size':18, 'weight':700})
ax.text(3+0.05, 150, 'AFTER', horizontalalignment='left', verticalalignment='center', fontdict={'size':18, 'weight':700})

# Decoration
ax.set_title("Defense vs Mega Defense", fontdict={'size':22})
ax.set(xlim=(0,4), ylim=(29,150), ylabel='Defense Value')
ax.set_xticks([1,3])
ax.set_xticklabels(["Normal", "Mega"])
plt.yticks(np.arange(50, 150, 10), fontsize=10,rotation=60)
plt.gca().spines["top"].set_alpha(.0)
plt.gca().spines["right"].set_alpha(.0)

数据表

	id	Name	Attack	Defense	Name.1	Attack.1	Defense.1
0	6	Charizard	84	78	Mega Charizard X	130	111
1	6	Charizard	84	78	Mega Charizard Y	104	78
2	150	Mewtwo	110	90	Mega Mewtwo X	190	100
3	150	Mewtwo	110	90	Mega Mewtwo Y	150	70

图像

111.png

解析

代码解析

mlines.Line2D（）

matplotlib中任意执行的画法

“# Vertical Lines”

画垂直的辅助线，把画布4等分

“# Points"

标记数据点的位置

”# Line Segmentsand Annotation“

画出红绿色的线段和标注文本的位置

”# 'Before' and 'After' Annotations“

标注Before、After的文本位置

”# Decoration“

ax.set() 设置x 坐标轴的长度，y轴长度，和名字

plt.gca().spines["top"].set_alpha(.0) 把顶上的框线去掉

ax.set_xticks([1,3]) 很关键 设置x轴的刻度，因为这张图x刻度不是连续的.

图像解释

y轴表示防御力的数值；x轴 间隔左边是mega 进化前 ，右边是mega 进化后，绿色表示增长，红色表示减少

结论

因为宝可梦的能力值很多都是相同的并且很少会有负增长的情况，所以梯形图在多数据的时候就显得很乱了。比如对比所有宝可梦mega后的速度变化。

112.png

下期预告

哑铃图Dumbbell Plot —— 宝可梦mega前后能力增长对比2

数据表格地址：https://raw.githubusercontent.com/ub3132003/pynotebook/master/data/pokemon.csv

灵感来自: machinelearningplus 感谢b站UP "菜菜TsaiTsai" 分享这个博客.

上期内容：