在txt.file中使用3列邊表

Question

我是NetworkX的新手，但是有問題。 我有.txt文件，該文件包含3列中的許多數據，這些數據列由制表符分隔，如下所示：

1   21   \N
2   61   \N
2   62   1201231
50  11   54432

如何僅使用前兩列使用read.edgelist('data.txt', create_using=nx.Graph(), nodetype=int) ？

Answer 1

不使用nx.read_edgelist()而是默認構造函數的解決方案如下。

>>> with open("file.txt") as f:
...     g = nx.Graph([line.split()[:2] for line in f])
... 
>>> g.edges()
EdgeView([('1', '21'), ('2', '61'), ('2', '62'), ('50', '11')])

如果節點類型必須為整數，則可以在讀取文件時將其轉換為int 。

例如，

[tuple(map(int,line.split()[:2])) for line in f]