|
16 | 16 |
|
17 | 17 | import pandas as pd |
18 | 18 | import datetime as dt |
| 19 | +import os |
19 | 20 | import warnings |
20 | 21 | import copy |
21 | 22 | import subprocess as sub |
@@ -819,8 +820,10 @@ def generate_network(self, mode1, mode2, colours=None, edge_helper=net_edges_sim |
819 | 820 | graph.node[n]["colour"] = "plum" |
820 | 821 | else: |
821 | 822 | graph.node[n]['colour'] = 'lightgrey' |
| 823 | + print('Created a MultiGraphPlus network object with {} nodes and {} edges.'.format(graph.number_of_nodes(), graph.number_of_edges())) |
822 | 824 | return graph |
823 | 825 |
|
| 826 | + |
824 | 827 | def write_edges(self, fname, mode1, mode2, helper=net_edges_simple, edge_attribute=['weight', 'date']): |
825 | 828 | """ |
826 | 829 | Writes an edge list with attributes. |
@@ -893,7 +896,7 @@ def write_edges(self, fname, mode1, mode2, helper=net_edges_simple, edge_attribu |
893 | 896 | f.write("\n") |
894 | 897 | # Close file and print summary. |
895 | 898 | f.close() |
896 | | - print("Wrote edgelist with attributes to {}.".format(fname)) |
| 899 | + print("Wrote edgelist with attributes to {} in {}.".format(fname, os.getcwd())) |
897 | 900 |
|
898 | 901 | def write_nodes(self, fname, mode1, mode2, keep_atom1=[], keep_vector1=[], keep_atom2=[], keep_vector2=[]): |
899 | 902 | """ |
@@ -970,4 +973,4 @@ def write_nodes(self, fname, mode1, mode2, keep_atom1=[], keep_vector1=[], keep_ |
970 | 973 | else: |
971 | 974 | f.write("\n") |
972 | 975 | f.close() |
973 | | - print("Wrote node attributes to {}.".format(fname)) |
| 976 | + print("Wrote node attributes to {} in {}.".format(fname, os.getcwd())) |
0 commit comments