如何将一个2D数组切分成多个块

要将一个2D数组切分成多个块，可以考虑使用以下几种方法，具体取决于如何定义块的划分规则和需求。如果你希望将2D数组均匀地切分成固定大小的小块，可以使用简单的循环和切片操作。

在这里插入图片描述

1、问题背景

Python 中, 如果有一个 raw 数据文件，将其读入到字节缓冲区（python 字符串），其中每一个数据值代表一个2d 数组中 8 位像素。已知此图片的宽度和高度，想将图片切分成多个块，并且每一个块的面积必须大于最小块面积（如：1024 字节），小于最大块面积（如：2048 字节）。这些块的高度和宽度是任意的，只要满足面积约束即可，并且块的大小不必相同。此外，输入数据的长度也不一定是2的幂。

2、解决方案

方法一:

为了代码尽量简洁，可以将数据存储为按行存储的行。

import collections
import operatordef split_(seq, size):return [seq[i:i+size] for i in range(0,len(seq),size)]def split_image(data, width, height, MIN_AREA, MAX_AREA):tiles = list()if width >= MIN_AREA:# 每行可以细分为多个块tile_width = width / (width / MIN_AREA) # 整数除法rows = split_(data, width)row_tiles = [split_(row, tile_width) for row in rows]tiles = reduce(operator.add, row_tiles)elif width < MIN_AREA:# 每个块由多行组成min_tile_height = int(MIN_AREA / width) + 1tile_height = height / (height / min_tile_height)tile_size = tile_height * widthtiles = split_(data, tile_size)if len(tiles[-1]) < MIN_AREA:if (tile_height > 2):tiles[-2] += tiles[-1]del tiles[-1]else: # tile_height == 2, the case 1 don't pass here# 特殊情况，我们需要垂直分割最后三行# 如果宽度也是3，那么我们就遇到了问题，但如果我们在这里# 那么MIN_AREA为4，MAX_AREA为8，行数大于等于5if width > 3:last_three_rows = split_(tiles[-2] + tiles[-1], width)tiles[-2] = reduce(operator.add,[row[:width/2] for row in last_three_rows])tiles[-1] = reduce(operator.add,[row[width/2:] for row in last_three_rows])else: # width = 3 and MIN_AREA = 4last_five_rows = reduce(operator.add, tiles[-3:])three_columns = [last_five_rows[i::3] for i in range(3)]tiles[-3:] = three_columnsreturn tiles

方法二:

如果是在处理图片，可以使用 PIL（Python Imaging Library）。

import Imagedef split_image(imagefile, MIN_AREA, MAX_AREA):i = Image.open(imagefile)width, height = i.sizedata = list(i.getdata())tiles = list()if width >= MIN_AREA:tile_width = width / (width / MIN_AREA) # 整数除法for y in range(0, height, tile_width):for x in range(0, width, tile_width):box = (x, y, x + tile_width, y + tile_width)region = i.crop(box)tiles.append(list(region.getdata()))elif width < MIN_AREA:min_tile_height = int(MIN_AREA / width) + 1tile_height = height / (height / min_tile_height)tile_size = tile_height * widthfor y in range(0, height, tile_height):for x in range(0, width):box = (x, y, x + width, y + tile_height)region = i.crop(box)tiles.append(list(region.getdata()))return tiles