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)
成果展示