如何在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 数据框的数据