Export single band from a multiband stacked raster with rasterio

Question

I have been trying to break down my multi-band raster into individual bands with rasterio and I am getting this error:

ValueError: Source shape (11356, 9488) is inconsistent with given indexes 1

here's the code:

import os
import rasterio


 src = rasterio.open(wv_tif)
    for band in range(1, src.count):
        single_band = src.read(band)

        # get the output name
        out_name = os.path.basename(wv_tif)
        file, ext = os.path.splitext(out_name)
        name = file + "_" + "B" + str(band) + ".tif"
        out_img = os.path.join(folder, name)

        print(out_img + " done")

        # Copy the metadata
        out_meta = src.meta.copy()

        out_meta.update({"count": 1})

        # save the clipped raster to disk
        with rasterio.open(out_img, "w", **out_meta) as dest:
            dest.write(single_band)

I have no idea what is going wrong

Answer 1

You're telling rasterio to write a dataset so it's expecting an array shape of (bands, rows, cols), but you're passing a single band, an array shape of (rows, cols).

Either reshape your array:

dest.write(single_band[np.newaxis,...])`

Or tell rasterio to write a single band:

dest.write(single_band, 1)

Answer 2

I had the same issue and with some changes I reached one script that worked fine.

import os
import rasterio

Defining different objects to be used:,

src = 'C:/home/User/Documents/file.tif'
src2 = rasterio.open('C:/home/User/Documents/file.tif')
folder = 'C:/home/User/Documents'

for band in range(1, src2.count +1): #The +1 allows the writing of the last band
        single_band = src2.read(band)

        # get the output name
        out_name = os.path.basename(src)
        file, ext = os.path.splitext(out_name)
        name = file + "_" + "B" + str(band) + ".tif"
        out_img = os.path.join(folder, name)

        print(out_img + " done")

        # Copy the metadata
        out_meta = src2.meta.copy()

        out_meta.update({"count": 1})

        # save the clipped raster to disk
        with rasterio.open(out_img, "w", **out_meta) as dest:
            dest.write(single_band, 1)