-
Notifications
You must be signed in to change notification settings - Fork 2
/
load_data.py
80 lines (59 loc) · 2.33 KB
/
load_data.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
# -*- coding: utf-8 -*-
"""
Created on Thu Oct 29 10:39:12 2020
This python file contains functions to load data stored in this git-hub
repository and store it in a dictionary.
@author: Vincent Bazinet
"""
import numpy as np
import os
def load_dict(path):
'''
Function to load a dictionary containing the required data to generate
our results and figures. See main_script.py for information about the
entries in this dictionary.
Parameters
----------
path : str
Relative path to the data folder (from the current working dictionary).
Example: 'data/LAU1000'.
Returns
-------
data : dict
Dictionary containing the data to generate our results and figures.
'''
data = {}
data['sc'] = np.load(os.path.join(os.getcwd(),
path,
'sc.npy'))
data['fc'] = np.load(os.path.join(os.getcwd(),
path,
'fc.npy'))
data['coords'] = np.load(os.path.join(os.getcwd(),
path,
'coords.npy'))
data['lhannot'] = os.path.join(os.getcwd(),
path,
'lh.annot')
data['rhannot'] = os.path.join(os.getcwd(),
path,
'rh.annot')
data['noplot'] = None
data['order'] = 'RL'
data['rsn'] = np.load(os.path.join(os.getcwd(),
path,
'rsn.npy'))
data['rsn_names'] = np.load(os.path.join(os.getcwd(),
path,
'rsn_names.npy')).tolist()
data['ve'] = np.load(os.path.join(os.getcwd(),
path,
've.npy'))
data['ve_names'] = np.load(os.path.join(os.getcwd(),
path,
've_names.npy')).tolist()
data['ci'] = []
ci_path = os.path.join(os.getcwd(), path, 'sc_ci')
for ci in os.listdir(os.path.join(ci_path)):
data['ci'].append(np.load(os.path.join(ci_path, ci)))
return data