PILを使った適当なグラフ生成サンプル

draw.py

#! /usr/bin/env python

from PIL import Image, ImageDraw
import random
import socialdata as sd

def drawnetwork(people,links):
    img = Image.new('RGB',(500,500),(255,255,255))
    draw = ImageDraw.Draw(img)
    
    pos=dict(people)

    for (a,b) in links:
        draw.line((pos[a],pos[b]),fill=(255,0,0))

    for n,p in pos.items():
        draw.text(p,n,(0,0,0))

    img.show()

def mk_socialdata():
    people_num = 10
    links_loop = 15
    upper="ABCDEFGHIJKLMNOPQRSTUVWXYZ"
    lower="abcdefghijklmnopqrstuvwxyz"

    a = 50
    b = 450
    people = [(random.choice(upper)
               +random.choice(lower)
               +random.choice(lower)
               +random.choice(lower),
               (random.randint(a,b),random.randint(a,b)))
              for x in range(people_num)]
    linkset = set([(random.choice(people)[0],random.choice(people)[0])
               for i in range(links_loop)])
    new_linkset = [x for x in linkset if len(set(x))!=1]

    return (people, new_linkset)

#if True:
if False:
    p,l = mk_socialdata()
    drawnetwork(p,l)
else:
    drawnetwork(sd.people, sd.links)

mk_link.py

#! /usr/bin/env python
import random

people_num = 10
links_loop = 15
upper="ABCDEFGHIJKLMNOPQRSTUVWXYZ"
lower="abcdefghijklmnopqrstuvwxyz"

a = 50
b = 450
people = [(random.choice(upper)
           +random.choice(lower)
           +random.choice(lower)
           +random.choice(lower),
           (random.randint(a,b),random.randint(a,b)))
          for x in range(people_num)]
print "people =", people


linkset = set([(random.choice(people)[0],random.choice(people)[0])
               for i in range(links_loop)])
new_linkset = [x for x in linkset if len(set(x))!=1]
print "links =",
print new_linkset

mkdata.sh

#! /bin/sh
python mk_link.py > socialdata.py

socialdata.py

people = [('Qsxh', (69, 359)), ('Mhzb', (173, 144)), ('Njne', (292, 130)), ('Cmer', (285, 295)), ('Xqpt', (315, 107)), ('Xkgh', (343, 344)), ('Wlme', (135, 310)), ('Qvut', (58, 391)), ('Dxkr', (241, 314)), ('Fnvk', (52, 76))]
links = [('Xqpt', 'Wlme'), ('Qsxh', 'Fnvk'), ('Wlme', 'Mhzb'), ('Cmer', 'Mhzb'), ('Xqpt', 'Cmer'), ('Wlme', 'Cmer'), ('Wlme', 'Njne'), ('Njne', 'Mhzb'), ('Njne', 'Xqpt'), ('Wlme', 'Fnvk'), ('Dxkr', 'Cmer'), ('Cmer', 'Qvut'), ('Xkgh', 'Qvut'), ('Njne', 'Xkgh')]