尝试将y变量读取到pyplot.scatter时发生IndexError



我试图创建的最终产品是一个程序,它从csv(附加,称为OutputC.csv(中获取输入,并允许用户从他们选择的列中绘制散点图。我决定先创建预选列的图形,然后再使用rawinput和变量。因此,我遇到的问题是python将数据读取到pyplot中的方式。数据取自输入csv,并写入它自己的文件(Qout14.csv(,我试图从中绘制。不幸的是,即使x和y变量列表的长度相同,我也会收到IndexError。在阅读后,我想知道问题是否是y数据没有逗号分隔实例中的数据,所以它可能只是将所有y作为字符串读取,所以它在列表中没有其他内容。令人困惑的是,同一行代码适用于变量的另一端,例如x。有人能指出我的错误在哪里吗?

Example of Inputs(Columns 1 and 7 of OutputC.csv):
pcp Qout14
2.3 7.20E-03
3   1.34E-02
3.3 1.50E-02
2.3 8.25E-03
3   1.32E-02
2.5 9.47E-03
3   1.28E-02
3.6 1.81E-02
2.5 1.02E-02
2.5 9.44E-03
2   6.00E-03
2.8 1.17E-02
2.8 1.12E-02
2.8 1.16E-02

Traceback(上次调用(:文件"C:\Users\jmiyama\Desktop\Python Files\OutputCtoScatterplot.py",第行48,英寸main((文件"C:\Users\jmiyama\Desktop\Python Files\OutputCtoScatterplot.py",第35行,在mainy=[row.split(','([1]用于数据中的行[1]IndexError:列出索引超出范围

import csv
import numpy as np
import matplotlib.pyplot as plt
import pandas
import collections
def getColumn(filename, column):
results = csv.reader(open(filename), dialect='excel')   
return [result[column] for result in results]
def dictcolumn():
pcp = getColumn("C:\Users\jmiyama\Desktop\Python Files\OutputC.csv",1)
Qout14 = getColumn("C:\Users\jmiyama\Desktop\Python Files\OutputC.csv",7)
scatter={}
scatter["pcp"]= pcp
scatter["Qout14"]= Qout14
return scatter
def main():
scatter = dictcolumn()
file = 'Qout14.csv'
with open(file, 'w+') as csv_file:
writer = csv.writer(csv_file, lineterminator='n')
writer.writerows(zip(*scatter.values()))
with open("Qout14.csv") as f:
data = f.read()
data = data.split('n')
# trying to split the zipped values into columns
x = [row.split(',')[0] for row in data[1:]]
#THE CODE WORKS UP TO HERE
y = [row.split(',')[1] for row in data[1:]]
fig = plt.figure()
ax = fig.add_subplot(111)
ax.grid()
plt.scatter(x,y)
plt.xlabel('pcp')
plt.ylabel('Qout14')
plt.title('Qout14 vs pcp')
plt.show()
if __name__ == '__main__':
main()

data = data.split('n')似乎在末尾添加了额外的"空白"行,因此下面的行可以工作。

x = [row.split(',')[0] for row in data[1:-1]]
y = [row.split(',')[1] for row in data[1:-1]]

最新更新