沧海拾珠

Pandas 基本绘图

1. 图的类型。主要有3种,线状图,直方图,饼图。

如果是数列数据

1
2
3
4
5
6
7
8
import matplotlib.pyplot as plt
from matplotlib import rcParams
rcParams['figure.figsize'] = 5, 4 #设置图片大小
x = range(1,6)
y = [2,4,6,8,10]
plt.plot(x,y) #line chart
plt.bar(x,y) #bar chart
plt.bar(y) #pie chart

如果是dataframe数据

1
2
3
4
DataFrame.plot(kind = 'line')
DataFrame.plot(kind = 'bar')
DataFrame.plot(kind = 'barh')
DataFrame.plot(kind = 'pie')

2. 图的颜色和尺寸。

可以使用color定义单一颜色,如果想要颜色更加丰富多彩,可以使用colormap。

1
2
3
4
DataFrame.plot(color = 'red', figsize = (10,3))
DataFrame.plot(colormap = 'rainbow', figsize = (10,3))
#rcParams['figure.figsize'] = 5,4 也可以用来设置图片大小

3. 使用seaborn画图。Seaborn是基于matplotlib的一个可视化工具库,画的图更加的精美。需要单独安装。

$pip install seaborn

1
2
3
4
import seaborn as sns
df = pd.read_csv('olympics.csv',skiprows = 4)
new_df = df[df.Edition == 1896]
sns.countplot(x = 'Medal', data = new_df, hue = 'Gender')