创建双模网络

Question

使用Python 3.2我试图将数据从CSV文件转换为双模网络。对于那些不知道这意味着什么的人来说，这个想法很简单：

这是我的数据集的片段：

Project_ID    Name_1    Name_2    Name_3    Name_4 ... Name_150
    1           Jean      Mike
    2           Mike
    3           Joe       Sarah     Mike      Jean        Nick
    4           Sarah     Mike
    5           Sarah     Jean      Mike      Joe

我想创建一个CSV，将Project_ID放在CSV的第一行，每个唯一的名称放在第一列（单元格A1为空白），然后在i，j单元格中为1，如果该人员在给定项目。注意：我的数据具有全名（中间首字母），没有两个人具有相同的名称，因此不会有任何重复。

最终的数据输出如下：

            1             2              3              4              5 
Jean        1             0              1              0              1
Mike        1             1              1              1              1
Joe         0             0              1              0              1
Sarah       0             0              1              1              1
...        ...           ...            ...            ...            ...
Nick        0             0              1              0              0

Answer 1

首先使用CVS reader

import csv

with open('some.csv', 'rb') as f:
    reader = csv.reader(f)
    for row in reader:
        print row

请注意，行将读为每行的数组。

应该在开始之前创建输出数组。从this question开始，您可以这样做

buckets = [[0 for col in range(5)] for row in range(10)]