我只是在学习python,所以我很感激帮助。我有一个双列数据集,第一列是唯一的 id,第二列是一串项目。我正在使用 networkX 从数据中创建一棵树(见下文)。我需要知道每个级别的项目频率。例如,对于 A (1,2,3,4) 中的路径,每个节点的计数应为 1:4、2:2、3:2 和 4:2。如何获取节点数?
我的数据如下所示:
A 1, 2, 3, 4
B 1, 2, 1, 4
C 1, 3, 4, 3
D 1, 4, 3, 2
我到目前为止的代码如下:
#create graph
G = nx.MultiGraph()
#read in strings from csv
testfile = 'C:…file.txt'
with open(testfile, "r") as f:
line = f.readline
f = (i for i in f if '\t' in i.rstrip())
for line in f:
customerID, path = line.rstrip().split("\t")
path2 = path.rstrip("\\").rstrip("}").split(",")
pathInt = list()
for x in path2:
if x is not None:
newx = int(x)
pathInt.append(newx)
print(pathInt)
varlength = len(pathInt)
pathTuple = tuple(pathInt)
G.add_path([pathTuple[:i+1] for i in range(0, varlength)])
nx.draw(G)
plt.show() # display