Python新建一个image并写入tar包|极客笔记

Python新建一个image并写入tar包

在实际开发中，有时候我们需要将一个文件或者文件夹打包成一个tar包，并进行传输或者存储。本文将介绍如何使用Python新建一个image，并将其写入到tar包中。

准备工作

在开始之前，我们需要先安装Python的tarfile模块，该模块用于创建和解压tar包。可以使用以下命令安装：

pip install tarfile

创建一个image

首先，我们需要创建一个image，用于存储到tar包中。以下是一个简单的示例，创建一个名为test.jpg的image：

import numpy as np
from PIL import Image

# 创建一个随机的100x100的RGB数组
data = np.random.randint(0, 255, (100, 100, 3), dtype=np.uint8)
image = Image.fromarray(data)

# 保存image为test.jpg
image.save('test.jpg')

运行以上代码后，当前目录下会生成一个test.jpg的image文件。

将image写入tar包

接下来，我们将使用tarfile模块将test.jpg文件写入到一个tar包中。以下是一个示例代码：

import tarfile

# 创建一个名为test.tar的tar包，并打开
with tarfile.open('test.tar', 'w') as tar:
    # 将test.jpg文件写入到tar包中
    tar.add('test.jpg')

print('文件写入成功！')

运行以上代码后，当前目录下会生成一个test.tar的tar包文件，其中包含了test.jpg的image文件。