1
我正在带一个标题行的csv文件(称为aggregate_file
),按列进行排序,并将其重新写入另一个csv文件(称为sorted_file
)。我想要排序的列标题由variable_names
指定。为什么我的csv排序功能不起作用?
def sortbyCounty(aggregate_file, sorted_file, *variable_names):
f = open(aggregate_file, 'r')
readit = csv.reader(f)
headers = readit.next()
col_indices = []
for var in variable_names:
col_indices.append(headers.index(var))
print col_indices
thedata = list(readit)
thedata.sort(key=operator.itemgetter(col_indices))
fx = open(sorted_file, 'w')
writeit = csv.writer(fx)
writeit.writerow(headers)
writeit.writerows(thedata)
writeit.close()
return sorted_file
接下来,我调用这个函数下面几行:
aggregate_file = "Aggregate_test90.csv"
sorted_file = "County_test90.csv"
variable_names = 'CTYCODE90'
test = sortbyCounty(aggregate_file, sorted_file, *variable_names)
这是我的错误信息:
col_indices.append(headers.index(var))
ValueError: list.index(x): x not in list
然而,当我print
我headers
名单,我可以清楚地看到我的变量是存在的:
['_STATE90', 'HEIGHT90', 'WEIGHT90', '_BMI90', 'AGE90', 'CTYCODE90', 'IYEAR90', 'SEX90', '_RFOBESE90']
所以我不明白为什么我收到这个错误信息。我错过了什么?