如何绘制透明度随数组数据动态变化的x-y折线？

Hey there! I see you're trying to make a line plot where the transparency changes along the line—great idea! The problem with your current code is that Matplotlib's plot() method doesn't accept an array for the alpha parameter; it only takes a single scalar value. That's why it works when you use a fixed alpha but breaks when you pass Bb[:,2].

Luckily, there are two solid ways to achieve what you want:

Method 1: Use LineCollection (Recommended for Large Datasets) #

This is the most efficient approach, especially if you have a lot of data points. LineCollection lets you create a collection of line segments, each with its own transparency setting.

Here's how to implement it with your data:

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.collections import LineCollection

# Assuming Bb is your 2D array with shape (N, 3)
x = Bb[:, 0]
y = Bb[:, 1]
alphas = Bb[:, 2]

# Create pairs of adjacent points to form line segments
points = np.column_stack((x, y))
segments = np.array([points[i:i+2] for i in range(len(points)-1)])

# Create the LineCollection with red color and varying alpha
lc = LineCollection(segments, colors='r', alpha=alphas[:-1])
# Note: alphas[:-1] because we have N-1 segments for N points

# Plot the collection and set equal aspect ratio
fig, ax = plt.subplots()
ax.add_collection(lc)
ax.set_aspect('equal')

# Make sure the axes cover all your data (LineCollection doesn't auto-scale)
ax.set_xlim(x.min(), x.max())
ax.set_ylim(y.min(), y.max())

plt.show()

Method 2: Loop Through Segments (Simple for Small Datasets) #

If you're working with a small number of points, you can loop through each adjacent pair of points and plot each tiny segment with its own alpha value. This is more straightforward but less efficient for large datasets.

Example code:

import numpy as np
import matplotlib.pyplot as plt

fig, ax = plt.subplots()

# Iterate over each adjacent point pair
for i in range(len(Bb) - 1):
    # Extract the x and y values for the current segment
    x_seg = Bb[i:i+2, 0]
    y_seg = Bb[i:i+2, 1]
    # Use the alpha value from either the start or end point (your choice!)
    ax.plot(x_seg, y_seg, 'r', alpha=Bb[i, 2])

ax.set_aspect('equal')
plt.show()

Either of these methods should give you the varying transparency line you're looking for. The LineCollection method is definitely the way to go if you have a lot of data points, as it's much faster than plotting hundreds of tiny segments individually.

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