绘制折线图&遇到问题记录

刘义申汉

已于 2024-12-12 20:58:49 修改

阅读量350

点赞数 2

CC 4.0 BY-SA版权

分类专栏： # 数据可视化文章标签： python 开发语言数据可视化 matplotlib

于 2024-12-12 20:55:21 首次发布

本文链接：https://blog.csdn.net/weixin_44498989/article/details/144435948

绘制折线图

主要参考：https://blog.csdn.net/qq_38029916/article/details/121611066

对应代码

import csv
import matplotlib.pyplot as plt
import pandas as pd
import numpy as np
plt.rcParams['font.sans-serif'] = ['SimHei']
plt.rcParams['font.family'] = 'sans-serif'
plt.rcParams['axes.unicode_minus'] = False

pd_data = pd.read_csv("results.csv")
# %%capture output
## 捕获横坐标
epoch_data = pd_data["epoch"].astype(int).values
type(pd_data["epoch"])

%%capture output
pd_data["train/box_loss"][np.isinf(pd_data["train/box_loss"])] = np.nan
train_cls_loss_data = pd_data["train/box_loss"].fillna(method="backfill",axis=0,inplace=False).astype(float).values

def ge