【接前篇】Python Plotly数据可视化全国Used BMW

import pandas as pd
import numpy as np
import plotly.express as px
import plotly

data = pd.read_csv('baoma_more.csv')
data_sorted = data.sort_values(by=['Name'])
models = data_sorted['Name']

bmw_1 = []
bmw_2 = []
bmw_3 = [] 
bmw_4 = []
bmw_5 = []
...
...
...

for model_info in models:
	if '宝马1系' in str(model_info):
		bmw_1.append(model_info)
	if '宝马2系' in str(model_info):
		bmw_2.append(model_info)
	if '宝马3系' in str(model_info):
		bmw_3.append(model_info)
	if '宝马4系' in str(model_info):
		bmw_4.append(model_info)
	if '宝马5系' in str(model_info):
		bmw_5.append(model_info)
	if '宝马6系' in str(model_info):
		bmw_6.append(model_info)
    ...
    ...

半截发现之前那样属实头脑短路，应该定义函数来实现 后半部分定义个函数 

def get_model_list(model_name,model_list,sub_model_list):
	for model in model_list:
		if model_name in model:
			sub_model_list.append(model)

li_520 = []
li_525 = []
li_523 = []
li_528 = []
li_530li = []
li_535li = []
li_Le = []
li_540 = []
li_5_gt = []
get_model_list('525',bmw_5,li_525)
get_model_list('530Li',bmw_5,li_530li)
get_model_list('520',bmw_5,li_520)
get_model_list('525',bmw_5,li_525)
get_model_list('523',bmw_5,li_523)
get_model_list('528',bmw_5,li_528)
get_model_list('535Li',bmw_5,li_535li)
get_model_list('Le',bmw_5,li_Le)
get_model_list('540',bmw_5,li_540)
get_model_list('GT',bmw_5,li_5_gt)

data = dict(
    character=["宝马1系", "宝马2系", "宝马3系", "宝马4系",'宝马5系','宝马6系','宝马7系','宝马M_Power','宝马X1','宝马X2','宝马X3','宝马X4','宝马X5','宝马X6','宝马Z4','宝马i8',
'宝马520','宝马525','宝马523','宝马528','宝马530','宝马535','宝马Le','宝马540','宝马5GT'
],
    parent=['BMW','BMW','BMW','BMW','BMW','BMW','BMW','BMW','BMW','BMW','BMW','BMW','BMW','BMW','BMW','BMW','宝马5系','宝马5系','宝马5系','宝马5系','宝马5系','宝马5系','宝马5系','宝马5系','宝马5系'],
    value=[len(bmw_1),len(bmw_2),len(bmw_3),len(bmw_4),len(bmw_5),len(bmw_6),len(bmw_7),len(bmw_M),len(bmw_x1),len(bmw_x2),len(bmw_x1),len(bmw_x2),len(bmw_x5),len(bmw_x6),len(bmw_z4),len(bmw_i8),
len(li_520),len(li_525),len(li_523),len(li_528),len(li_530li),len(li_535li),len(li_Le),len(li_540),len(li_5_gt)
]
)

fig =px.sunburst(
    data,
    names='character',
    parents='parent',
    values='value',
)

plotly.offline.plot(fig)

成果展示