У меня было два графика, g1 и g2. Я применил некоторые правила и сопоставил их, как показано ниже.

График :

enter image description here

Я хотел бы преобразовать отображенный граф обратно в фрейм данных pandas / table /.

From        Attribute        To         
10/10       Start          130/21
130/21      Left           190/190
190/190     Right          240/204
240/204     End             -

Есть ли способ сделать это в Python Pandas?

0
user6053895 15 Июн 2020 в 10:51

1 ответ

Лучший ответ
def make_node_df(G):
    nodes = {}
    for node, attribute in G.nodes(data=True):
        if not nodes.get('node'):
            nodes['node'] = [node]
        else:
            nodes['node'].append(node)

        for key, value in attribute.items():
            if not nodes.get(key):
                nodes[key] = [value]
            else:
                nodes[key].append(value)

    return pd.DataFrame(nodes)

def make_edge_df(G):
    edges = {}
    for source, target, attribute in G.edges(data=True):

        if not edges.get('source'):
            edges['source'] = [source]
        else:
            edges['source'].append(source)

        if not edges.get('target'):
            edges['target'] = [target]
        else:
            edges['target'].append(target)

        for key, value in attribute.items():
            if not edges.get(key):
                edges[key] = [value]
            else:
                edges[key].append(value)
    return pd.DataFrame(edges)


def node_df_to_ebunch(df, nodename='node'):

    attributes = [col for col in df.columns if not col==nodename]

    ebunch = []

    for ix, row in df.iterrows():
        ebunch.append((row[nodename], {attribute:row[attribute] for attribute in attributes}))

    return ebunch

Итак, вы можете преобразовать узлы графика в фрейм данных с помощью make_node_df, а края - с помощью make_edge_df.

Если вы хотите перейти от фрейма данных к графу, вы можете использовать встроенную функцию nx.from_pandas_edgelist с краевым фреймом данных, а для узлов вы можете сделать:

G = nx.Graph()
nodes = node_df_to_ebunch(df, nodename='node')
G.add_nodes_from(nodes)
0
warped 15 Июн 2020 в 12:50