制作沒有標題的CVS文件列表

Question

該函數查找參數為空時給出特定狀態或所有狀態的填充狀態的所有狀態。 當我將參數留空時，盡管它給我一個錯誤，因為文件的前3行中有標頭。

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

只需跳過前三行：

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