e31fb50e9d0af2e42118474f8042751a6c5692f3,examples/sandwich.py,,plot_neighborhood_graph,#,86

Before Change

def plot_neighborhood_graph(x, nn, y, axis=pyplot, cols="rbgmky"):
  for i in xrange(x.shape[0]):
    xs = [x[i,0], x[nn[i,1], 0]]
    ys = [x[i,1], x[nn[i,1], 1]]
    axis.plot(xs, ys, cols[y[i]])


if __name__ == "__main__":
  sandwich_demo()

After Change

def plot_neighborhood_graph(x, nn, y, axis=plt, colors="rbgmky"):
  for i, a in enumerate(x):
    b = x[nn[i,1]]
    axis.plot((a[0], b[0]), (a[1], b[1]), colors[y[i]])


if __name__ == "__main__":
  sandwich_demo()

Instances