У меня было два графика, g1 и g2. Я применил некоторые правила и сопоставил их, как показано ниже.
График :
Я хотел бы преобразовать отображенный граф обратно в фрейм данных 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?
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)
Похожие вопросы
Новые вопросы
python
Python - это многопарадигмальный, динамически типизированный, многоцелевой язык программирования. Он разработан для быстрого изучения, понимания и использования, а также для обеспечения чистого и единообразного синтаксиса. Обратите внимание, что Python 2 официально не поддерживается с 01.01.2020. Тем не менее, для вопросов о Python, связанных с версией, добавьте тег [python-2.7] или [python-3.x]. При использовании варианта Python (например, Jython, PyPy) или библиотеки (например, Pandas и NumPy) включите его в теги.