重新运行(相同)代码后出现KeyError

问题描述

当我尝试运行以下代码时返回KeyError：

import pandas as pd
import networkx as nx
from matplotlib import pyplot as plt

G = nx.from_pandas_edgelist(df, 'Source', 'Target',  edge_attr=True)
df_pos = nx.spring_layout(G,k = 0.3) 

nx.draw_networkx(G, df_pos)
plt.show()

node_color = [
    '#1f78b4' if G.nodes[v]["Label_Source"] == 0 # This returns the error
    else '#33a02c' for v in G]

我的数据框有以下列：

Source  Target      Label_Source Label_Target
string1  string2          1           0
string1  string4          1           1
string4  string5          1           0

我知道KeyError名称有很多问题，但奇怪的是，在不更改任何内容的情况下，在运行代码时出现了这个错误，但在此之前，我可以运行代码而没有错误。一切都没有改变。 这方面的任何帮助都是最好的！

回溯：

KeyError                                  Traceback (most recent call last)
<ipython-input-9-1266cd6e1844> in <module>
----> 1 node_color = [
      2     '#1f78b4' if G[v]["Label_Source"] == 0
      3     else '#33a02c' for v in G]

<ipython-input-9-1266cd6e1844> in <listcomp>(.0)
      1 node_color = [
----> 2     '#1f78b4' if G[v]["Label_Source"] == 0
      3     else '#33a02c' for v in G]

~/opt/anaconda3/lib/python3.8/site-packages/networkx/classes/coreviews.py in __getitem__(self, key)
     49 
     50     def __getitem__(self, key):
---> 51         return self._atlas[key]
     52 
     53     def copy(self):

KeyError: 'Label_Source'

---

解决方案

尝试：

node_color = np.where(df['Label_Source'] == 0, '#1f78b4', '#33a02c').tolist()
print(node_color)

# Output:
['#33a02c', '#33a02c', '#33a02c']

或使用networkx：

node_color = ['#1f78b4' if v == 0 else '#33a02c'
                  for v in nx.get_edge_attributes(G, 'Label_Source')]
print(node_color)

# Output:
['#33a02c', '#33a02c', '#33a02c']