TypeError：列表索引必须是整数或切片，而不是元组

Question

fvecs = []

for line in open(filename):
    stats = line.split(',')
    labels.append(int(stats[0]))
    fvecs.append([float(x) for x in stats[5,6,12,27,29,37,39,41]])

I have a big csv. file that I am using as a dataset containing 43 columns and hundreds of rows, I am attempting to extract specific columns to be used as individual records and I can't seem to work this out. The error is caused by the final line of code and produces the error message in the title, it works perfectly when the range is set to, stats[30:38] for example.

I have tried storing the required columns in a separate array and calling it like stats[requiredcolumns] but this produces the same error.

I have considered using pandas but this is just a small snippet of code from a much larger program, which all functions correctly, and the implementation of pandas would require a complete overhaul of the full program which is not possible due to time constraints.

Any help would be greatly appreciated

Answer 1

If you have few columns, you can try this:

for line in open(filename):
    stats = line.split(',')
    labels.append(int(stats[0]))
    fvecs.append([float(x) for x in stats[5],stats[6],stats[12],stats[27], stats[29], stats[37], stats[39], stats[41]])

This code will return a list of lists; otherwise, the first comment is right about indexing and NumpPy.