PyPNG: Map to PNG function (array to PNG)

map2png.py

#https://pythonhosted.org/pypng/ex.html

import png
import time



def map2png(name, amap, w, h):
	amap2 = list(amap)
	for i in range(len(amap2)):
		if amap2[i] == 100:
			amap2[i] = 255
		if amap2[i] == -1:
			amap2[i] = 100

	l = []
	for i in range(0, len(amap2), w):
		l += [tuple(amap2[i:i+w])]
	f = open(name+".png", 'wb')
	w = png.Writer(w, h, greyscale=True)
	w.write(f, l)
	f.close()

#m = [-1,0,0,0,0,100]
#map2png("map_"+time.strftime("%H:%M:%S"),m, 3, 2)