#!/usr/bin/env python3
# -*- coding: utf-8 -*- #
import pygame
def color_surface(surface, color):
red, green, blue = color
arr = pygame.surfarray.pixels3d(surface)
arr[:,:,0] = red
arr[:,:,1] = green
arr[:,:,2] = blue