从python中的列表中绘制数据

Question

我需要绘制一些物体（汽车）的速度。

每个速度都通过例程计算并写入文件，大致通过这个（我删除了一些行以简化）：

thefile_v= open('vels.txt','w') 

for car in cars:
    velocities.append(new_velocity) 

     if len(car.velocities) > 4:
          try:
              thefile_v.write("%s\n" %car.velocities) #write vels once we get 5 values
              thefile_v.close

          except:
              print "Unexpected error:", sys.exc_info()[0]
              raise

结果是一个文本文件，其中包含每辆车的速度列表。

类似这样的：

[0.0, 3.8, 4.5, 4.3, 2.1, 2.2, 0.0]
[0.0, 2.8, 4.0, 4.2, 2.2, 2.1, 0.0]
[0.0, 1.8, 4.2, 4.1, 2.3, 2.2, 0.0]
[0.0, 3.8, 4.4, 4.2, 2.4, 2.4, 0.0]

然后我想绘制每个速度

with open('vels.txt') as f:
    lst = [line.rstrip() for line in f]

plt.plot(lst[1]) #lets plot the second line
plt.show()

这是我发现的。这些值被视为一个字符串并将它们作为 yLabel。

我解决了这个问题：

from numpy import array

y = np.fromstring( str(lst[1])[1:-1], dtype=np.float, sep=',' )
plt.plot(y)
plt.show()

我了解到，我之前构建的一组速度列表被视为数据行。

我必须将它们转换为数组才能绘制它们。然而，括号 [] 进入了方式。通过将数据行转换为字符串并通过此删除括号（即 [1:-1]）。

它现在正在工作，但我确信有更好的方法来做到这一点。

有没有cmets？

Answer 1

只要说您有数组 [0.0, 3.8, 4.5, 4.3, 2.1, 2.2, 0.0]，要绘制此图，代码将类似于：

import matplotlib.pyplot as plt

ys = [0.0, 3.8, 4.5, 4.3, 2.1, 2.2, 0.0]
xs = [x for x in range(len(ys))]

plt.plot(xs, ys)

# Make sure to close the plt object once done
plt.close()

如果您想为 x 轴设置不同的间隔，那么：

interval_size = 2.4 #example interval size
xs = [x * interval_size for x in range(len(ys))]

另外在从文本文件中读取您的值时，请确保您已将您的值从字符串转换回整数。这可能是您的代码假设您的输入是 y 标签的原因。

Answer 2

只是一种可能的简单解决方案。使用map 函数。假设在您的文件中，您存储的数据类似于，没有任何 [ 和 ] 不可转换字母。

#file_name: test_example.txt
0.0, 3.8, 4.5, 4.3, 2.1, 2.2, 0.0
0.0, 2.8, 4.0, 4.2, 2.2, 2.1, 0.0
0.0, 1.8, 4.2, 4.1, 2.3, 2.2, 0.0
0.0, 3.8, 4.4, 4.2, 2.4, 2.4, 0.0

那么下一步就是；

import matplotlib.pyplot as plt

path = r'VAR_DIRECTORY/test_example.txt' #the full path of the file


with open(path,'rt') as f:
    ltmp = [list(map(float,line.split(','))) for line in f]

plt.plot(ltmp[1],'r-')
plt.show()

在顶部，我只是假设您要绘制第二行 0.0, 2.8, 4.0, 4.2, 2.2, 2.1, 0.0。然后这是结果。

Answer 3

这个例子并不完整，所以这里必须做一些假设。一般来说，使用 numpy 或 pandas 来存储您的数据。

假设car 是一个对象，具有velocity 属性，您可以将所有速度写在一个列表中，用numpy 将此列表保存为文本文件，用numpy 再次读取并绘制它。

import numpy as np
import matplotlib.pyplot as plt

class Car():
    def __init__(self):
        self.velocity = np.random.rand(5)

cars = [Car() for _ in range(5)]
velocities = [car.velocity for car in cars]
np.savetxt("vels.txt", np.array(velocities))

#### 

vels = np.loadtxt("vels.txt")
plt.plot(vels.T)
## or plot only the first velocity
#plt.plot(vels[0]

plt.show()