全文共3138字,预计学习时长20分钟或更长

可视化令数据一目了然。然而,成功的数据可视化往往很难实现。此外,向更多受众呈现这些可视化的数据,也需要耗费更多时间和精力。

大家都知道如何制作条形图、散点图和直方图,但却不注重美化它们。这在无形中会损害我们在同行和上级心中的可靠性。

此外,代码重用也很重要。你该不会想每次访问数据集都从头开始吧?利用一些可重用的图形能更快地找到你想要的信息。

本文涉及三个实用的可视化工具:

· 图形分类相关性

· 散点图矩阵

· 使用Seaborn的分类散点图注释和图注释

总的来说,本文会教大家制作一些好看又中用的图表。

本文将使用kaggle上的国际足联2019年完整的球员数据集,其最新版数据库包括了每个注册在内的球员的详细信息。

由于该数据集有许多列,因此我们只关注分类列和连续列的子集。

import numpy as np

import pandas as pd

import seaborn as sns

import ma as plt

%matplotlib inline

# We dont Probably need the Gridlines. Do we? If yes comment this line

(style="ticks")

player_df = ("../inpu;)

numcols = [

'Overall',

'Potential',

'Crossing','Finishing', 'ShortPassing', 'Dribbling','LongPassing', 'BallControl', 'Acceleration',

'SprintSpeed', 'Agility', 'Stamina',

'Value','Wage']

catcols = ['Name','Club','Nationality','Preferred Foot','Position','Body Type']

# Subset the columns

player_df = player_df[numcols+ catcols]

# Few rows of data

(5)

球员数据

虽然该数据格式良好,但是因为工资和值列是以欧元为单位,并包含字符串,需要进行一些预处理,才能使它们为后续分析提供数值。

def wage_split(x):

try:

return in("K")[0][1:])

except:

return 0

player_df['Wage'] = player_df['Wage'].apply(lambda x : wage_split(x))

def value_split(x):

try:

if 'M' in x:

return floa("M")[0][1:])

elif 'K' in x:

return floa("K")[0][1:])/1000

except:

return 0

player_df['Value'] = player_df['Value'].apply(lambda x : value_split(x))

图形分类相关性

简单来说,相关性是衡量两个变量如何一起运动的指标。

例如,在现实生活中,收入与支出呈正相关,其中一个变量随着另一个变量的增加而增加。

学习成绩和电子游戏的使用呈负相关,其中一个变量的增加意味着另一个变量的减少。

因此如果预测变量与目标变量呈正相关或负相关,那么该变量就有研究价值。

研究不同变量之间的相关性对于理解数据非常有意义。

使用Seaborn即可轻松创建出相当不错的关系图。

corr = ()

g = (corr, vmax=.3, center=0,

square=True, linewidths=.5, cbar_kws={"shrink": .5}, annot=True, fmt='.2f', cmap='coolwarm')

()

g.(14,10)

()

所有的分类变量都去哪了?

你有注意到什么问题吗?

有问题,因为该图仅计算了数值列之间的相关性。

如果目标变量是club或position,会出现什么情况?

如果想得到三种不同情况之间的相关性,可使用以下相关性度量来计算。

1. 数值变量

该变量可通过Pearson相关性的方式得到,用于度量两个变量如何一起运动,范围为[-1,1]。

2. 分类变量

使用克莱姆V系数来分类案例。该系数是两个离散变量之间的相互关联,并与具有两个或多层次的变量一起使用。它也是一个对称的度量,因为变量的顺序无关紧要,即克莱姆(A,B)==克莱姆(B,A)。

例如,在数据集中,Club和Nationality一定有某种关联。

可用堆叠图来验证这一点,这是理解分类变量和分类变量间分布的一个绝佳方法,因为在该数据中有很多国籍和俱乐部,所以使用数据的子集。

只保留最好的球队(保留波尔图足球俱乐部只是为了让样本更加多样化)和最常见的国籍。

俱乐部偏好在很大程度上反映了“国籍”:了解前者有助于预测后者。

由图可知,英国球员更可能效力于切尔西队或曼联队,而不是在巴塞罗那队、拜仁慕尼黑队或波尔图队。

同理,克莱姆V系数也也捕获到了同样的信息。

如果所有俱乐部拥有的球员的国籍比例相同,那么克莱姆V系数则为0。

如果每个俱乐部偏好单一国籍的球员,则克莱姆系数V==1,例如,所有的英国球员在曼联队效力,所有的德国球员在拜仁慕尼黑队效力等等。

在所有其他情况下,范围则为[0,1]。

3. 数值变量和分类变量

对连续分类案例使用相关比率。

在不涉及太多数学的情况下,该变量用于离散程度的衡量。

如果给定一个数字,就能找出它的类别吗?

例如,假设数据集中有“SprintSpeed”和“Position”两列分类,那么:

守门员:58(De Gea)、52(T. Courtois)、58(M. Neuer)、 43(G. Buffon)

中后卫:68(D. Godin)、59(V. Kompany)、73(S. Umtiti)、 75(M. Benatia)

前锋:91)、94(G. Bale)、80)、 76(R. Lewandowski)

由上可知,这些数字很好地预测了他们所处的位置,因此相关性很高。

如果某球员冲刺速度超过85,那么该球员肯定是前锋。

这个比率也在[0,1]之间。

执行此操作的代码取自dython包,代码不会很多,最终结果如下:

player_df = (0)

results = associations(player_df,nominal_columns=catcols,return_results=True)

分类vs.分类、分类vs.数值、数值vs.数值,这些使图表更为有趣。

很美,不是吗?

只要看看数据,就能对足球有如此多的了解,例如:

· 球员的位置与运球能力高度相关。总不能让梅西踢后卫吧!

