python 读取csv文件绘图

最新推荐文章于 2024-09-19 21:52:54 发布

哈利马其

最新推荐文章于 2024-09-19 21:52:54 发布

阅读量6.2k

点赞数 1

分类专栏： linux 文章标签： python

本文链接：https://blog.csdn.net/lybhit/article/details/124086378

版权

linux 专栏收录该内容

4 篇文章 0 订阅

订阅专栏

该博客展示了如何使用Python从CSV文件中读取数据，并利用matplotlib库绘制3D角度变化图。代码首先定义了列类型，然后打开CSV文件，逐行读取并提取pos_x, pos_y, pos_z, roll, pitch, yaw的数据。最后，使用plot函数绘制了pos_x, pos_y, pos_z的轨迹图。

摘要由CSDN通过智能技术生成

python 读取csv文件数据，然后通过plot绘图。

#!/bin/bash python
 
 import csv
 import numpy as np
 from matplotlib import pyplot as plt
 
 col_types = [float, float, float, float, float, float, float]
 
 with open('/home/syy001/Desktop/a.csv') as f:
     f_csv = csv.reader(f)
 
     headers = next(f_csv)
 
     pos_x,pos_y,pos_z,roll,pitch,yaw = [],[],[],[],[],[]
 
     index = 0
 
     for row in f_csv:
         row = tuple(convert(value) for convert, value in zip(col_types, row))
         pos_x.append(row[0])
         pos_y.append(row[1])
         pos_z.append(row[2])
         roll.append(row[3])
         pitch.append(row[4])
         yaw.append(row[5])
         index += 1
 
     x_len = index * 0.1
     t = np.arange(0, x_len, 0.1)
 
 #    fig=plt.figure(dpi=128, figsize=(10, 6))
 
     plt.plot(t, pos_x, color="green", label="angular_x")
     plt.plot(t, pos_y, color="red", label="angular_y")
     plt.plot(t, pos_z, color="blue", label="angular_z")
 #    plt.plot(t, roll, color="yellow", label="acc_x")
 #    plt.plot(t, pitch, color="skyblue", label="acc_y")
 #    plt.plot(t, yaw, color="black", label="acc_z")
     plt.show()