NetworkX导出GraphML格式时可作为属性的有效数据类型咨询

嘿，我正好对NetworkX导出GraphML的属性类型这块熟，给你详细说清楚：

GraphML本身的规范定义了几种原生数据类型，NetworkX在实现导出时，只支持能映射到这些原生类型的Python数据，具体包括：

• 数值类型：int（整数）、float（浮点数），甚至numpy的数值类型（比如numpy.int64、numpy.float32）也能被自动转换后支持
• 字符串类型：str（普通Python字符串，注意不能是bytes类型）
• 布尔类型：bool（True/False，会被映射为GraphML的boolean类型）

像你遇到的list、dict这类复合类型，还有set、自定义类实例等，GraphML没有对应的原生类型，所以NetworkX会判定为无效属性，导出时抛出错误。

既然没有内置的忽略函数，咱们可以自己写一个遍历图的节点、边、全局属性的处理函数，要么过滤掉无效属性，要么把它们转换成支持的类型（比如转成JSON字符串保留信息）。这里给你一个实用的示例：

import networkx as nx
import json

def prepare_graph_for_graphml_export(G):
    # 定义NetworkX支持的GraphML属性类型
    SUPPORTED_TYPES = (int, float, str, bool)
    
    # 处理图的全局属性
    cleaned_graph_attrs = {}
    for attr_key, attr_value in G.graph.items():
        if isinstance(attr_value, SUPPORTED_TYPES):
            cleaned_graph_attrs[attr_key] = attr_value
        else:
            # 两种选择：要么跳过（注释下面一行，换成pass），要么转成JSON字符串保留信息
            cleaned_graph_attrs[attr_key] = json.dumps(attr_value)
    G.graph = cleaned_graph_attrs
    
    # 处理节点属性
    for node_id, node_attrs in G.nodes(data=True):
        cleaned_node_attrs = {}
        for attr_key, attr_value in node_attrs.items():
            if isinstance(attr_value, SUPPORTED_TYPES):
                cleaned_node_attrs[attr_key] = attr_value
            else:
                cleaned_node_attrs[attr_key] = json.dumps(attr_value)
        G.nodes[node_id] = cleaned_node_attrs
    
    # 处理边属性
    for u, v, edge_attrs in G.edges(data=True):
        cleaned_edge_attrs = {}
        for attr_key, attr_value in edge_attrs.items():
            if isinstance(attr_value, SUPPORTED_TYPES):
                cleaned_edge_attrs[attr_key] = attr_value
            else:
                cleaned_edge_attrs[attr_key] = json.dumps(attr_value)
        G.edges[u, v] = cleaned_edge_attrs
    
    return G

使用方法 #

# 假设你的原始图是G
original_G = nx.Graph()
# 给图、节点、边添加属性（包括list/dict类型）...

# 处理属性
ready_G = prepare_graph_for_graphml_export(original_G)
# 导出为GraphML
nx.write_graphml(ready_G, "exported_graph.graphml")

如果不需要保留list/dict这类属性的信息，直接把函数里else块的cleaned_xxx_attrs[attr_key] = json.dumps(attr_value)换成pass就行，这样无效属性就会被过滤掉，不会参与导出。

内容的提问来源于stack exchange，提问作者ajreckof