快捷導(dǎo)航

python如何生成指定文件大小

更新時(shí)間：2025年06月04日 14:06:24 作者：lewis_0

這篇文章主要介紹了python如何生成指定文件大小的實(shí)現(xiàn)方式,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教

python生成指定文件大小

方法一(速度最快)

def create_file_fast(file_path, size_mb):
    """快速生成指定大小的空文件（全0填充）"""
    size_bytes = size_mb * 1024 * 1024
    with open(file_path, 'wb') as f:
        f.seek(size_bytes - 1)  # 定位到文件末尾前一字節(jié)
        f.write(b'\0')          # 寫入單個(gè)空字節(jié)

# 示例：生成500MB文件
if __name__ == '__main__:
	create_file_fast('100mb_file.bin', 500)

方法二(中等速度)

import os
import random

def create_random_file(file_path, size_mb):
    """生成包含隨機(jī)內(nèi)容的文件"""
    size_bytes = size_mb * 1024 * 1024
    chunk_size = 1024 * 1024  # 每次寫入1MB
    
    with open(file_path, 'wb') as f:
        while size_bytes > 0:
            chunk = min(chunk_size, size_bytes)
            # 生成隨機(jī)字節(jié)數(shù)據(jù)（比os.urandom更快）
            data = bytes(random.getrandbits(8) for _ in range(chunk))
            f.write(data)
            size_bytes -= chunk

# 示例：生成500MB文件
if __name__ == '__main__:
	create_random_file('random_500mb.bin', 500)

方法三(生成可讀文本文件–較慢)

import string
import random

def create_text_file(file_path, size_mb, line_length=80):
    """生成包含隨機(jī)文本的文件（帶換行符）"""
    size_bytes = size_mb * 1024 * 1024
    chars = string.ascii_letters + string.digits + string.punctuation + ' '
    
    with open(file_path, 'w') as f:
        while size_bytes > 0:
            # 生成一行文本
            line = ''.join(random.choices(chars, k=line_length))
            line += '\n'  # 添加換行符
            
            f.write(line)
            size_bytes -= len(line.encode('utf-8'))  # 計(jì)算UTF-8編碼后的字節(jié)大小

# 示例：生成500MB文本文件
if __name__ == '__main__:
	create_text_file('10mb_text.txt', 500)

方法四(使用內(nèi)存映射高效生成–超大文件)

import mmap
import os

def create_large_file(file_path, size_gb):
    """高效生成超大文件（GB級(jí)別）"""
    size_bytes = size_gb * 1024 ** 3
    
    # 創(chuàng)建空文件并設(shè)置大小
    with open(file_path, 'wb') as f:
        f.truncate(size_bytes)
    
    # 使用內(nèi)存映射填充隨機(jī)數(shù)據(jù)
    with open(file_path, 'r+b') as f:
        mm = mmap.mmap(f.fileno(), 0)
        chunk_size = 1024 * 1024 * 100  # 每次處理100MB
        for offset in range(0, size_bytes, chunk_size):
            end = min(offset + chunk_size, size_bytes)
            mm[offset:end] = os.urandom(end - offset)
        mm.close()

# 示例：生成5GB文件
if __name__ == '__main__:
	create_large_file('5gb_file.dat', 5)

方法五(生成特定模式文件)

def create_pattern_file(file_path, size_mb, pattern=b'ABCD'):
    """生成重復(fù)模式的文件（用于測試）"""
    size_bytes = size_mb * 1024 * 1024
    pattern_len = len(pattern)
    
    with open(file_path, 'wb') as f:
        # 計(jì)算完整模式塊數(shù)量
        full_blocks = size_bytes // pattern_len
        # 剩余字節(jié)
        remainder = size_bytes % pattern_len
        
        # 寫入完整塊
        for _ in range(full_blocks):
            f.write(pattern)
        
        # 寫入剩余部分
        if remainder:
            f.write(pattern[:remainder])

# 示例：生成500MB的重復(fù)模式文件
if __name__ == '__main__:
	create_pattern_file('pattern_50mb.dat', 500, b'TESTPATTERN')

方法六

def run(file_size):
    #輸入的文件大小去除首尾空格后切割，如果列表中只有一個(gè)數(shù)字說明是整數(shù)，否則就是小數(shù)
    file_path = "D:/dir/"
    file_size = file_size
    print('file_size', file_size)
    new_file_size = file_size.strip()
    file_size_list = new_file_size.split(".")
    start = time.time()
    if len(file_size_list) == 1:
        int_size_mb(file_size)
        print("文件大小{}MB,已存入地址{}".format(file_size, file_path))
    else:
        int_size_mb(file_size)
        float_size_mb(file_size)
        print("文件大小{}MB,已存入地址{}".format(file_size,file_path))
    end = time.time()
    timer = end - start
    speed = float(file_size) / timer
    # print(start, end, timer, speed)
    logger.info('寫入文件共耗時(shí):' + str(timer) + ',寫入速度為:' + str(speed))
def int_size_mb(file_size):
    file_path = "D:/dir/"
    file_name = 'test_file'
    new_file_size = file_size.strip()
    file_size_list = new_file_size.split(".")
    #整數(shù)部分用寫入文件w方式
    with open(file_path+file_name,"w") as file:
        #b-kb-mb文件大小轉(zhuǎn)化
        for i in range(int(file_size_list[0])):
            for j in range(1024):
                file.write("01"*512)
def float_size_mb(file_size):
    file_path = "D:/dir/"
    file_name = 'test_file'
    new_file_size = file_size.strip()
    file_size_list = new_file_size.split(".")
    #小數(shù)部分用追加寫入a方法
    file_size = file_size
    with open(file_path+file_name,"a") as file:
        #獲取小數(shù)（單位mb）
        float_size_mb=float(file_size)-int(file_size_list[0])
        for i in range(1024):
            file.write("1"*int(1024*float_size_mb))

# 示例：生成500MB的文件           
if __name__ == '__main__':
	run(500)