用python绘制CSV数据：列表索引超出范围

Question

我正在尝试从一些数据中绘制图表，但即使数据有 2 列，我也会遇到索引超出范围错误。代码如下：

import csv
from matplotlib import pyplot as plt


def getColumn(filename, column):
    results = csv.reader(open(filename), delimiter='\t')
    return [result[column] for result in results]

wavelength = getColumn('Bb69.dat.fix',0)
flux = getColumn('Bb69.dat.fix',1)

plt.figure('Bb69')
plt.xlabel('Wavelength (angstrom)')
plt.ylabel('flux (erg/cm^2/s/angstrom)')
plt.plot(wavelength,flux)
plt.show()

这是错误

Traceback (most recent call last):

  File "SP12.py", line 14, in <module>

    flux = getColumn('Bb69.dat.fix',1)

  File "SP12.py", line 11, in getColumn

    return [result[column] for result in results]

IndexError: list index out of range

Answer 1

def getColumn(filename, column):
    results = csv.reader(open(filename), delimiter='\t')
    return [result[column] for result in results]

有可能不是results 的所有元素总是包含等待的2 个元素。例如 csv 文件包含空行。

将getColumn方法的最后一行改为：

    return [result[column] for result in results if len(result) > column]

然而这是一个风险，因为如果某些行不包含足够的字段，它可能会返回更长的列表以用于较低的 column。

更好的是：

wavelength = []
flux = []

with open('Bb69.dat.fix') as f:
    for row in csv.reader(f, delimiter='\t'):
        if len(row) >= 2:
            wavelength.append(float(row[0]))
            flux.append(float(row[1]))

这保证wavelength 和flux 的长度相同。