如何读一个文件夹里png图片python

要读取一个文件夹里的 PNG 图片文件，可以使用 Python 的 os 和 PIL 库。以下是主要步骤：使用 os 库遍历文件夹、使用 PIL 库读取图片、处理图片。其中，使用 os 库遍历文件夹是关键步骤之一。下面将详细描述如何在 Python 中读取文件夹里的 PNG 图片文件。

一、使用 os 库遍历文件夹

在处理图片之前，需要先获取文件夹中所有 PNG 文件的路径。使用 os 库可以方便地遍历指定文件夹中的所有文件。

1. 获取文件列表

首先，导入 os 库，并使用 os.listdir() 函数获取指定文件夹中的所有文件名。然后，过滤出所有 PNG 文件。

import os
def get_png_files(folder_path):
    files = os.listdir(folder_path)
    png_files = [file for file in files if file.endswith('.png')]
    return png_files
folder_path = 'path/to/your/folder'
png_files = get_png_files(folder_path)
print(png_files)

2. 拼接文件路径

获取文件名后，还需要将其与文件夹路径拼接，形成完整的文件路径。可以使用 os.path.join() 函数实现这一操作。

def get_full_file_paths(folder_path):
    png_files = get_png_files(folder_path)
    full_paths = [os.path.join(folder_path, file) for file in png_files]
    return full_paths
full_paths = get_full_file_paths(folder_path)
print(full_paths)

二、使用 PIL 库读取图片

在获取到所有 PNG 文件的完整路径后，可以使用 PIL 库（Python Imaging Library）读取图片文件。PIL 库的 Image 模块提供了打开和操作图片的函数。

1. 安装 PIL 库

首先，确保安装了 PIL 库。可以使用 pip 安装：

pip install pillow

2. 读取图片

使用 PIL 库的 Image 模块打开和读取图片文件。可以通过遍历文件路径列表，依次读取每个 PNG 文件。

from PIL import Image
def read_images(file_paths):
    images = []
    for file_path in file_paths:
        image = Image.open(file_path)
        images.append(image)
    return images
images = read_images(full_paths)
for img in images:
    img.show()  # 显示图片

三、处理图片

读取图片后，可以对其进行各种处理操作，如调整大小、裁剪、转换颜色等。以下是一些常见的图片处理操作示例。

1. 调整图片大小

可以使用 PIL 库的 resize() 函数调整图片的大小。

def resize_images(images, size):
    resized_images = [img.resize(size) for img in images]
    return resized_images
new_size = (100, 100)
resized_images = resize_images(images, new_size)

2. 裁剪图片

可以使用 PIL 库的 crop() 函数裁剪图片。

def crop_images(images, box):
    cropped_images = [img.crop(box) for img in images]
    return cropped_images
crop_box = (10, 10, 200, 200)
cropped_images = crop_images(images, crop_box)

3. 转换颜色模式

可以使用 PIL 库的 convert() 函数转换图片的颜色模式，如将图片转换为灰度图。

def convert_images_to_grayscale(images):
    grayscale_images = [img.convert('L') for img in images]
    return grayscale_images
grayscale_images = convert_images_to_grayscale(images)

四、保存处理后的图片

处理完图片后，可以将其保存到指定文件夹。使用 PIL 库的 save() 函数可以将图片保存到文件。

def save_images(images, folder_path):
    if not os.path.exists(folder_path):
        os.makedirs(folder_path)
    for i, img in enumerate(images):
        img.save(os.path.join(folder_path, f'image_{i}.png'))
output_folder = 'path/to/output/folder'
save_images(grayscale_images, output_folder)

五、综合示例

将以上步骤综合起来，完成一个读取文件夹中 PNG 图片、处理图片并保存处理后图片的完整示例。

import os
from PIL import Image
def get_png_files(folder_path):
    files = os.listdir(folder_path)
    png_files = [file for file in files if file.endswith('.png')]
    return png_files
def get_full_file_paths(folder_path):
    png_files = get_png_files(folder_path)
    full_paths = [os.path.join(folder_path, file) for file in png_files]
    return full_paths
def read_images(file_paths):
    images = []
    for file_path in file_paths:
        image = Image.open(file_path)
        images.append(image)
    return images
def resize_images(images, size):
    resized_images = [img.resize(size) for img in images]
    return resized_images
def crop_images(images, box):
    cropped_images = [img.crop(box) for img in images]
    return cropped_images
def convert_images_to_grayscale(images):
    grayscale_images = [img.convert('L') for img in images]
    return grayscale_images
def save_images(images, folder_path):
    if not os.path.exists(folder_path):
        os.makedirs(folder_path)
    for i, img in enumerate(images):
        img.save(os.path.join(folder_path, f'image_{i}.png'))
主程序
folder_path = 'path/to/your/folder'
output_folder = 'path/to/output/folder'
full_paths = get_full_file_paths(folder_path)
images = read_images(full_paths)
resized_images = resize_images(images, (100, 100))
cropped_images = crop_images(resized_images, (10, 10, 90, 90))
grayscale_images = convert_images_to_grayscale(cropped_images)
save_images(grayscale_images, output_folder)