gdf形式のデータをD3のフォースレイアウトで使えるように変換する

【D3.js】Facebookでフレンドがどれだけ「いいね！」しているかの視覚化
↑の補足。

Facebookから取得できるソーシャルグラフデータ(gdf)をD3で使えるようにする手順

netvizzからダウンロードしたgdfデータのフォーマットは以下

nodedef>name VARCHAR,label VARCHAR,gender VARCHAR,locale VARCHAR,agerank
INT
23917067,Jorge,male,en_US,106
23931909,Haruna,female,en_US,105
35702006,Joseph,male,en_US,104
503839109,Damian,male,en_US,103
532735006,Isaac,male,es_LA,102
. . .
edgedef>node1 VARCHAR,node2 VARCHAR
23917067,35702006
23917067,629395837
23917067,747343482
23917067,755605075
23917067,1186286815
. . .

まずは「nodedef」のデータと「edgedef」で分け、二つのcsvデータを作る

「name」のカラム名は分かりやすいように「id」に変更した

nodes.csv

id,label,gender,locale,agerank
23917067,Jorge,male,en_US,106
23931909,Haruna,female,en_US,105
35702006,Joseph,male,en_US,104
503839109,Damian,male,en_US,103
532735006,Isaac,male,es_LA,102
. . .

links.csv

node1,node2
23917067,35702006
23917067,629395837
23917067,747343482
23917067,755605075
23917067,1186286815
. . .

D3 Force Layout で使うには

フォースレイアウトでは、linkデータとしてnodeの結びつきをインデックスで表さないとならない。

（例

var nodes = [ {id:1, name:"hoge"}, {id:2, name:"fuga"} ]

hogeからfugaへの結びつき表すにはlinkデータで以下のように表現する。

var links = [ {source:0 ,target:1 }];

あくまでnodes配列のインデックス値での指定しなくてはならないことに注意。

そこで、gdfのデータからlink(source, target)データを作成する

d3.csv('nodes.csv', function(nodes){
    d3.csv('links.csv', function(links){
        d3main(nodes, links);
    });
});

function d3main(nodes, links){
    var nodeID = {}; //idの位置(配列のindex)を保存する
    for(var i=0; i < nodes.length; i++){
        var id = nodes[i].id;
        nodeID[id] = i; 
    }

    //source,targetの取得
    for(var i=0; i < links.length; i++){
        links[i]['source'] = nodeID[links[i]['node1']];
        links[i]['target'] = nodeID[links[i]['node2']];
    }

    graph = {'nodes': nodes, 'links': links };
    console.log(graph);

}

作成したデータをフォースレイアウトで表示する

 var width = 960,
    height = 500;

var force = d3.layout.force()
    .charge(-120)
    .linkDistance(30)
    .size([width, height]);

force
    .nodes(graph.nodes)
    .links(graph.links)
    .start();

var link = svg.selectAll(".link")
    .data(graph.links)
    .enter()
    .append("line")
    .attr("class", "link")

var node = svg.selectAll(".node")
    .data(graph.nodes)
    .enter()
    .append("circle")
    .attr("class", "node")
    .attr("r", 5)
    .style("fill", "blue")
    .call(force.drag);

node.append("title")
    .text(function(d) { return d.label; });

force.on("tick", function() {
    link.attr("x1", function(d) { return d.source.x; })
        .attr("y1", function(d) { return d.source.y; })
        .attr("x2", function(d) { return d.target.x; })
        .attr("y2", function(d) { return d.target.y; });

    node.attr("cx", function(d) { return d.x; })
        .attr("cy", function(d) { return d.y; });
});