Search code examples
pythongraph-tool

Setting default property value in graph_tool


I need to count how many times a given condition (e.g., "ACondition") is satisfied for each vertex in a graph. To do so, I need to make sure that the vertex property is initialized to zero, which I do explicitly. See the code below.

# Instantiates the graph object and the vertex property. 
import graph_tool.all as gt
g1 = gt.Graph()
g1.vp.AProperty = g1.new_vertex_property("int32_t")

# Sets the vertex property to zero (prior to counting).
for v1 in g1.vertices():
    g1.vp.AProperty[v1] = 0

# Counts the number of times "ACondition" is satisfied for each vertex.
for v1 in g1.vertices():
    if(ACondition == True):
        g1.vp.AProperty[v1] += 1

Is there a way to specify a default value of a property so that I don't need to set its initial value explicitly (i.e, the second block of code above)?


Solution

  • new_vertex_property accepts a single value or a sequence that will be used to initialize the property: g1.new_vertex_property("int32_t", 0)


    I'm not sure why you say you "need to make sure that the vertex property is initialized to zero", because if you don't provide a default, it will be initialized to zero anyway:

    >>> g = gt.Graph()
    >>> g.add_vertex(10)
    >>> g.new_vertex_property('int').a
    PropertyArray([0, 0, 0, 0, 0, 0, 0, 0, 0, 0], dtype=int32)
    

    If the property is a truth value, you should use bool instead.

    You can also use sum and get_array() to count satisfied properties.

    import graph_tool.all as gt
    g = gt.Graph()
    
    # Initialize property foo with False value
    g.vp['foo'] = g.new_vertex_property('bool')
    
    # How many vertices satisfy property foo
    sum(g.vp['foo'].a)