Python 在文件结束之前退出 CSV 文件

Question

我有一个函数，它接受在我的 django 应用程序中上传的 csv 或 txt 文件，读取所有文件并将其处理到 python 列表中。我注意到，如果由于某种原因文件包含空行，它不会读取到文件末尾并返回一个空列表。你认为我错过了什么吗？这是我的代码。

是的，我可以告诉用户确保文件不包含任何空行，但我想我可以让脚本足够智能，在阅读时直到文件末尾才中断。

django #form #class #view

def read_the_file_content(self,文件,文件类型,消息,发件人): 打印“读文件内部”

    data = []


    for row in csv.reader(file):

        print row
        num = str(re.sub("[^0-9]","",row[0]))
        #print num
        if num.startswith('0'):
           phone_num =  '234' + num[1:]

        else:
            phone_num =  num



        data.append(phone_num)     


    if data == []:

        return []
    else:
        return set(data)

csvFILEData

0703706261107063724920

0806553737708168571687

Answer 1

您可以通过检查返回列表来跳过 csv 阅读器中的空行

for row in csv.reader(file):
    if not row:
       continue
    ...