在这种情况下如何拆分csv文件[关闭]

Question

    _filename = f'summary_download_{_type}_{_from}_to_{_to}_{_encode}{_account}.zip'
    attachment = 'attachment; filename*=utf-8\'\'' + urllib.parse.quote(_filename)

    if _type == 'query':
        header = ['date', 'user_name', 'sdgid', 'number_of_query', 'number_of_device']
    if _type == 'threat':
        header = ['date', 'user_name', 'sdgid', 'threat', 'number_of_query']
    if _type == 'category':
        header = ['date', 'user_name', 'sdgid', 'category', 'number_of_query']
    if _type == 'specified_domain':
        header = ['date', 'user_name', 'sdgid', 'specified_domain_list', 'number_of_query']

    auditlog_data_frame = pd.DataFrame(data, columns=header)
    auditlog_data_csv = auditlog_data_frame.to_csv(index=False, encoding='utf8')
    auditlog_data_csv.split(';')
    with TemporaryDirectory() as tmpdirname:
        temp_zip_file_name = os.path.join(tmpdirname, os.path.basename(_filename))
        zip_file = pyzipper.AESZipFile(
            temp_zip_file_name,
            'w',
            compression=pyzipper.ZIP_DEFLATED,
            encryption=pyzipper.WZ_AES
        )
        zip_file.setpassword(settings.ZIP_FILE_PASSWORD)
        zip_file.writestr('summarydownload.csv', auditlog_data_csv)
        zip_file.close()

        response = HttpResponse(
            open(temp_zip_file_name, 'rb'),
            content_type='application/zip'
        )
    response['Content-Disposition'] = attachment
    return response

Answer 1

如果我正确理解了您的问题，那么您正在寻找类似的东西

with open(filename) as input_file:
    output_file = None
    for index, line in enumerate(input_file):
        if index % 1_000_000 == 0:
            if output_file:
                output_file.close()
            output_file = open(f'output{index}', "w")
        output_file.write(line)
    output_file.close()