python - 无法在 Python 的 networkx 中加载简单的 csv

Question

我是 Python 的完全新手，我想使用 networkx 包研究数据集。我不明白这里出了什么问题:

我有一个看起来像这样的 csv(摘录):

['152027', '-6167']
['152027', '-4982']
['152027', '-3810']
['152027', '-2288']
['152027', '-1253']
['152100', '-152100']
['152100', '-86127']

让我们称这个 .csv 文件为 nodes。这里的数字没有特别的含义。它们只是匿名的名字:所以 152027 是一个与个人 -6167、个人 -4982 等相关的人。

我在 Python 中使用以下代码

import csv
import networkx as nx

file = csv.reader(open('nodes', 'rb'), delimiter=',')

G=nx.read_edgelist(file, delimiter=',',nodetype=float,encoding='utf-8')
G.number_of_nodes()

我得到了悲伤的Out[71]: 0我不明白这里出了什么问题。你能帮帮我吗？

Answer 1

nx.read_edgelist 期望第一个变量是文件句柄或文件名字符串，而不是 csv.reader 对象。

根本不要使用csv；试试看

G = nx.read_edgelist('nodes', delimiter=',', nodetype=int, encoding="utf-8")

编辑:如果您需要跳过标题行，您可以这样做

with open('nodes', 'rb') as inf:
    next(inf, '')   # skip a line
    G = nx.read_edgelist(inf, delimiter=',', nodetype=int, encoding="utf-8")