Skip to content

Commit da715a9

Browse files
author
Anton Eriksson
committed
feat(python): Add write_state_network and write_pajek
1 parent fdb3ce1 commit da715a9

File tree

1 file changed

+29
-0
lines changed

1 file changed

+29
-0
lines changed

interfaces/python/infomap.py

Lines changed: 29 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2001,9 +2001,38 @@ def write_csv(self, filename, states=False):
20012001
"""
20022002
return self.writeCsvTree(filename, states)
20032003

2004+
def write_state_network(self, filename):
2005+
"""Write internal state network to file.
2006+
2007+
See Also
2008+
--------
2009+
write_pajek
2010+
2011+
Parameters
2012+
----------
2013+
filename : str
2014+
"""
2015+
return self.network.writeStateNetwork(filename)
2016+
2017+
def write_pajek(self, filename, flow=False):
2018+
"""Write network to a Pajek file.
2019+
2020+
See Also
2021+
--------
2022+
write_state_network
2023+
2024+
Parameters
2025+
----------
2026+
filename : str
2027+
flow : bool, optional
2028+
If the flow should be included. Default ``False``.
2029+
"""
2030+
return self.network.writePajekNetwork(filename, flow)
2031+
20042032
# for the method "write"
20052033
write_ftree = write_flow_tree
20062034
write_nwk = write_newick
2035+
write_net = write_pajek
20072036

20082037

20092038
def main():

0 commit comments

Comments
 (0)