adversarial_v0.0.2a.py

# import statements
import numpy as np
import math
import random

# initial conditions
# population
paladin = 0
informant = 40
villain = 960
apathetic = 0
total = paladin + informant + villain + apathetic

w_s = 5   # witness subsample size

time = 0    # total time as given by poisson arrival process
turn = 0    # index for turns

# payoff rates for different interactions
reward = 0.3    # reward of crime
punishment = 0.6    # punishment of conviction
image = 0.2   # credibility reduction

# population array where majority of manipulation occurs
pop = np.array([paladin, informant, villain, apathetic , total, time])

history = np.copy(pop)    # data array, keeps record of each round

def arrival():
    """Return a 1-second average Poisson Arrival Process time."""
    return -math.log(1.0 - random.random())

def p_R(q):
    """Returns ratio of Paladins to total population.
    
    Keyword arguments:
    q -- population array
    """
    return float(q[0] * q[4]**(-1))
def i_R(q):
    """Returns ratio of Informants to total population.
    
    Keyword arguments:
    q -- population array
    """
    return float(q[1] * q[4]**(-1))
def v_R(q):
    """Returns ratio of Villains to total population.
    
    Keyword arguments:
    q -- population array
    """
    return float(q[2] * q[4]**(-1))
def a_R(q):
    """Returns ratio of Apathetics to total population.
    
    Keyword arguments:
    q -- population array
    """
    return float(q[3] * q[4]**(-1))

def crim_Choice(q):
    """Returns the index (1 or 2) for the next victimizer in the
    population arrray.
    
    Keyword arguments:
    q -- population array
    """
    inf = i_R(q) / (i_R(q) + v_R(q))
    vil = v_R(q) / (i_R(q) + v_R(q))
    victimizer = np.random.choice(4, p=[0,inf,vil,0])
    return victimizer
def pop_Choice(q):
    """Returns the index (0-3) for a random individual in the
    population arrray.
    
    Keyword arguments:
    q -- population array
    """
    pal = p_R(q)
    inf = i_R(q)
    vil = v_R(q)
    ap = a_R(q)
    victim = np.random.choice(4, p=[pal,inf,vil,ap])
    return victim

def step_One(q):
    """Returns 2 indicies in an array for the victimizer and victim.
    
    Keyword arguments:
    q -- population array
    """
    vm = crim_Choice(q)
    q[vm] -= 1
    q[4] -= 1
    v = pop_Choice(q)
    q[v] -= 1
    q[4] -= 1
    return np.array([vm, v])
def step_Two(vm_v):
    """Returns whether or not a crime was reported to authorities.
    
    Keyword arguments:
    vm_v -- victimizer/victim choice array
    """
    if (v_vm[1]==2) or (v_vm[1]==3):
        return False
    else:
        return True

def step_Three(q, w):
    """Returns the indicies of the witnessing subpopulation
    
    Keyword arguments:
    q -- population array
    w -- number of witnesses to be selected
    """
    pop_choice = np.array([pop_Choice(q)])
    for a in range(w-1):
        pop_choice = np.append(pop_choice, np.array([pop_Choice(q)]))
    return pop_choice

print(step_Two(step_One(pop)))
print(pop)
print(step_Three(pop, w_s))
print(pop)
print(history)

print(arrival())