From 8ff56b8929e18c3cd55b2b3f0608e1725ef9d453 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=8A=A6=E7=AC=99?= <3140038914@qq.com> Date: Thu, 25 Apr 2024 13:10:14 +0800 Subject: [PATCH] =?UTF-8?q?=E5=AE=8C=E6=88=90=E4=BA=86=E5=AF=B9=E4=BA=8Eex?= =?UTF-8?q?cel=E5=8A=9F=E8=83=BD=E7=9A=84=E4=B9=A6=E5=86=99=EF=BC=8C?= =?UTF-8?q?=E5=AE=8C=E6=88=90=E4=BA=86=E6=B5=8B=E8=AF=95=E7=94=A8=E4=BE=8B?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- service/CsvService.py | 3 +- service/ExcelService.py | 109 ++++++++++++++++++++- service/IFileService.py | 3 +- test/Excel_test.py | 90 +++++++++++++++++- tool/tttt.py | 205 ++++++++++++++++++++++++++++++++++++++++ 5 files changed, 403 insertions(+), 7 deletions(-) create mode 100644 tool/tttt.py diff --git a/service/CsvService.py b/service/CsvService.py index 8df1c85..6e3a8fd 100644 --- a/service/CsvService.py +++ b/service/CsvService.py @@ -1,7 +1,6 @@ from IFileService import IFileService -from entity.BilibiliVideo import BilibiliVideo class CsvService(IFileService): - def save(self, filePath, videoList: list[BilibiliVideo]): + def save(self, filePath, videoList): raise NotImplementedError diff --git a/service/ExcelService.py b/service/ExcelService.py index f3f488c..934688a 100644 --- a/service/ExcelService.py +++ b/service/ExcelService.py @@ -1,5 +1,7 @@ + from IFileService import IFileService from entity.BilibiliVideo import BilibiliVideo +from tool import tttt class ExcelService(IFileService): @@ -9,5 +11,108 @@ class ExcelService(IFileService): """ pass - def save(self, filePath, videoList: list[BilibiliVideo]): - raise NotImplementedError + def save(self, filePath, videoList): + tttt.write_to_excel(videoList,filePath) + tttt.calculate_ratio_and_update(filePath, 'Sheet') + import openpyxl + from openpyxl.chart import ScatterChart, BarChart, AreaChart, Reference + + # 读取Excel文件 + wb = openpyxl.load_workbook('bilibili_videos.xlsx') + ws = wb.active + + # 创建柱状图1 + bar_chart1 = BarChart() + bar_chart1.title = "观众对于视频的认可度" + bar_chart1.y_axis.title = "Data" + bar_chart1.x_axis.title = "Index" + + # 设置柱状图1数据 + bar_data1 = Reference(ws, min_col=17, min_row=2, max_row=ws.max_row) + bar_categories1 = Reference(ws, min_col=2, min_row=2, max_row=ws.max_row) + bar_chart1.add_data(bar_data1, titles_from_data=True) + bar_chart1.set_categories(bar_categories1) + + # 添加柱状图1到工作表 + ws.add_chart(bar_chart1, "V1") + + # 创建柱状图 + bar_chart = BarChart() + bar_chart.title = "收益" + bar_chart.y_axis.title = "Data" + bar_chart.x_axis.title = "Index" + + # 设置柱状图数据 + bar_data = Reference(ws, min_col=18, min_row=2, max_row=ws.max_row) + bar_categories = Reference(ws, min_col=2, min_row=2, max_row=ws.max_row) + bar_chart.add_data(bar_data, titles_from_data=True) + bar_chart.set_categories(bar_categories) + + # 添加柱状图到工作表 + ws.add_chart(bar_chart, "AD1") + + # 创建面积图 + area_chart = AreaChart() + area_chart.title = "视频实用性" + area_chart.y_axis.title = "Data" + area_chart.x_axis.title = "Index" + + # 设置面积图数据 + area_data = Reference(ws, min_col=19, min_row=2, max_row=ws.max_row) + area_categories = Reference(ws, min_col=2, min_row=2, max_row=ws.max_row) + area_chart.add_data(area_data, titles_from_data=True) + area_chart.set_categories(area_categories) + + # 添加面积图到工作表 + ws.add_chart(area_chart, "AL1") + + # 创建柱状图 + bar_chart_2 = BarChart() + bar_chart_2.title = "视频惊艳程度" + bar_chart_2.y_axis.title = "Data" + bar_chart_2.x_axis.title = "Index" + + # 设置柱状图数据 + bar_data_2 = Reference(ws, min_col=20, min_row=2, max_row=ws.max_row) + bar_categories_2 = Reference(ws, min_col=2, min_row=2, max_row=ws.max_row) + bar_chart_2.add_data(bar_data_2, titles_from_data=True) + bar_chart_2.set_categories(bar_categories_2) + + # 添加柱状图到工作表 + ws.add_chart(bar_chart_2, "AT1") + + # 创建柱状图 + bar_chart_3 = BarChart() + bar_chart_3.title = "视频的互动性" + bar_chart_3.y_axis.title = "Data" + bar_chart_3.x_axis.title = "Index" + + # 设置柱状图数据 + bar_data_3 = Reference(ws, min_col=21, min_row=2, max_row=ws.max_row) + bar_categories_3 = Reference(ws, min_col=2, min_row=2, max_row=ws.max_row) + bar_chart_3.add_data(bar_data_3, titles_from_data=True) + bar_chart_3.set_categories(bar_categories_3) + + # 添加柱状图到工作表 + ws.add_chart(bar_chart_3, "BB1") + + # 保存Excel文件 + wb.save('bilibili_videos_with_charts.xlsx') + + texts = [ + "approve", + "money", + "practical", + "Stunning", + "interaction" + ] + tttt.write_english_texts(filePath, 'Sheet', texts) + + + + + + + + + diff --git a/service/IFileService.py b/service/IFileService.py index ab4c576..b0d65be 100644 --- a/service/IFileService.py +++ b/service/IFileService.py @@ -1,11 +1,10 @@ from abc import abstractmethod, ABCMeta -from entity.BilibiliVideo import BilibiliVideo class IFileService(metaclass=ABCMeta): @abstractmethod - def save(self,filePath,videoList:list[BilibiliVideo]): + def save(self, filePath, videoList): """ 保存到文件，高级参数在init里面写 :param filePath: 文件保存路径 diff --git a/test/Excel_test.py b/test/Excel_test.py index c10a958..8dba95c 100644 --- a/test/Excel_test.py +++ b/test/Excel_test.py @@ -1,3 +1,91 @@ +from entity.BilibiliVideo import BilibiliVideo +from tool import tttt + + class TestExcel: def test_Excel(self): - pass \ No newline at end of file + import openpyxl + + def write_to_excel(videos, filename): + # 创建一个新的Excel工作簿 + wb = openpyxl.Workbook() + ws = wb.active + ws.title = "Bilibili Videos" + + # 写入表头 + headers = ["bvId", "title", "url", "uploadTime", "uploadTimeText", "topNo", + "viewCount", "likeCount", "coinCount", "favoriteCount", + "commentCount", "bulletCount", "creatorId", "creatorName", "creatorFanCount"] + for col_num, header in enumerate(headers, 1): + ws.cell(row=1, column=col_num, value=header) + + # 写入视频数据 + for row_num, video in enumerate(videos, 2): + ws.cell(row=row_num, column=1, value=video.bvId) + ws.cell(row=row_num, column=2, value=video.title) + ws.cell(row=row_num, column=3, value=video.url) + ws.cell(row=row_num, column=4, value=video.uploadTime) + ws.cell(row=row_num, column=5, value=video.uploadTimeText) + ws.cell(row=row_num, column=6, value=video.topNo) + ws.cell(row=row_num, column=7, value=video.viewCount) + ws.cell(row=row_num, column=8, value=video.likeCount) + ws.cell(row=row_num, column=9, value=video.coinCount) + ws.cell(row=row_num, column=10, value=video.favoriteCount) + ws.cell(row=row_num, column=11, value=video.commentCount) + ws.cell(row=row_num, column=12, value=video.bulletCount) + ws.cell(row=row_num, column=13, value=video.creatorId) + ws.cell(row=row_num, column=14, value=video.creatorName) + ws.cell(row=row_num, column=15, value=video.creatorFanCount) + + # 保存Excel文件 + wb.save(filename) + + # 示例用法 + video1 = BilibiliVideo("bv123456", "视频标题1", "http://video1.com", 1620000000, 1, 1000, 500, 200, 50, 100, + 300, "creator123", "up主1", 10000) + video2 = BilibiliVideo("bv789012", "视频标题2", "http://video2.com", 1621000000, 2, 2000, 1000, 400, 60, 200, + 400, "creator456", "up主2", 20000) + + videos = [video1, video2] + write_to_excel(videos, "bilibili_videos.xlsx") + + pass + def test_WTExcel(): + # 测试数据 + test_cases = [ + BilibiliVideo("BV1a4411C7i2", "视频标题1", "https://www.bilibili.com/video/BV1a4411C7i2", 1648927200, 1, + 100000, 5000, 2000, 3000, 1000, 500, "up123456", "UP主A", 1000000), + BilibiliVideo("BV1F5411S8h3", "视频标题2", "https://www.bilibili.com/video/BV1F5411S8h3", 1648830800, 2, + 80000, 4000, 1500, 2000, 800, 400, "up234567", "UP主B", 800000), + BilibiliVideo("BV1bW411d9c4", "视频标题3", "https://www.bilibili.com/video/BV1bW411d9c4", 1648734400, 3, + 60000, 3000, 1000, 1500, 600, 300, "up345678", "UP主C", 600000), + BilibiliVideo("BV1qy411z5k5", "视频标题4", "https://www.bilibili.com/video/BV1qy411z5k5", 1648638000, 4, + 40000, 2000, 800, 1000, 400, 200, "up456789", "UP主D", 400000), + BilibiliVideo("BV1R4411J0a6", "视频标题5", "https://www.bilibili.com/video/BV1R4411J0a6", 1648541600, 5, + 20000, 1000, 500, 700, 200, 100, "up567890", "UP主E", 200000), + BilibiliVideo("BV1gW411H8d7", "视频标题6", "https://www.bilibili.com/video/BV1gW411H8d7", 1648445200, 6, + 10000, 500, 300, 400, 100, 50, "up678901", "UP主F", 100000), + BilibiliVideo("BV1py411t6v8", "视频标题7", "https://www.bilibili.com/video/BV1py411t6v8", 1648348800, 7, + 5000, 300, 200, 300, 50, 30, "up789012", "UP主G", 50000), + BilibiliVideo("BV1Qy411S3x9", "视频标题8", "https://www.bilibili.com/video/BV1Qy411S3x9", 1648252400, 8, + 2000, 100, 100, 200, 20, 10, "up890123", "UP主H", 20000), + BilibiliVideo("BV1a4411C4y0", "视频标题9", "https://www.bilibili.com/video/BV1a4411C4y0", 1648156000, 9, + 1000, 50, 50, 100, 10, 5, "up901234", "UP主I", 10000), + BilibiliVideo("BV1F5411C2r1", "视频标题10", "https://www.bilibili.com/video/BV1F5411C2r1", 1648059600, + 10, 500, 30, 20, 50, 5, 3, "up012345", "UP主J", 5000) + ] + + # 将测试数据写入 Excel 文件 + write_to_excel(test_cases, 'bilibili_videos.xlsx') + def test_c_r_a_update(): + + # 示例用法 + file_path = 'bilibili_videos.xlsx' # Excel 文件路径 + sheet_name = 'Sheet' # 工作表名称 + + tttt.calculate_ratio_and_update(file_path, sheet_name) + print("Data analysis written to the Excel file.") + + + + diff --git a/tool/tttt.py b/tool/tttt.py new file mode 100644 index 0000000..37fce5a --- /dev/null +++ b/tool/tttt.py @@ -0,0 +1,205 @@ +import openpyxl +from entity.BilibiliVideo import BilibiliVideo + +# 创建一个函数，用于将 BilibiliVideo 实例的属性写入 Excel 表格中 +def write_to_excel(video_instances, excel_filename): + wb = openpyxl.Workbook() + ws = wb.active + + # 添加表头 + ws.append(['bvId', 'title', 'url', 'uploadTime', 'uploadTimeText', 'topNo', 'viewCount', 'likeCount', 'coinCount', + 'favoriteCount', 'commentCount', 'bulletCount', 'creatorId', 'creatorName', 'creatorFanCount']) + + # 遍历 BilibiliVideo 实例列表，逐个写入表格 + for video_instance in video_instances: + ws.append([video_instance.bvId, video_instance.title, video_instance.url, video_instance.uploadTime, + video_instance.uploadTimeText, video_instance.topNo, video_instance.viewCount, video_instance.likeCount, + video_instance.coinCount, video_instance.favoriteCount, video_instance.commentCount, + video_instance.bulletCount, video_instance.creatorId, video_instance.creatorName, + video_instance.creatorFanCount]) + + # 保存 Excel 文件 + wb.save(excel_filename) +if __name__ == '__main__': + # 测试数据 + test_cases = [ + BilibiliVideo("BV1a4411C7i2", "视频标题1", "https://www.bilibili.com/video/BV1a4411C7i2", 1648927200, 1, 100000, 5000, 2000, 3000, 1000, 500, "up123456", "UP主A", 1000000), + BilibiliVideo("BV1F5411S8h3", "视频标题2", "https://www.bilibili.com/video/BV1F5411S8h3", 1648830800, 2, 80000, 4000, 1500, 2000, 800, 400, "up234567", "UP主B", 800000), + BilibiliVideo("BV1bW411d9c4", "视频标题3", "https://www.bilibili.com/video/BV1bW411d9c4", 1648734400, 3, 60000, 3000, 1000, 1500, 600, 300, "up345678", "UP主C", 600000), + BilibiliVideo("BV1qy411z5k5", "视频标题4", "https://www.bilibili.com/video/BV1qy411z5k5", 1648638000, 4, 40000, 2000, 800, 1000, 400, 200, "up456789", "UP主D", 400000), + BilibiliVideo("BV1R4411J0a6", "视频标题5", "https://www.bilibili.com/video/BV1R4411J0a6", 1648541600, 5, 20000, 1000, 500, 700, 200, 100, "up567890", "UP主E", 200000), + BilibiliVideo("BV1gW411H8d7", "视频标题6", "https://www.bilibili.com/video/BV1gW411H8d7", 1648445200, 6, 10000, 500, 300, 400, 100, 50, "up678901", "UP主F", 100000), + BilibiliVideo("BV1py411t6v8", "视频标题7", "https://www.bilibili.com/video/BV1py411t6v8", 1648348800, 7, 5000, 300, 200, 300, 50, 30, "up789012", "UP主G", 50000), + BilibiliVideo("BV1Qy411S3x9", "视频标题8", "https://www.bilibili.com/video/BV1Qy411S3x9", 1648252400, 8, 2000, 100, 100, 200, 20, 10, "up890123", "UP主H", 20000), + BilibiliVideo("BV1a4411C4y0", "视频标题9", "https://www.bilibili.com/video/BV1a4411C4y0", 1648156000, 9, 1000, 50, 50, 100, 10, 5, "up901234", "UP主I", 10000), + BilibiliVideo("BV1F5411C2r1", "视频标题10", "https://www.bilibili.com/video/BV1F5411C2r1", 1648059600, 10, 500, 30, 20, 50, 5, 3, "up012345", "UP主J", 5000) + ] + + # 将测试数据写入 Excel 文件 + write_to_excel(test_cases, 'bilibili_videos.xlsx') + + + + +import openpyxl + +def calculate_ratio_and_update(file_path, sheet_name): + # 打开 Excel 文件 + wb = openpyxl.load_workbook(file_path) + sheet = wb[sheet_name] + + # 遍历除第一行外的每一行数据 + for row_num in range(2, sheet.max_row + 1): + + value1 = sheet.cell(row=row_num, column=7).value #观看 + value2 = sheet.cell(row=row_num, column=8).value#点赞 + value3 = sheet.cell(row=row_num, column=9).value#投币 + value4 = sheet.cell(row=row_num, column=10).value#收藏 + value5 = sheet.cell(row=row_num, column=11).value#评论 + value6 = sheet.cell(row=row_num, column=15).value#粉丝 + value7 = sheet.cell(row=row_num, column=12).value # 弹幕 + + if value1 is not None and value2 is not None: + value1 = float(value1) + value2 = float(value2) + value3 = float(value3) + value4 = float(value4) + value5 = float(value5) + value6 = float(value6) + + ratio1 = value2 / value1 + sheet.cell(row=row_num, column=17).value = ratio1 # + ratio2 = value3 / value1 + sheet.cell(row=row_num, column=18).value = ratio2 + ratio3 = value4 / value1 + sheet.cell(row=row_num, column=19).value = ratio3 + ratio4 = value6 / value1 + sheet.cell(row=row_num, column=20).value = ratio4 + ratio5 = value5 + value7 + sheet.cell(row=row_num, column=21).value = ratio5 + + # 保存文件 + wb.save(file_path) +if __name__ == '__main__': + # 示例用法 + file_path = 'bilibili_videos.xlsx' # Excel 文件路径 + sheet_name = 'Sheet' # 工作表名称 + + calculate_ratio_and_update(file_path, sheet_name) + print("Data analysis written to the Excel file.") + + + + +import openpyxl +from openpyxl.chart import ScatterChart, BarChart, AreaChart, Reference + +# 读取Excel文件 +wb = openpyxl.load_workbook('bilibili_videos.xlsx') +ws = wb.active + +# 创建柱状图1 +bar_chart1 = BarChart() +bar_chart1.title = "观众对于视频的认可度" +bar_chart1.y_axis.title = "Data" +bar_chart1.x_axis.title = "Index" + +# 设置柱状图1数据 +bar_data1 = Reference(ws, min_col=17, min_row=2, max_row=ws.max_row) +bar_categories1 = Reference(ws, min_col=2, min_row=2, max_row=ws.max_row) +bar_chart1.add_data(bar_data1, titles_from_data=True) +bar_chart1.set_categories(bar_categories1) + +# 添加柱状图1到工作表 +ws.add_chart(bar_chart1, "V1") + +# 创建柱状图 +bar_chart = BarChart() +bar_chart.title = "收益" +bar_chart.y_axis.title = "Data" +bar_chart.x_axis.title = "Index" + +# 设置柱状图数据 +bar_data = Reference(ws, min_col=18, min_row=2, max_row=ws.max_row) +bar_categories = Reference(ws, min_col=2, min_row=2, max_row=ws.max_row) +bar_chart.add_data(bar_data, titles_from_data=True) +bar_chart.set_categories(bar_categories) + +# 添加柱状图到工作表 +ws.add_chart(bar_chart, "AD1") + +# 创建面积图 +area_chart = AreaChart() +area_chart.title = "视频实用性" +area_chart.y_axis.title = "Data" +area_chart.x_axis.title = "Index" + +# 设置面积图数据 +area_data = Reference(ws, min_col=19, min_row=2, max_row=ws.max_row) +area_categories = Reference(ws, min_col=2, min_row=2, max_row=ws.max_row) +area_chart.add_data(area_data, titles_from_data=True) +area_chart.set_categories(area_categories) + +# 添加面积图到工作表 +ws.add_chart(area_chart, "AL1") + +# 创建柱状图 +bar_chart_2 = BarChart() +bar_chart_2.title = "视频惊艳程度" +bar_chart_2.y_axis.title = "Data" +bar_chart_2.x_axis.title = "Index" + +# 设置柱状图数据 +bar_data_2 = Reference(ws, min_col=20, min_row=2, max_row=ws.max_row) +bar_categories_2 = Reference(ws, min_col=2, min_row=2, max_row=ws.max_row) +bar_chart_2.add_data(bar_data_2, titles_from_data=True) +bar_chart_2.set_categories(bar_categories_2) + +# 添加柱状图到工作表 +ws.add_chart(bar_chart_2, "AT1") + +# 创建柱状图 +bar_chart_3 = BarChart() +bar_chart_3.title = "视频的互动性" +bar_chart_3.y_axis.title = "Data" +bar_chart_3.x_axis.title = "Index" + +# 设置柱状图数据 +bar_data_3 = Reference(ws, min_col=21, min_row=2, max_row=ws.max_row) +bar_categories_3 = Reference(ws, min_col=2, min_row=2, max_row=ws.max_row) +bar_chart_3.add_data(bar_data_3, titles_from_data=True) +bar_chart_3.set_categories(bar_categories_3) + +# 添加柱状图到工作表 +ws.add_chart(bar_chart_3, "BB1") + +# 保存Excel文件 +wb.save('bilibili_videos_with_charts.xlsx') + +def write_english_texts(file_path, sheet_name, texts): + # 打开 Excel 文件 + wb = openpyxl.load_workbook(file_path) + sheet = wb[sheet_name] + + # 循环写入不同的英文文本到第一行的第17到21列 + for i, text in enumerate(texts, start=17): + sheet.cell(row=1, column=i).value = text + + # 保存文件 + wb.save(file_path) + + +file_path = 'bilibili_videos_with_charts.xlsx' # Excel 文件路径 +sheet_name = 'Sheet' # 工作表名称 + +english_texts = [ + "approve", + "money", + "practical", + "Stunning", + "interaction" +] # 要写入的英文文本列表 + +write_english_texts(file_path, sheet_name, english_texts) + +print("Data analysis chart written to the Excel file.") \ No newline at end of file