在Python中转换文件大小的更好方法

Question

我正在使用一个读取文件的库,并以字节为单位返回其大小.

然后将此文件大小显示给最终用户; 为了让它们更容易理解它,我明确地将文件大小转换为MB除以1024.0 * 1024.0.当然这有效,但我想知道在Python中有更好的方法吗？

更好的是,我的意思是stdlib函数可以根据我想要的类型操作大小.就像我指定的那样MB,它会自动将其除以1024.0 * 1024.0.有些人在这些方面.

Answer 1

这是我使用的:

import math

def convert_size(size_bytes):
   if size_bytes == 0:
       return "0B"
   size_name = ("B", "KB", "MB", "GB", "TB", "PB", "EB", "ZB", "YB")
   i = int(math.floor(math.log(size_bytes, 1024)))
   p = math.pow(1024, i)
   s = round(size_bytes / p, 2)
   return "%s %s" % (s, size_name[i])

注意:大小应以字节发送.

Answer 2

有一个hurry.filesize,它将以字节为单位大小,如果有的话,可以创建一个很好的字符串.

>>> from hurry.filesize import size
>>> size(11000)
'10K'
>>> size(198283722)
'189M'

或者如果你想要1K == 1000(这是大多数用户所假设的):

>>> from hurry.filesize import size, si
>>> size(11000, system=si)
'11K'
>>> size(198283722, system=si)
'198M'

它也有IEC支持(但没有记录):

>>> from hurry.filesize import size, iec
>>> size(11000, system=iec)
'10Ki'
>>> size(198283722, system=iec)
'189Mi'

因为它是由Awesome Martijn Faassen编写的,所以代码很小,清晰且可扩展.编写自己的系统很容易.

这是一个:

mysystem = [
    (1024 ** 5, ' Megamanys'),
    (1024 ** 4, ' Lotses'),
    (1024 ** 3, ' Tons'), 
    (1024 ** 2, ' Heaps'), 
    (1024 ** 1, ' Bunches'),
    (1024 ** 0, ' Thingies'),
    ]

像这样使用:

>>> from hurry.filesize import size
>>> size(11000, system=mysystem)
'10 Bunches'
>>> size(198283722, system=mysystem)
'189 Heaps'

Answer 3

而不是1024 * 1024你的大小除数可以使用<< 按位移位运算符,即1<<20获得兆字节,1<<30以获得千兆字节等.

我定义了一个常量MBFACTOR = float(1<<20),然后可以用于字节,即:megas = size_in_bytes/MBFACTOR.

Answer 4

这是计算尺寸的紧凑功能

def GetHumanReadable(size,precision=2):
    suffixes=['B','KB','MB','GB','TB']
    suffixIndex = 0
    while size > 1024 and suffixIndex < 4:
        suffixIndex += 1 #increment the index of the suffix
        size = size/1024.0 #apply the division
    return "%.*f%s"%(precision,size,suffixes[suffixIndex])

有关更详细的输出和反之亦然操作,请参阅:http://code.activestate.com/recipes/578019-bytes-to-human-human-to-bytes-converter/

Answer 5

这里是：

def convert_bytes(size):
    for x in ['bytes', 'KB', 'MB', 'GB', 'TB']:
        if size < 1024.0:
            return "%3.1f %s" % (size, x)
        size /= 1024.0

    return size

输出

>>> convert_bytes(1024)
'1.0 KB'
>>> convert_bytes(102400)
'100.0 KB'

Answer 6

万一有人在寻找这个问题的反面(我确实这样做),这对我有用:

def get_bytes(size, suffix):
    size = int(float(size))
    suffix = suffix.lower()

    if suffix == 'kb' or suffix == 'kib':
        return size << 10
    elif suffix == 'mb' or suffix == 'mib':
        return size << 20
    elif suffix == 'gb' or suffix == 'gib':
        return size << 30

    return False

Answer 7

如果您已经知道所需的内容，请参见下文，以一种快速且相对易于阅读的方式在一行代码中打印文件大小。这些单行代码将上面@ccpizza的出色答案与一些方便的格式化技巧结合在一起，我在这里阅读了如何以千位分隔符的形式用逗号打印数字？。

字节数

print ('{:,.0f}'.format(os.path.getsize(filepath))+" B")

长袍

print ('{:,.0f}'.format(os.path.getsize(filepath)/float(1<<7))+" kb")

千字节

print ('{:,.0f}'.format(os.path.getsize(filepath)/float(1<<10))+" KB")

兆位

print ('{:,.0f}'.format(os.path.getsize(filepath)/float(1<<17))+" mb")

兆字节

print ('{:,.0f}'.format(os.path.getsize(filepath)/float(1<<20))+" MB")

千兆位

print ('{:,.0f}'.format(os.path.getsize(filepath)/float(1<<27))+" gb")

千兆字节

print ('{:,.0f}'.format(os.path.getsize(filepath)/float(1<<30))+" GB")

兆兆字节

print ('{:,.0f}'.format(os.path.getsize(filepath)/float(1<<40))+" TB")

显然，他们假设您一开始就大致知道要处理的大小，就我而言（西南伦敦电视台的视频编辑器），大小为MB，有时对于视频剪辑则为GB。

使用PATHLIB更新 为了回应Hildy的评论，以下是我的建议：仅使用Python标准库提供一对紧凑的函数（保持事物为“原子的”而不是合并它们）：

from pathlib import Path    

def get_size(path = Path('.')):
    """ Gets file size, or total directory size """
    if path.is_file():
        size = path.stat().st_size
    elif path.is_dir():
        size = sum(file.stat().st_size for file in path.glob('*.*'))
    return size

def format_size(path, unit="MB"):
    """ Converts integers to common size units used in computing """
    bit_shift = {"B": 0,
            "kb": 7,
            "KB": 10,
            "mb": 17,
            "MB": 20,
            "gb": 27,
            "GB": 30,
            "TB": 40,}
    return "{:,.0f}".format(get_size(path) / float(1 << bit_shift[unit])) + " " + unit

# Tests and test results
>>> get_size("d:\\media\\bags of fun.avi")
'38 MB'
>>> get_size("d:\\media\\bags of fun.avi","KB")
'38,763 KB'
>>> get_size("d:\\media\\bags of fun.avi","kb")
'310,104 kb'

Answer 8

UNITS = {1000: ['KB', 'MB', 'GB'],
            1024: ['KiB', 'MiB', 'GiB']}

def approximate_size(size, flag_1024_or_1000=True):
    mult = 1024 if flag_1024_or_1000 else 1000
    for unit in UNITS[mult]:
        size = size / mult
        if size < mult:
            return '{0:.3f} {1}'.format(size, unit)

approximate_size(2123, False)