将函数更改为 return 整列

Question

函数 return 是单个州和人口，但我需要 return 列中的所有人口但跳过前 6 行

def findpop(state=None):
    f = open(getMediaPath("population_state_reduced (2).csv"), "rt")
    index = 1
    for line in f:
        if index > 3:
            parts = line.split(',')
            if state is None:
                return [(parts[4], int(parts[5]))]
            else:
                for line in f:
                    if parts[4] == state.capitalize():
                        return int(parts[5])
        index += 1

print findpop()

Answer 1

如果我理解你更正这正是你想要的：

import csv

def findpop(state=None):
    res = []
    with open(getMediaPath("population_state_reduced (2).csv")) as f:
        reader = csv.reader(f)
        for i, line in enumerate(reader):
            if i > 5: # skip first 6 rows
                if state is None:
                    res.append((line[4], int(line[5])))
                else:
                    if line[4] == state.capitalize():
                        return int(line[5])
    return res

将函数更改为 return 整列

change function to return entire column

python

jython