五十常用图的条形图 树形图代码注释

条形图

import pandas as pd
import matplotlib.pyplot as plt
import random

# Import Data
df_raw = pd.read_csv("实例.csv")

# Prepare Data
df = df_raw.groupby('manufacturer').size().reset_index(name='counts')
n = df['manufacturer'].unique().__len__()+1#unique唯一值，将数据类型变成集合，__len__用法和len相差不远。
all_colors = list(plt.cm.colors.cnames.keys())#数字或者颜色参数转化为颜色值（字典形式）取key
random.seed(100)
c = random.choices(all_colors, k=n)#在all_colors中随机选取n个数值

# Plot Bars
plt.figure(figsize=(16,10), dpi= 80)#画布
plt.bar(df['manufacturer'], df['counts'], color=c, width=.5)#x轴标签，高度，条形宽度，默认0—1
for i, val in enumerate(df['counts'].values):#将列表变成一个索引序列
    plt.text(i, val, float(val), horizontalalignment='center', verticalalignment='bottom', fontdict={'fontweight':500, 'size':12})#x,y对应的值，float标注的文字，可以简写ha，va，字体的粗细与大小

# Decoration
plt.gca().set_xticklabels(df['manufacturer'], rotation=60, horizontalalignment= 'right')#获得axes对象，x轴坐标斜着60度显示，lebel显示位置
plt.title("Number of Vehicles by Manaufacturers", fontsize=22)
plt.ylabel('# Vehicles')
plt.ylim(0, 45)
plt.show()

树形图

import pandas as pd
import matplotlib.pyplot as plt
import squarify#将数据生成基于matplotlib的树状图可视化
df_raw = pd.read_csv("实例.csv")

# Prepare Data
df = df_raw.groupby('class').size().reset_index(name='counts')#按照class的值分组，size统计数据，reset_index建立一个新的索引
labels = df.apply(lambda x: str(x[0]) + "\n (" + str(x[1]) + ")", axis=1)#lambda一种函数，冒号前是参数，可以有多个，冒号后为表达式；axis=0:将一列数据作为lamda的数据
#df.apply(func,axis=1/0)
sizes = df['counts'].values.tolist()#values.tolist将数组转化为列表
colors = [plt.cm.Spectral(i/float(len(labels))) for i in range(len(labels))]#不同的取值颜色不同

# Draw Plot
plt.figure(figsize=(12,8), dpi= 80)#figsize表示页面布局的大小，单位为英寸，dpi每英寸的像素点数
squarify.plot(sizes=sizes,label=labels, color=colors, alpha=.8)#sizes的个数说明了分支数的个数，有几就是几，label显示标签，colors颜色，不同的取值有不同的颜色，alpha表示颜色的淡化程度

# Decorate
plt.title('Treemap of Vechile Class')
plt.axis('off')#不显示坐标轴
plt.show()