Apply vertex paint to a vertex

Question

Through the user interface, you paint a vertex by hitting V in 3D View, and clicking on a vertex.

How can this be achieved in Python?

Answer 1

This function should do the trick:

import bpy

def color_vertex(obj, vert, color):
    """Paints a single vertex where vert is the index of the vertex
    and color is a tuple with the RGB values."""

    mesh = obj.data 
    scn = bpy.context.scene

    #check if our mesh already has Vertex Colors, and if not add some... (first we need to make sure it's the active object)
    scn.objects.active = obj
    obj.select = True
    if mesh.vertex_colors:
        vcol_layer = mesh.vertex_colors.active
    else:
        vcol_layer = mesh.vertex_colors.new()

    for poly in mesh.polygons:
        for loop_index in poly.loop_indices:
            loop_vert_index = mesh.loops[loop_index].vertex_index
            if vert == loop_vert_index:
                vcol_layer.data[loop_index].color = color

#example usage
color = (1.0, 0.0, 1.0)  # pink
color_vertex(bpy.context.scene.objects['Cube'], 1, color)