How to plot the outline of the outer edges on a Matplotlib line in Python?
The objects in LineCollection do not have distinct edgecolor and facecolor. By trying to set the linestyle, you are affecting the style of the entire line segment. I found it easier to create the desired effect by using a series of patches. Each patch represents an edge of the graph. The edgecolor, linestyle, linewidth, and facecolor of the patches can be manipulated individually. The trick is building a function to convert an edge into a rotated Rectangle patch.
import matplotlib.path as mpath
import matplotlib.patches as mpatches
import numpy as np
from matplotlib import pyplot as plt
import networkx as nx
G = nx.Graph()
for i in range(10):
G.add_node(i)
for i in range(9):
G.add_edge(9, i)
# make a square figure so the rectangles look nice
plt.figure(figsize=(10,10))
plt.xlim(-1.1, 1.1)
plt.ylim(-1.1, 1.1)
def create_patch(startx, starty, stopx, stopy, width, w=.1):
# Check if lower right corner is specified.
direction = 1
if startx > stopx:
direction = -1
length = np.sqrt((stopy-starty)**2 + (stopx-startx)**2)
theta = np.arctan((stopy-starty)/(stopx-startx))
complement = np.pi/2 - theta
patch = mpatches.Rectangle(
(startx+np.cos(complement)*width/2, starty-np.sin(complement)*width/2),
direction * length,
width,
angle=180/np.pi*theta,
facecolor='#000000',
linestyle=':',
linewidth=width*10,
edgecolor='k',
alpha=.3
)
return patch
# Create layout before building edge patches
pos = nx.circular_layout(G)
for i, edge in enumerate(G.edges()):
startx, starty = pos[edge[0]]
stopx, stopy = pos[edge[1]]
plt.gca().add_patch(create_patch(startx, starty, stopx, stopy, (i+1)/10))
plt.show()
In your example, you noticed that we can use the X and Y positions of the edges to find the angle of rotation. We use the same trick here. Notice also that sometimes the magnitude of the rectangle length is negative. The Rectangle Patch assumes that the x and y inputs refer to the lower left corner of the rectangle. We run a quick check to make sure that's true. If false, we've specified the top first. In that case, we draw the rectangle backwards along the same angle.
Another gotcha: it's important to run your layout algorithm before creating the patches. Once the pos
is specified, we can use the edges to look up the start and stop locations.
Opportunity for Improvement: Rather than plotting each patch as you generate it, you can use a PatchCollection and manipulate the patches in bulk. The docs claim that PatchCollection is faster, but it may not fit all use cases. Since you expressed a desire to set properties on each patch independently, the collection might not offer the flexibility you need.
The problem of surrounding a line with a certain width by another line is that the line is defined in data coordinates, while the linewidth is in a physical unit, namely points. This is in general desireable, because it allows to have the linewidth to be independent of the data range, zooming level etc. It also ensures that the end of the line is always perpendicular to the line, independent of the axes aspect.
So the outline of the line is always in a mixed coordinate system and the final appearance is not determined before drawing the actual line with the renderer. So for a solution that takes the (possibly changing) coordinates into account, one would need to determine the outline for the current state of the figure.
One option is to use a new artist, which takes the existing LineCollection
as input and creates new transforms depending on the current position of the lines in pixel space.
In the following I chose a PatchCollection
. Starting off with a rectangle, we can scale and rotate it and then translate it to the position of the original line.
import numpy as np
import matplotlib.pyplot as plt
from matplotlib.collections import LineCollection, PatchCollection
import matplotlib.transforms as mtrans
class OutlineCollection(PatchCollection):
def __init__(self, linecollection, ax=None, **kwargs):
self.ax = ax or plt.gca()
self.lc = linecollection
assert np.all(np.array(self.lc.get_segments()).shape[1:] == np.array((2,2)))
rect = plt.Rectangle((-.5, -.5), width=1, height=1)
super().__init__((rect,), **kwargs)
self.set_transform(mtrans.IdentityTransform())
self.set_offsets(np.zeros((len(self.lc.get_segments()),2)))
self.ax.add_collection(self)
def draw(self, renderer):
segs = self.lc.get_segments()
n = len(segs)
factor = 72/self.ax.figure.dpi
lws = self.lc.get_linewidth()
if len(lws) <= 1:
lws = lws*np.ones(n)
transforms = []
for i, (lw, seg) in enumerate(zip(lws, segs)):
X = self.lc.get_transform().transform(seg)
mean = X.mean(axis=0)
angle = np.arctan2(*np.squeeze(np.diff(X, axis=0))[::-1])
length = np.sqrt(np.sum(np.diff(X, axis=0)**2))
trans = mtrans.Affine2D().scale(length,lw/factor).rotate(angle).translate(*mean)
transforms.append(trans.get_matrix())
self._transforms = transforms
super().draw(renderer)
Note how the actual transforms are only calculated at draw
time. This ensures that they take the actual positions in pixel space into account.
Usage could look like
verts = np.array([[[5,10],[5,5]], [[5,5],[8,2]], [[5,5],[1,4]], [[1,4],[2,0]]])
plt.rcParams["axes.xmargin"] = 0.1
fig, (ax1, ax2) = plt.subplots(ncols=2, sharex=True, sharey=True)
lc1 = LineCollection(verts, color="k", alpha=0.5, linewidth=20)
ax1.add_collection(lc1)
olc1 = OutlineCollection(lc1, ax=ax1, linewidth=2,
linestyle=":", edgecolor="black", facecolor="none")
lc2 = LineCollection(verts, color="k", alpha=0.3, linewidth=(10,20,40,15))
ax2.add_collection(lc2)
olc2 = OutlineCollection(lc2, ax=ax2, linewidth=3,
linestyle="--", edgecolors=["r", "b", "gold", "indigo"],
facecolor="none")
for ax in (ax1,ax2):
ax.autoscale()
plt.show()
Now of course the idea is to use the linecollection
object from the question instead of the lc1
object from the above. This should be easy enough to replace in the code.