生成 CVS 文件列表 w/o header

Question

此函数查找给出特定州或所有州的人口参数留空时的所有州。当我将参数留空时，虽然它给我一个错误，因为文件的前 3 行中有 headers。

def findpop(state=None):
     f=open(getMediaPath("population_state_reduced (2).csv"),"rt")
     for line in f: 
       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])
print findpop()

Answer 1

跳过前 3 行：

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()

生成 CVS 文件列表 w/o header

make list of CVS file w/o header

python

jython