Simplest way to save array into raster file in Python

Question

With one 2-d array in the shape of (100, 100), I want to save it into raster file in .tiff format.

I can use gdal package to read tiff files which are already exist. But I still can't find a simple way to transform the 2-d array into tiff file.

Using plt.imsave("xx.tif",array) or

def array_to_raster(array):
    """Array > Raster
    Save a raster from a C order array.
    :param array: ndarray
     """
    dst_filename = 'xxx.tiff'
    x_pixels = 100  # number of pixels in x
    y_pixels = 100  # number of pixels in y
    driver = gdal.GetDriverByName('GTiff')
    dataset = driver.Create(
           dst_filename,
           x_pixels,
           y_pixels,
           1,
           gdal.GDT_Float32, )
    dataset.GetRasterBand(1).WriteArray(array)
    dataset.FlushCache()  # Write to disk.
    return dataset, dataset.GetRasterBand(1)

They all failed to achieve my target. The second method was adapted from here which can transform an array into a geotiff with a projection.

Is there some simple way to save array into .tiff, so I can call it by import the tiff file next time.

Any advices would be appreciate.

sneeze_shiny · Accepted Answer · 2022-03-17 09:12:28Z

8

A tif raster could be considered as 'array+proj+geotransforms'. If you want to write an array to a tiff ,you can refer to the following code:

dst_filename = 'xxx.tiff'
x_pixels = 100  # number of pixels in x
y_pixels = 100  # number of pixels in y
driver = gdal.GetDriverByName('GTiff')
dataset = driver.Create(dst_filename,x_pixels, y_pixels, 1,gdal.GDT_Float32)
dataset.GetRasterBand(1).WriteArray(array)

# follow code is adding GeoTranform and Projection
geotrans=data0.GetGeoTransform()  #get GeoTranform from existed 'data0'
proj=data0.GetProjection() #you can get from a exsited tif or import 
dataset.SetGeoTransform(geotrans)
dataset.SetProjection(proj)
dataset.FlushCache()
dataset=None

edited Mar 17, 2022 at 9:12

sneeze_shiny

3381 gold badge10 silver badges34 bronze badges

answered Jan 14, 2017 at 13:24

Cobin

9301 gold badge15 silver badges26 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

La Cordillera Over a year ago

For the record, the last 4 lines should read: dataset.SetGeoTransform(geotrans) dataset.SetProjection(proj) dataset.FlushCache() dataset=None

xbsd Over a year ago

What is outds ?

do-me · Accepted Answer · 2022-01-09 17:11:41Z

2

Easiest way with imageio

If you don't care about its projection it's a oneliner:

import imageio
imageio.imwrite('test.tiff', [[0,255],[255,0]])

answered Jan 9, 2022 at 17:11

do-me

2,3682 gold badges15 silver badges18 bronze badges

Collectives™ on Stack Overflow

Simplest way to save array into raster file in Python

2 Answers 2

2 Comments

Easiest way with imageio

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

2 Comments

Easiest way with imageio

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related