矩阵仅使用一列用于标题和行

时间:2016-07-24 20:40:27

标签: python numpy pandas

假设我在csv中有一个像这样的名字列表:

Nom;Link;NonLink
Deb;John;
John;Deb;
Martha;Travis;
Travis;Martha;
Allan;;
Lois;;
Jayne;;
Brad;;Abby
Abby;;Brad

我使用numpy导入它:

import numpy as np
file = np.genfromtxt('liste.csv', dtype=None, delimiter =';',skip_header=1)

现在,我正在隔离我的第一栏:

Nom = np.array(file[:,0])

我想创建一个仅使用第一列的矩阵来得到像这样的结果:

      Deb    John    Martha    etc...   
Deb     0       0         0     ...
John    0       0         0     ...
Martha  0       0         0     ...
etc...

那是否有一个numpy函数?

编辑:我的最终目标是制作一个小程序来分配桌子上的席位,其中Link中的人必须坐在同一张桌子上,NonLink不能在同一张桌子上。

谢谢,

1 个答案:

答案 0 :(得分:1)

您可以使用pandas并使用var flower = element.all(by.repeater('flower in model.flower')).first(); var addButton = flower.element(by.buttonText("Add")); addButton.click(); 变量创建数据框。 像这样:

Nom