Python抖音数据分析可视化源码

# Python分析抖音信息可视化 from pyecharts.charts import Pie, Bar, TreeMap, Map, Geo from wordcloud import WordCloud, ImageColorGenerator from pyecharts import options as opts import matplotlib.pyplot as plt from PIL import Image import pandas as pd import numpy as np import jieba df = pd.read_csv('douyin.csv', header=0, encoding='utf-8-sig') print(df) def create_gender(data): df = data.copy() # 修改数值 df.loc[df.gender == '0', 'gender'] = '未知' df.loc[df.gender == '1', 'gender'] = '男性' df.loc[df.gender == '2', 'gender'] = '女性' # 根据性别分组 gender_message = df.groupby(['gender']) # 对分组后的结果进行计数 gender_com = gender_message['gender'].agg(['count']) gender_com.reset_index(inplace=True) # 饼图数据 attr = gender_com['gender'] v1 = gender_com['count'] # 初始化配置 pie = Pie(init_opts=opts.InitOpts(width="800px", height="400px")) # 添加数据，设置半径 pie.add("", [list(z) for z in zip(attr, v1)], radius=["40%", "75%"]) # 设置全局配置项，标题、图例、工具箱(下载图片) pie.set_global_opts(title_opts=opts.TitleOpts(title="抖音大V性别分布情况", pos_left="center"), legend_opts=opts.LegendOpts(orient="vertical", pos_left="left"), toolbox_opts=opts.ToolboxOpts(is_show=True, feature={"saveAsImage": {}})) # 设置系列配置项，标签样式 pie.set_series_opts(label_opts=opts.LabelOpts(is_show=True, formatter="{b}:{d}%")) pie.render("抖音大V性别分布情况.html") create_gender(df) def create_likes(df): # 排序，降序 df = df.sort_values('likes', ascending=False) # 获取TOP10的数据 attr = df['name'][0:10] v1 = [float('%.1f' % (float(i) / 100000000)) for i in df['likes'][0:10]] # 初始化配置 bar = Bar(init_opts=opts.InitOpts(width="800px", height="400px")) # x轴数据 bar.add_xaxis(list(reversed(attr.tolist()))) # y轴数据 bar.add_yaxis("", list(reversed(v1))) # 设置全局配置项，标题、工具箱(下载图片)、y轴分割线 bar.set_global_opts(title_opts=opts.TitleOpts(title="抖音大V点赞数TOP10(亿)", pos_left="center", pos_top="18"), toolbox_opts=opts.ToolboxOpts(is_show=True, feature={"saveAsImage": {}}), xaxis_opts=opts.AxisOpts(splitline_opts=opts.SplitLineOpts(is_show=True))) # 设置系列配置项，标签样式 bar.set_series_opts(label_opts=opts.LabelOpts(is_show=True, position="right", color="black")) bar.reversal_axis() bar.render("抖音大V点赞数TOP10(亿).html") create_likes(df) def create_cut_likes(df): # 将数据分段 Bins = [0, 1000000, 5000000, 10000000, 25000000, 50000000, 100000000, 5000000000] Labels = ['0-100', '100-500', '500-1000', '1000-2500', '2500-5000', '5000-10000', '10000以上'] len_stage = pd.cut(df['likes'], bins=Bins, labels=Labels).value_counts().sort_index() # 获取数据 attr = len_stage.index.tolist() v1 = len_stage.values.tolist() # 生成柱状图 bar = Bar(init_opts=opts.InitOpts(width="800px", height="400px")) bar.add_xaxis(attr) bar.add_yaxis("", v1) bar.set_global_opts(title_opts=opts.TitleOpts(title="抖音大V点赞数分布情况(万)", pos_left="center", pos_top="18"), toolbox_opts=opts.ToolboxOpts(is_show=True, feature={"saveAsImage": {}}), yaxis_opts=opts.AxisOpts(splitline_opts=opts.SplitLineOpts(is_show=True))) bar.set_series_opts(label_opts=opts.LabelOpts(is_show=True, position="top", color="black")) bar.render("抖音大V点赞数分布情况(万).html") create_cut_likes(df) def create_fans(df): df = df.sort_values('fans', ascending=False) attr = df['name'][0:10] v1 = ['%.1f' % (float(i) / 10000) for i in df['fans'][0:10]] bar = Bar(init_opts=opts.InitOpts(width="800px", height="400px")) bar.add_xaxis(list(reversed(attr.tolist()))) bar.add_yaxis("", list(reversed(v1))) bar.set_global_opts(title_opts=opts.TitleOpts(title="抖音大V粉丝数TOP10(万)", pos_left="center", pos_top="18"), toolbox_opts=opts.ToolboxOpts(is_show=True, feature={"saveAsImage": {}}), xaxis_opts=opts.AxisOpts(splitline_opts=opts.SplitLineOpts(is_show=True))) bar.set_series_opts(label_opts=opts.LabelOpts(is_show=True, position="right", color="black")) bar.reversal_axis() bar.render("抖音大V粉丝数TOP10(万).html") create_fans(df) def create_cut_fans(df): Bins = [0, 1500000, 2000000, 5000000, 10000000, 25000000, 200000000] Labels = ['0-150', '150-200', '200-500', '500-1000', '1000-2500', '5000以上'] len_stage = pd.cut(df['fans'], bins=Bins, labels=Labels).value_counts().sort_index() attr = len_stage.index.tolist() v1 = len_stage.values.tolist() bar = Bar(init_opts=opts.InitOpts(width="800px", height="400px")) bar.add_xaxis(attr) bar.add_yaxis("", v1) bar.set_global_opts(title_opts=opts.TitleOpts(title="抖音大V粉丝数分布情况(万)", pos_left="center", pos_top="18"), toolbox_opts=opts.ToolboxOpts(is_show=True, feature={"saveAsImage": {}}), yaxis_opts=opts.AxisOpts(splitline_opts=opts.SplitLineOpts(is_show=True))) bar.set_series_opts(label_opts=opts.LabelOpts(is_show=True, position="top", color="black")) bar.render("抖音大V粉丝数分布情况(万).html") create_cut_fans(df) def create_comments(df): df = df.sort_values('comments', ascending=False) attr = df['name'][0:10] v1 = ['%.1f' % (float(i) / 100000000) for i in df['comments'][0:10]] bar = Bar(init_opts=opts.InitOpts(width="800px", height="400px")) bar.add_xaxis(list(reversed(attr.tolist()))) bar.add_yaxis("", list(reversed(v1))) bar.set_global_opts(title_opts=opts.TitleOpts(title="抖音大V评论数TOP10(亿)", pos_left="center", pos_top="18"), toolbox_opts=opts.ToolboxOpts(is_show=True, feature={"saveAsImage": {}}), xaxis_opts=opts.AxisOpts(splitline_opts=opts.SplitLineOpts(is_show=True))) bar.set_series_opts(label_opts=opts.LabelOpts(is_show=True, position="right", color="black")) bar.reversal_axis() bar.render("抖音大V评论数TOP10(亿).html") create_comments(df) def create_shares(df): df = df.sort_values('shares', ascending=False) attr = df['name'][0:10] v1 = ['%.1f' % (float(i) / 100000000) for i in df['shares'][0:10]] bar = Bar(init_opts=opts.InitOpts(width="800px", height="400px")) bar.add_xaxis(list(reversed(attr.tolist()))) bar.add_yaxis("", list(reversed(v1))) bar.set_global_opts(title_opts=opts.TitleOpts(title="抖音大V分享数TOP10(亿)", pos_left="center", pos_top="18"), toolbox_opts=opts.ToolboxOpts(is_show=True, feature={"saveAsImage": {}}), xaxis_opts=opts.AxisOpts(splitline_opts=opts.SplitLineOpts(is_show=True))) bar.set_series_opts(label_opts=opts.LabelOpts(is_show=True, position="right", color="black")) bar.reversal_axis() bar.render("抖音大V分享数TOP10(亿).html") create_shares(df) def create_type_likes(df): # 分组求和 likes_type_message = df.groupby(['category']) likes_type_com = likes_type_message['likes'].agg(['sum']) likes_type_com.reset_index(inplace=True) # 处理数据 dom = [] for name, num in zip(likes_type_com['category'], likes_type_com['sum']): data = {} data['name'] = name data['value'] = num dom.append(data) print(dom) # 初始化配置 treemap = TreeMap(init_opts=opts.InitOpts(width="800px", height="400px")) # 添加数据 treemap.add('', dom) # 设置全局配置项，标题、工具箱(下载图片) treemap.set_global_opts(title_opts=opts.TitleOpts(