在2月6日的编程课程中,我成功地掌握了以下几项技能:
- 了解了CSV文件的概念;
- 熟悉了pandas库的使用;
- 理解了DataFrame的结构;
- 学会了使用pandas库的read_csv()函数读取CSV文件;
- 学会了操作DataFrame的index、columns和values属性;
- 掌握了使用matplotlib.pyplot模块的colorbar()函数添加颜色映射;
- 学会了使用坐标轴对象的imshow()方法绘制热力图;
- 掌握了使用坐标轴对象的set_xticks()方法设置横轴刻度标签;
- 掌握了使用坐标轴对象的set_yticks()方法设置纵轴刻度标签;
- 学会了使用坐标轴对象的set_ticks_position()方法设置横轴刻度的显示位置。
本节课代码:
import pandas as pd
import matplotlib.pyplot as plt
import matplotlib.font_manager as fm
font = fm.FontProperties(fname='./simhei.ttf')
data = pd.read_csv('data/relationShips.csv', index_col=0)
columns = data.columns
values = data.values
index = data.index
figure, ax = plt.subplots(num='人物共现关系分析')
im = ax.imshow(values,cmap='Oranges',origin='upper')
plt.colorbar(im)
ax.set_yticks(range(20),labels=index,fontproperties=font)
ax.set_xticks(range(20),labels=index,fontproperties=font,rotation=90)
ax.xaxis.set_ticks_position('top')
ax.set_title('《三国演义》中重要人物共现次数热力图',fontproperties=font,fontsize=12)
plt.show()