如何使用可选属性调用networkx.add_node(..)?

时间:2019-07-10 00:32:06

标签: python python-3.x networkx graphml

我正在遍历由JSON构造的对象的字典,并使用networkx从这些对象创建顶点。我遇到的问题是某些JSON对象缺少属性,如果执行此操作:

self.graph.add_node(valueToCheck, 
                id=self.vertexDict[valueToCheck], 
                namespace=component["namespace"], 
                tenant=component["tenant"], 
                type=component.get("type")+"Component",
                artifactFileName=component.get("artifactFileName"),
                className=component.get("className"),
                userConfig=component.get("userConfig"),
                sourceType=component.get("sourceType"),
                sinkType=component.get("sinkType"))

然后,我无法使用nx.write_graphml(..)导出图形,因为某些顶点属性的值为None(缺少该属性时,这是component.get(..)的预期输出)。

当JSON对象中可能缺少某些属性时,如何使用networkx构造顶点?

这是我的JSON的样子:

[{'type': 'function',
  'namespace': 'campaigns',
  'name': 'campaign-record-transformer',
  'tenant': 'osp',
  'artifactFileName': 'osp-functions-1.1-SNAPSHOT-jar-with-dependencies.jar',
  'className': 'com.overstock.dataeng.pulsar.functions.CampaignRecordTransformer',
  'inputs': ['persistent://osp/campaigns/campaign-manager'],
  'logTopic': 'persistent://osp/logging/pulsar-log-topic',
  'output': 'persistent://osp/campaigns/campaign-records'},
 {'type': 'function',
  'namespace': 'campaignsTest',
  'name': 'campaign-metadata-transformer',
  'tenant': 'osp',
  'artifactFileName': 'osp-functions-1.1-SNAPSHOT-jar-with-dependencies.jar',
  'className': 'com.overstock.dataeng.pulsar.functions.CampaignMetadataTransformer',
  'logTopic': 'persistent://osp/logging/pulsar-log-topic',
  'output': 'persistent://osp/campaigns/campaign-metadata-output'}]

请注意,第二个对象缺少inputs属性。在实际数据中,至少有8个可选属性可能会以不同组合丢失,并且有数百个这样的对象。

1 个答案:

答案 0 :(得分:1)

我没有发表评论的名声,所以尽管这不是一个完整的答案,但我还是这样发表

您是否尝试过仅排除add_node步骤中缺少的属性?

也就是说,如果缺少键,则不提供键/值对,而不要提供键值为“ None”的键对。

您可以通过使用python加载json然后解压缩组件来轻松实现此目标:

components = json.load(...)
for component in components:
    self.graph.add_node(value, **component)

请参见https://docs.python.org/3/tutorial/controlflow.html#unpacking-argument-lists