· 值与传球和控球的相关性比运球更高。规则是永远传球,正如内马尔的传球。

· “俱乐部”和“收入”有很高的相关性并且可预测。

· “体型”与“踢球偏好的脚”高度相关。这是否意味着如果某球员是瘦子,就很可能喜欢用左脚踢球?这可能没啥实际意义,需要进一步调查。

此外,通过这个简单的图表,就能找到上述这么多信息,这在没有分类变量的典型相关图中是见不到的。

大家可深入研究这张图表,得到更多有意义的结果,但关键是图表能让大家在现实生活中更容易找到某种规律。

散点图矩阵

虽然前文谈到了很多相关性,但它是一个变幻无常的指标,为了让大家理解,我们来看一个例子。

“Anscombe四重奏”由四个相关性几乎近似于1的数据集组成,但具有非常不同的数据分布,并且在绘制时呈现出非常不同的效果。

Anscombe四重奏:相关性变化无常

因此,有时绘制相关数据变得至关重要,并且需要单独查看分布。

现在数据集中有很多列,把它们全都绘制成图形会很费力。

其实只需几行代码就可以解决。

filtered_player_df = player_df[(player_df['Club'].isin(['FC Barcelona', 'Paris Saint-Germain',

'Manchester United', 'Manchester City', 'Chelsea', 'Real Madrid','FC Porto','FC Bayern München'])) &

(player_df['Nationality'].isin(['England', 'Brazil', 'Argentina',

'Brazil', 'Italy','Spain','Germany']))

]

# Single line to create pairplot

g = (filtered_player_df[['Value','SprintSpeed','Potential','Wage']])

非常好,在该图中可看到很多信息。

· 工资和价值高度相关。

· 大多数其它值也是相关的,然而“潜力”与“价值”的比的趋势是不寻常的。可以看到,当到达特定的潜在阀值时,价值如何呈指数增长。这些信息对建模很有帮助,可以对“潜力”进行转换使其更具有相关性吗?

警告:没有分类列!

在此基础上能做得更好吗?总能做到。

g = (filtered_player_df[['Value','SprintSpeed','Potential','Wage','Club']],hue = 'Club')

图上的信息很多,只需添加“hue”参数到分类变量“club”中即可。

· 波尔图队的工资分配趋向于工资少的那一端。

· 该图无法看出波尔图队球员价值的急剧分布,波尔图队的球员们总是在寻求机会。

· 许多粉红点(代表切尔西队)在“潜力”和“工资”图上形成了一个集群。切尔西队有很多工资较低的高潜力球员,需要更多关注。

还可从工资/价值子图中获取一些信息。

年薪50万的蓝点是梅西。此外,比梅西更有价值的橙点是内马尔。

尽管该技巧仍然不能解决分类问题,但还有一些其它方法来研究分类变量分布,虽然是个例。

分类散点图

如何查看分类数据和数字数据之间的关系?

就像输入名字一样,输入分类散点图的图片。为每个类别绘制一组点,在y轴上稍微分散,以便于查看。

这是我们目前绘制这种关系的首选方法。

g = (y = "Club",

x = 'Wage',

data = filtered_player_df,

# Decrease the size of the points to avoid crowding

size = 7)

# remove the top and right line in graph

()

g.(14,10)

()

分类散点图

为什么不用箱形图呢?中位数在哪?可以绘制出来吗?当然可以。在顶部覆盖一个条形图,就得到了一个好看的图形。

g = (y = "Club",

x = 'Wage',

data = filtered_player_df, whis=np.inf)

g = (y = "Club",

x = 'Wage',

data = filtered_player_df,

# Decrease the size of the points to avoid crowding

size = 7,color = 'black')

# remove the top and right line in graph

()

g.(12,8)

()

有趣的分类散点图+箱形图

很好,在图表上可看出各个点的分布和一些统计数据,并能明确地了解工资差异。

图中最右边的点是梅西,这样一来,就不用通过图表下方的文字来说明。

该图可用来做演示,如果老板要求把梅西写在这张图上,那么就可加上图片注释。

max_wage = ()

max_wage_player = filtered_player_df[(player_df['Wage'] == max_wage)]['Name'].values[0]

g = (y = "Club",

x = 'Wage',

data = filtered_player_df, whis=np.inf)

g = (y = "Club",

x = 'Wage',

data = filtered_player_df,

# Decrease the size of the points to avoid crowding

size = 7,color='black')

# remove the top and right line in graph

()

# Annotate. xy for coordinate. max_wage is x and 0 is y. In this plot y ranges from 0 to 7 for each level

# xytext for coordinates of where I want to put my text

(s = max_wage_player,

xy = (max_wage,0),

xytext = (500,1),

# Shrink the arrow to avoid occlusion

arrowprops = {'facecolor':'gray', 'width': 3, 'shrink': 0.03},

backgroundcolor = 'white')

g.(12,8)

()

带注释的统计信息和点群,可用于演讲中。

· 看看该图下方的波尔图队,工资预算如此之小,难以和其它高收入的球队们竞争。

· 皇马和巴塞罗那有许多高薪球员。

· 曼联的工资中位数最高。

· 曼联和切尔西注重平等,许多球员的工资水平都差不多。

· 虽然内马尔比梅西更受重视,但梅西和内马尔的工资差距巨大。

由此可知,在这个疯狂的世界中,有些正常只是表面的。

本文讨论了计算和读取不同变量类型之间的相关性、绘制数值数据之间的相关性、以及使用分类散点图绘制分类数据和数值数据之间的相关性。十分有趣的是,可以在Seaborn 中将图表元素叠加在一起。

留言 点赞 关注

我们一起分享AI学习与发展的干货

欢迎关注全平台AI垂类自媒体 “读芯术”

相关推荐