这一学习一下条形图和散点图的画法。
首先是条形图:
import pandas as pd
import matplotlib.pyplot as plt
import numpy as np
from numpy import arange
data =pd.read_csv('fandango_scores.csv')
data.head()
这是一个电影评分的数据集,包含了电影名称和不同对象的评分。
columns =['Fandango_Stars','RT_norm','RT_user_norm','RT_norm_round','Metacritic_norm_round','IMDB_norm_round']
data1 = data.loc[0,columns].values
data1
从数据集中选出一部分数据,来进行画图,这次选取部分列的第一行。
print(data1)
bar_positions = arange(6)+1.75
print(bar_positions)
fig, axes=plt.subplots()
axes.bar(bar_positions,data1, 0.3)
plt.show()
然后进行画图部分。bar_position是为了制定柱状图的柱子距离原点的距离。
data1是指定各个柱子的高度。
fig, axes=plt.subplots()画出一块空白画布。
axes.bar传递进去三个参数,柱子距离,柱子高度,柱子宽度。
然后进行展示。
bar_positions = arange(6)+1.75
print(bar_positions)
fig, axes=plt.subplots()
axes.barh(bar_positions,data1, 0.3,color='y')
plt.show()
把bar()和barh()分别用来画垂直和水平柱状图,可以选择颜色参数来改变柱子颜色。
bar_positions = arange(6)+1.75
fig, axes=plt.subplots()
axes.barh(bar_positions,data1, 0.3,color='y')
axes.set_xticklabels(columns,rotation=45)
axes.set_xlabel('Rating Source')
axes.set_ylabel('Average Rating')
axes.set_title('Average User Rating')
增加x,y轴名称,表头信息。
散点图(scatter):
fig,ax = plt.subplots()
ax.scatter(data['Fandango_Stars'],data['RT_user_norm'])
ax.set_xlabel('Fandango_Stars')
ax.set_ylabel('RT_user_norm')
plt.show()
设定一张散点图,从数据集中选取两组对象的评分数据,传入散点图的函数中,再对x,y轴进行命名,就完成了绘制。