使用来自 csv 文件的 matplotlib 在 python 中绘制时间序列图

Question

我有以下格式的一些 csv 数据。

Ln    Dr    Tag Lab    0:01    0:02    0:03    0:04    0:05    0:06    0:07    0:08   0:09
L0   St     vT  4R       0       0       0       0       0      0        0       0      0
L2   Tx     st  4R       8       8       8       8       8      8        8       8      8
L2   Tx     ss  4R       1       1       9       6       1      0        0       6      7

我想使用列（ Ln 、 Dr 、 Tg 、 Lab ）作为键和0:0n字段作为时间序列图上的值绘制时间序列图。

我有以下代码。

#!/usr/bin/env python

import matplotlib.pyplot as plt
import datetime
import numpy as np
import csv
import sys


with open("test.csv", 'r', newline='') as fin:
    reader = csv.DictReader(fin)
    for row in reader:
          key = (row['Ln'], row['Dr'], row['Tg'],row['Lab'])
          #code to extract the values and plot a timeseries.

如何提取0:0n列中的所有值而不单独指定它们中的每一个。 我希望所有时间序列都绘制在一个时间序列上？

Answer 1

我建议使用熊猫：

import pandas as pd
a=pd.read_csv('yourfile.txt',delim_whitespace=True)
for x in a.iterrows():
    x[1][4:].plot(label=str(x[1][0])+str(x[1][1])+str(x[1][2])+str(x[1][3]))

plt.ylim(-1,10)
plt.legend()

Answer 2

我不确定你到底想做什么，但np.loadtxt是去这里的方法。 确保为您的文件正确设置分隔符

data = np.loadtxt(fname="test.csv",delimiter=',',skiprows=1)

现在第 n 列data是文件的第 n 列，行也相同。

您可以按行访问数据： data[n]或按列访问数据： data[:,n]