Python networkx.read_yaml() Examples
The following are 8
code examples of networkx.read_yaml().
You can vote up the ones you like or vote down the ones you don't like,
and go to the original project or source file by following the links above each example.
You may also want to check out all available functions/classes of the module
networkx
, or try the search function
.
.
Example #1
| Source File: networkx.py From Verum with Apache License 2.0 | 5 votes |
|
def read_graph(self, subgraph_file=None):
if subgraph_file is None:
subraph_file = self.context_graph_file
logging.info("Writing graph.")
# write the graph out
file_format = subgraph_file.split(".")[-1]
if file_format == "graphml":
return nx.read_graphml(subgraph_file)
elif file_format == "gml":
return nx.read_gml(subgraph_file)
elif file_format == "gexf":
return nx.read_gexf(subgraph_file)
elif file_format == "net":
return nx.read_pajek(subgraph_file)
elif file_format == "yaml":
return nx.read_yaml(subgraph_file)
elif file_format == "gpickle":
return nx.read_gpickle(subgraph_file)
else:
logging.warning("File format not found, returning empty graph.")
return nx.MultiDiGraph()
Example #2
| Source File: graph_generator.py From RandWire_tensorflow with MIT License | 5 votes |
|
def graph_generator(model, graph_param, save_path, file_name):
graph_param[0] = int(graph_param[0])
if model == 'ws':
graph_param[1] = int(graph_param[1])
graph = nx.random_graphs.connected_watts_strogatz_graph(*graph_param)
elif model == 'er':
graph = nx.random_graphs.erdos_renyi_graph(*graph_param)
elif model == 'ba':
graph_param[1] = int(graph_param[1])
graph = nx.random_graphs.barabasi_albert_graph(*graph_param)
if os.path.isfile(save_path + '/' + file_name + '.yaml') is True:
print('graph loaded')
dgraph = nx.read_yaml(save_path + '/' + file_name + '.yaml')
else:
dgraph = nx.DiGraph()
dgraph.add_nodes_from(graph.nodes)
dgraph.add_edges_from(graph.edges)
in_node = []
out_node = []
for indeg, outdeg in zip(dgraph.in_degree, dgraph.out_degree):
if indeg[1] == 0:
in_node.append(indeg[0])
elif outdeg[1] == 0:
out_node.append(outdeg[0])
# print(in_node, out_node)
sorted = list(nx.topological_sort(dgraph))
# nx.draw(dgraph)
# plt.draw()
# plt.show()
if os.path.isdir(save_path) is False:
os.makedirs(save_path)
if os.path.isfile(save_path + '/' + file_name + '.yaml') is False:
print('graph_saved')
nx.write_yaml(dgraph, save_path + '/' + file_name + '.yaml')
return dgraph, sorted, in_node, out_node
Example #3
| Source File: nx_yaml.py From qgisSpaceSyntaxToolkit with GNU General Public License v3.0 | 5 votes |
|
def read_yaml(path):
"""Read graph in YAML format from path.
YAML is a data serialization format designed for human readability
and interaction with scripting languages [1]_.
Parameters
----------
path : file or string
File or filename to read. Filenames ending in .gz or .bz2
will be uncompressed.
Returns
-------
G : NetworkX graph
Examples
--------
>>> G=nx.path_graph(4)
>>> nx.write_yaml(G,'test.yaml')
>>> G=nx.read_yaml('test.yaml')
References
----------
.. [1] http://www.yaml.org
"""
try:
import yaml
except ImportError:
raise ImportError("read_yaml() requires PyYAML: http://pyyaml.org/")
G=yaml.load(path)
return G
# fixture for nose tests
Example #4
| Source File: test_yaml.py From qgisSpaceSyntaxToolkit with GNU General Public License v3.0 | 5 votes |
|
def assert_equal(self, G, data=False):
(fd, fname) = tempfile.mkstemp()
nx.write_yaml(G, fname)
Gin = nx.read_yaml(fname);
assert_nodes_equal(G.nodes(), Gin.nodes())
assert_edges_equal(G.edges(data=data), Gin.edges(data=data))
os.close(fd)
os.unlink(fname)
Example #5
| Source File: nx_yaml.py From Carnets with BSD 3-Clause "New" or "Revised" License | 5 votes |
|
def read_yaml(path):
"""Read graph in YAML format from path.
YAML is a data serialization format designed for human readability
and interaction with scripting languages [1]_.
Parameters
----------
path : file or string
File or filename to read. Filenames ending in .gz or .bz2
will be uncompressed.
Returns
-------
G : NetworkX graph
Examples
--------
>>> G=nx.path_graph(4)
>>> nx.write_yaml(G,'test.yaml')
>>> G=nx.read_yaml('test.yaml')
References
----------
.. [1] http://www.yaml.org
"""
try:
import yaml
except ImportError:
raise ImportError("read_yaml() requires PyYAML: http://pyyaml.org/")
G = yaml.load(path, Loader=yaml.FullLoader)
return G
# fixture for nose tests
Example #6
| Source File: test_yaml.py From Carnets with BSD 3-Clause "New" or "Revised" License | 5 votes |
|
def assert_equal(self, G, data=False):
(fd, fname) = tempfile.mkstemp()
nx.write_yaml(G, fname)
Gin = nx.read_yaml(fname)
assert_nodes_equal(list(G), list(Gin))
assert_edges_equal(G.edges(data=data), Gin.edges(data=data))
os.close(fd)
os.unlink(fname)
Example #7
| Source File: nx_yaml.py From aws-kube-codesuite with Apache License 2.0 | 5 votes |
|
def read_yaml(path):
"""Read graph in YAML format from path.
YAML is a data serialization format designed for human readability
and interaction with scripting languages [1]_.
Parameters
----------
path : file or string
File or filename to read. Filenames ending in .gz or .bz2
will be uncompressed.
Returns
-------
G : NetworkX graph
Examples
--------
>>> G=nx.path_graph(4)
>>> nx.write_yaml(G,'test.yaml')
>>> G=nx.read_yaml('test.yaml')
References
----------
.. [1] http://www.yaml.org
"""
try:
import yaml
except ImportError:
raise ImportError("read_yaml() requires PyYAML: http://pyyaml.org/")
G=yaml.load(path)
return G
# fixture for nose tests
Example #8
| Source File: test_yaml.py From aws-kube-codesuite with Apache License 2.0 | 5 votes |
|
def assert_equal(self, G, data=False):
(fd, fname) = tempfile.mkstemp()
nx.write_yaml(G, fname)
Gin = nx.read_yaml(fname)
assert_nodes_equal(list(G), list(Gin))
assert_edges_equal(G.edges(data=data), Gin.edges(data=data))
os.close(fd)
os.unlink(fname)
