python3 绘图

YMWM_

已于 2024-10-29 11:30:51 修改

阅读量628

点赞数

CC 4.0 BY-SA版权

分类专栏： Python学习文章标签： python3 绘图

于 2023-02-19 10:18:40 首次发布

本文链接：https://blog.csdn.net/YMWM_/article/details/129107996

Python学习专栏收录该内容

71 篇文章

订阅专栏

本文展示了如何使用Python的数据可视化库matplotlib，结合numpy和pandas，来绘制曲线图。具体步骤包括设置字体显示，创建图形，定义曲线颜色和标记，添加图例，以及设置坐标轴标签和网格。示例代码中，绘制了一条从-1到5的二次曲线图。

1 绘制二维曲线图

import matplotlib.pyplot as plt  # 为方便简介为plt
import numpy as np  # 画图过程中会使用numpy
import pandas as pd  # 画图过程中会使用pandas

def PythonPlot(x: list, y: list):
    #绘制曲线图
    plt.rcParams['font.family'] = ['SimHei'] #确保图中中文字体正确显示
    plt.rcParams['axes.unicode_minus'] = False #确保负号正确显示
    fig = plt.figure() #先创建一个窗口
    plt.plot(x, y, color = 'blue', marker = 'o') #plot()画出曲线，并设置曲线颜色，线标记
    plt.legend(labels = ['二次曲线图'], loc = 'best') #best表示自动分配最佳位置
    plt.xlabel('x轴')
    plt.ylabel('y轴')
    plt.title('曲线图')
    plt.grid()
    plt.show() #显示图像    

x = np.arange(-1,5,0.01)
y = [a * a - 4 * a + 3 for a in x]
PythonPlot(x, y)

结果为，
在这里插入图片描述

2 绘制三维曲线图

import os 
import sys 
import time 
import matplotlib.pyplot as plt

if __name__ == "__main__":
    start_t = time.time()
        
    input_txt_path = "KeyFrameTrajectory_kitti_00.txt"
    x_coordinates = []
    y_coordinates = []
    z_coordinates = []
    with open(input_txt_path, "r", encoding = "utf-8") as fin:
        lines = fin.readlines()
        for line in lines:
            line = line.replace("\n", "")
            ele = line.split()
            x = float(ele[0])
            y = float(ele[1])
            z = float(ele[2])
            x_coordinates.append(x)
            y_coordinates.append(y)
            z_coordinates.append(z)

    #plot trajectory
    fig = plt.figure()
    ax = fig.add_subplot(111, projection='3d')
    ax.plot(x_coordinates, y_coordinates, z_coordinates, label='3D Curve')
    ax.set_xlabel('X')
    ax.set_ylabel('Y')
    ax.set_zlabel('Z')
    ax.legend()
    plt.savefig("traj.png")
    plt.show()

    end_t = time.time()
    print(f"used time: {(end_t - start_t) / 60.0} minutes!")