python - 如何在Python中存储csv中的行值？

Question

我有一个用 C++ 构建的主程序，可以生成 CSV 文件。我已经使用 PyQt5 制作了一个界面，我想从该 CSV 中读取一些特定的列。

我可以阅读，但我想存储它们以使用 matplotlib 绘制绘图。可能我的错是我试图像处理数组一样处理该行。这是我的 readfile 函数的样子:

def readCSV(self):
    try:
        with open('resultado.csv') as csvFile:
            reader = csv.DictReader(csvFile, delimiter=';')
            i = 0
            x, y = []
            for row in reader:
                print(row["TOA (ns)"], row["Frecuencia Inicial (MHz)"])
                x[i] = row["TOA (ns)"]
                y[i] = row["Frecuencia Inicial (MHz)"]
                i = i+1
    except:
        print("Error: ", sys.exc_info()[0])

异常打印:错误:类“ValueError”

这是我的 csv 的前 2 行，下一行都是浮点值。我只想要 TOA 和 Frecuencia Inicial 列:

Answer 1

对代码进行一些更改:只需将值附加到 x 和 y 数组，使用精确的列名

 def readCSV():
     with open('resultado.csv') as csvFile:
        reader = csv.DictReader(csvFile, delimiter=';')
        x, y = [], [] 
        for row in reader:
            try:
                print(row["TOA (ns)"], row["Frecuencia Inicial"])
                # guess you want
                x.append(row["TOA (ns)"])
                y.append(row["Frecuencia Inicial"])
                #i = i+1
            except Exception as e:
                print("Error: ", e)

     # do something with x, y
     # since you use 'self' in the function def I asume this is
     # a class method, so you could make x and y class properties
     # and then use self.x and self.y in this code

     print(x)
     print(y)

您可能想检查 Pandas 的使用:

import pandas as pd
df = pd.read_csv("/tmp/indata.csv", delimiter=";")
df
   TOA (ns)  Frecuencia Inicial  
0        10                2000   
1        20                3000

从 jupyter 笔记本中，您可以使用以下方法绘制图形:

df[['TOA (ns)', 'Frecuencia Inicial']].plot(figsize=(20,10))

您仍然可以使用 matplotlib 微调绘图并使用 Pandas 数据帧中的数据