Convert Numpy array to image

Question

I have a little nasty problem. I have a numpy array filled with float values. The background is that the array's values represent the water depth of a square grid.

345.34 888.78 999.35
300.00  98.00  88.45
440.89 423.56  11.68

I want to convert/save this array as image. Thereby, a range of values shall be represented by one color.

"dark blue" "midnight blue" "midnight blue"       #00008B #191970 #191970
"dark blue" "medium blue"   "medium blue"    or   #00008B #0000CD #0000CD 
"navy"      "navy"          "blue"                #000080 #000080 #0000FF

I circumvented this problem by saving the array as ASCII file and converting it in ArcGIS to a raster map but I want to avoid ArcGIS since I have too many arrays/maps to do it manually.

My attempt was to replace a range of values with integers through masks. Then I convert the replaced float values to strings and replace the strings with the RGB color codes. I finally use Image.fromarray to create an image. The result is a mess. It does not resemble an inundation map.

I hope somebody knows a doable way.

Saullo G. P. Castro · Accepted Answer · 2014-09-21 10:07:34Z

3

You can achieve that simply passing your data to matplotlib.pyplot.imshow(). The color scale can be adjusted by the color map of your choice. Assuming you have the data saved in a text file:

import numpy as np
import matplotlib.cm as cm
import matplotlib.pyplot as plt

data = np.loadtxt('data.txt')
plt.imshow(data, cmap=cm.Blues)
plt.show()

edited Sep 21, 2014 at 10:07

answered Sep 21, 2014 at 10:00

Saullo G. P. Castro

59.4k28 gold badges191 silver badges244 bronze badges

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

1 Comment

Scrox Over a year ago

I know there is something like that. great!

Collectives™ on Stack Overflow

Convert Numpy array to image

1 Answer 1

1 Comment

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Related