Question

我有一个CSV文件，我必须使用 Networkx 和 Pandas 库创建网络，数据是（net.csv） -

,1,2,3,4,5,6,7,8,9,10
1,0,0.257905291,0.775104118,0.239086843,0.002313744,0.416936603,0.194817214,0.163350301,0.252043807,0.251272559
2,0.346100279,0,0.438892758,0.598885794,0.002263231,0.406685237,0.523850975,0.257660167,0.206302228,0.161385794
3,0.753358102,0.222349243,0,0.407830809,0.001714776,0.507573592,0.169905687,0.139611318,0.187910832,0.326950557
4,0.185342928,0.571302688,0.51784403,0,0.003231018,0.295197533,0.216184462,0.153032751,0.216331326,0.317961522
5,0,0,0,0,0,0,0,0,0,0
6,0.478164621,0.418192795,0.646810223,0.410746629,0.002414973,0,0.609176897,0.203461461,0.157576977,0.636747837
7,0.24894327,0.522914349,0.33948832,0.316240267,0.002335929,0.639377086,0,0.410011123,0.540266963,0.587764182
8,0.234017887,0.320967208,0.285193773,0.258198079,0.003146737,0.224412057,0.411725737,0,0.487081815,0.469526333
9,0.302955306,0.080506624,0.261610132,0.22856311,0.001746979,0.014994905,0.63386228,0.486096957,0,0.664434415
10,0.232675407,0.121596312,0.457715027,0.310618067,0.001872929,0.57556548,0.473562887,0.32185564,0.482351246,0

仅供参考，它以有序的方式表示 -

    1           2           3           4           5           6           7           8           9           10
1   0           0.257905291 0.775104118 0.239086843 0.002313744 0.416936603 0.194817214 0.163350301 0.252043807 0.251272559
2   0.346100279 0           0.438892758 0.598885794 0.002263231 0.406685237 0.523850975 0.257660167 0.206302228 0.161385794
3   0.753358102 0.222349243 0           0.407830809 0.001714776 0.507573592 0.169905687 0.139611318 0.187910832 0.326950557
4   0.185342928 0.571302688 0.51784403  0           0.003231018 0.295197533 0.216184462 0.153032751 0.216331326 0.317961522
5   0           0           0           0           0           0           0           0           0           0
6   0.478164621 0.418192795 0.646810223 0.410746629 0.002414973 0           0.609176897 0.203461461 0.157576977 0.636747837
7   0.24894327  0.522914349 0.33948832  0.316240267 0.002335929 0.639377086 0           0.410011123 0.540266963 0.587764182
8   0.234017887 0.320967208 0.285193773 0.258198079 0.003146737 0.224412057 0.411725737 0           0.487081815 0.469526333
9   0.302955306 0.080506624 0.261610132 0.22856311  0.001746979 0.014994905 0.63386228  0.486096957 0           0.664434415
10  0.232675407 0.121596312 0.457715027 0.310618067 0.001872929 0.57556548  0.473562887 0.32185564  0.482351246 0

我有一个代码，应该使用这些值创建网络，但它不起作用（networkx3.py） -

import matplotlib.pyplot as plt
import pandas as pd
import networkx as nx

g = nx.Graph()

connection_success_rates = pd.read_csv('net.csv', index_col=[0])
connection_success = connection_success_rates.values.tolist()

temp = 1
for row in connection_success:
    for i in row:
        if type(row[i]) is str:
            g.add_edge(temp, int(i), weight=(float(row[i])))
    temp = temp+1

pos = nx.spring_layout(g, scale=100.)
nx.draw_networkx_nodes(g, pos)
nx.draw_networkx_edges(g, pos)
nx.draw_networkx_labels(g, pos)
plt.show()

我不知道出了什么问题，每当我运行它时都会抛出以下错误 -

Traceback (most recent call last):
  File "networkx3.py", line 13, in <module>
    if type(row[i]) is str:
TypeError: list indices must be integers, not float

（我帮助编写了@Adonis的原始代码。这是一个修改过的版本，当然不起作用）
每当我想进行小的修改时，我也会得到不同的错误谁能帮我弄清楚出了什么问题？（我不想修改CSV文件或其值）。第一行/标题和第一列/索引也将被忽略在此先感谢您的帮助:)（使用Ubuntu 14.04 32位VM）

Answer 1

for i in row提取元素，而不是索引。但由于您还需要元素索引，因此应使用enumerate(row)。所以你可以这样做：

import matplotlib.pyplot as plt
import pandas as pd
import networkx as nx

g = nx.Graph()

connection_success_rates = pd.read_csv('net.csv', index_col=[0])
connection_success = connection_success_rates.values.tolist()

temp = 1
for row in connection_success:
    for idx, i in enumerate(row):
        if type(i) is str:
            g.add_edge(temp, idx, weight=(float(i)))
    temp = temp+1

pos = nx.spring_layout(g, scale=100.)
nx.draw_networkx_nodes(g, pos)
nx.draw_networkx_edges(g, pos)
nx.draw_networkx_labels(g, pos)
plt.show()

如何使用read_csv将Pandas与Python Networkx一起使用？

1 个答案: