如何将excel中的矩阵导入numpy然后绘制图形?

时间:2013-12-09 17:15:58

标签: python excel numpy networkx

我在excel中有一个矩阵,我试图导入并转换为numpy矩阵,然后用networkx绘制它,我将如何进行此操作?我确实有一些代码但不确定我是否正确地执行此操作。

import networkx as nx
import pandas as pd
import numpy 
from numpy import genfromtxt

df=numpy.recfromcsv("Correlation_test.csv", delimiter=',', skiprows=1)

nx.Graph(df)

由于

这是我到目前为止所得到的,但我一直收到一条错误,说“输入不是正确的numpy矩阵或数组”。

1 个答案:

答案 0 :(得分:6)

如果数据是CSV文件,如下所示:

my_graph.csv

My graph data
0,1,1,0,0
1,0,0,1,0
1,0,0,1,1
0,1,1,0,1
0,0,1,1,0

加载和绘制结果图很简单:

import numpy as np
import networkx as nx
import pylab as plt

A = np.genfromtxt("my_graph.csv",delimiter=',',skiprows=1)
G = nx.Graph(A)
nx.draw(G)
plt.show()

enter image description here