**复习:**回顾学习完第一章,我们对泰坦尼克号数据有了基本的了解,也学到了一些基本的统计方法,第二章中我们学习了数据的清理和重构,使得数据更加的易于理解;今天我们要学习的是第二章第三节:数据可视化,主要给大家介绍一下Python数据可视化库Matplotlib,在本章学习中,你也许会觉得数据很有趣。在打比赛的过程中,数据可视化可以让我们更好的看到每一个关键步骤的结果如何,可以用来优化方案,是一个很有用的技巧。
# 加载所需的库
# 如果出现 ModuleNotFoundError: No module named 'xxxx'
# 你只需要在终端/cmd下 pip install xxxx 即可
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
#加载result.csv这个数据
result = pd.read_csv("result.csv")
result.head()
Unnamed: 0 | PassengerId | Survived | Pclass | Name | Sex | Age | SibSp | Parch | Ticket | Fare | Cabin | Embarked | |
---|---|---|---|---|---|---|---|---|---|---|---|---|---|
0 | 0 | 1 | 0 | 3 | Braund, Mr. Owen Harris | male | 22.0 | 1 | 0 | A/5 21171 | 7.2500 | NaN | S |
1 | 1 | 2 | 1 | 1 | Cumings, Mrs. John Bradley (Florence Briggs Th... | female | 38.0 | 1 | 0 | PC 17599 | 71.2833 | C85 | C |
2 | 2 | 3 | 1 | 3 | Heikkinen, Miss. Laina | female | 26.0 | 0 | 0 | STON/O2. 3101282 | 7.9250 | NaN | S |
3 | 3 | 4 | 1 | 1 | Futrelle, Mrs. Jacques Heath (Lily May Peel) | female | 35.0 | 1 | 0 | 113803 | 53.1000 | C123 | S |
4 | 4 | 5 | 0 | 3 | Allen, Mr. William Henry | male | 35.0 | 0 | 0 | 373450 | 8.0500 | NaN | S |
《Python for Data Analysis》第九章
【思考】最基本的可视化图案有哪些?分别适用于那些场景?(比如折线图适合可视化某个属性值随时间变化的走势)
【总结】常见可视化图表 更多图表可参考:https://blog.csdn.net/weixin_41545602/article/details/111828028
【注】可视化图表使用的注意事项
【参考】https://nn.sumaart.com/share/119.html
**matplotlib.pyplot.bar(x,height, width,*,align=‘center’,kwargs)
参数:
其他可选参数:
#代码编写
sex_survived = result.groupby(["Sex"])["Survived"].sum()
sex_survived
Sex
female 233
male 109
Name: Survived, dtype: int64
sex_survived_bar = plt.bar(sex_survived.index, height = sex_survived.values, width = 0.5)
plt.xlabel("Sex")
plt.ylabel("Survived")
plt.show()
【思考】计算出泰坦尼克号数据集中男女中死亡人数,并可视化展示?如何和男女生存人数可视化柱状图结合到一起?看到你的数据可视化,说说你的第一感受(比如:你一眼看出男生存活人数更多,那么性别可能会影响存活率)。
sex = result.groupby(["Sex"])["Sex"].count()
sex
Sex
female 314
male 577
Name: Sex, dtype: int64
#思考题回答
sex_dead = result.groupby(["Sex"])["Sex"].count()-result.groupby(["Sex"])["Survived"].sum()
sex_dead
Sex
female 81
male 468
dtype: int64
sex_dead_bar = plt.bar(sex_dead.index, sex_dead.values)
plt.show()
把上述两个柱状图结合起来,可以考虑使用堆积柱状图
法一:使用plt的绘图方法
先bar底部图,再bar顶部图,并用bottom设置底部数据,然后用legend函数检测元素
**matplotlib.pyplot.legend(*args, kwargs)
在坐标轴上放置一个图例。
当不传递任何额外参数时,将自动确定要添加到图例的元素。
需要指定个别标签的时候,可以传递列表参数进行指定。
plt.bar(sex_survived.index, sex_survived.values, label = "survived-1", width = 0.5)
plt.bar(sex_dead.index, sex_dead.values, bottom=sex_survived.values, label = "deaded-0", width = 0.5)
plt.legend()
plt.show()
法二:使用pandas的绘图方法
将Survived列的数据分组后变成表格结构(unstack),并在此基础上进行画图
**DataFrame.plot(*args, kwargs)
参数:
test = result.groupby(['Sex','Survived'])['Survived'].count().unstack()
test
Survived | 0 | 1 |
---|---|---|
Sex | ||
female | 81 | 233 |
male | 468 | 109 |
test.plot(kind='bar',stacked='True')
通过上图可以直观看出,男性的存活率低于女性存活率
【提示】对于这种统计性质的且用折线表示的数据,你可以考虑将数据排序或者不排序来分别表示。看看你能发现什么?
#代码编写
# 计算不同票价中生存与死亡人数 1表示生存,0表示死亡
fare_person = result.groupby(["Fare", "Survived"])["Fare"].count()
fare_person
Fare Survived
0.0000 0 14
1 1
4.0125 0 1
5.0000 0 1
6.2375 0 1
..
247.5208 1 1
262.3750 1 2
263.0000 0 2
1 2
512.3292 1 3
Name: Fare, Length: 330, dtype: int64
fare_person.sort_values(ascending=False)
Fare Survived
8.0500 0 38
7.8958 0 37
13.0000 0 26
7.7500 0 22
13.0000 1 16
..
20.2500 1 1
0 1
79.6500 0 1
18.7875 1 1
15.0500 0 1
Name: Fare, Length: 330, dtype: int64
fare_person.sort_values(ascending=False).plot(title = "sort")
fare_person.plot(title = "unsort")
可以看出,排序后的趋势更为明显,更容易观察趋势以及大概的分布情况。
#代码编写
# 1表示生存,0表示死亡
pclass_person = result.groupby(["Pclass","Survived"])["Survived"].count()
pclass_person
Pclass Survived
1 0 80
1 136
2 0 97
1 87
3 0 372
1 119
Name: Survived, dtype: int64
pclass_person.unstack().plot(kind = "bar", stacked = True)
pclass_person.unstack().plot(kind = "bar")
【思考】看到这个前面几个数据可视化,说说你的第一感受和你的总结
思考题回答:
女性存活人数以及存活率均高于男性
1级客舱的存活率最高;3级客舱的人数最多但存活率最低
#代码编写
age_person = result.groupby(["Age","Survived"])["Survived"].count().unstack()
age_person
Survived | 0 | 1 |
---|---|---|
Age | ||
0.42 | NaN | 1.0 |
0.67 | NaN | 1.0 |
0.75 | NaN | 2.0 |
0.83 | NaN | 2.0 |
0.92 | NaN | 1.0 |
... | ... | ... |
70.00 | 2.0 | NaN |
70.50 | 1.0 | NaN |
71.00 | 2.0 | NaN |
74.00 | 1.0 | NaN |
80.00 | NaN | 1.0 |
88 rows × 2 columns
age_person.plot(kind = "bar", figsize = (20,5), grid = True)
#代码编写
result.Age[result.Pclass == 1].plot(kind='kde', label ="1")
result.Age[result.Pclass == 2].plot(kind='kde', label ="2")
result.Age[result.Pclass == 3].plot(kind='kde', label ="3")
plt.legend()
sex_pclass = result.groupby(["Pclass", "Sex"])["Sex"].count().unstack()
sex_pclass
Sex | female | male |
---|---|---|
Pclass | ||
1 | 94 | 122 |
2 | 76 | 108 |
3 | 144 | 347 |
sex_pclass.plot(kind = "bar", stacked = True)
【思考】上面所有可视化的例子做一个总体的分析,你看看你能不能有自己发现
【补充】核密度图
核密度图本质是直方图的拟合曲线。
思考题回答:
通过上述图表,可以发现:
在这之前,我们也有部分观察结论:
【总结】到这里,我们的可视化就告一段落啦,如果你对数据可视化极其感兴趣,你还可以了解一下其他可视化模块,如:pyecharts,bokeh等。
如果你在工作中使用数据可视化,你必须知道数据可视化最大的作用不是炫酷,而是最快最直观的理解数据要表达什么,你觉得呢?