python indices_Python numpy.tril_indices_from() 使用实例

The following are code examples for showing how to use . They are extracted from open source Python projects. You can vote up the examples you like or vote down the exmaples you don’t like. You can also save this page to your account.

Example 1

def get_adjacency_matrix(out_dir, sid, expt_id):

"Returns the adjacency matrix"

vec_path = pjoin(out_dir, sid, '{}_graynet.csv'.format(expt_id))

edge_vec = np.genfromtxt(vec_path)

matrix_size = np.int64( (1.0 + np.sqrt(1.0+8.0*len(edge_vec)))/2.0 )

edge_mat = np.zeros([matrix_size, matrix_size])

# making this symmetric as required by nilearn's plot_connectome (stupid)

# upper tri; diag +1; # lower tri; diag -1

upper_tri = np.triu_indices_from(edge_mat, +1)

lower_tri = np.tril_indices_from(edge_mat, -1)

edge_mat[upper_tri] = edge_vec