Matplotlib中如何调整x时间列表匹配y相位嵌套列表以正确绘图

Hey there, I get exactly what you're trying to do here—you want to plot each nested segment of your phase data as a separate line (no connecting lines between segments), which means your flat time list needs to be split into a nested structure that matches your phase list. Let's walk through how to do this properly, even when the total length and segment lengths change dynamically.

Step 1: Split the Time List to Match Phase's Nested Structure #

First, we need to break your flat x (time) list into chunks where each chunk's length matches the corresponding sublist in y (phase). Here are two straightforward ways to do this:

Method 1: Basic Loop (Easy to Follow)

This approach uses a simple loop to track our position in the time list and slice out segments matching each phase sublist's length:

import matplotlib.pyplot as plt

# Your dynamic data (example values)
x = [0.0, 1.0, 2.0, 3.0, 4.0, 5.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0]
y = [[1.0, 1.0, 1.0, 1.0, 1.0, 1.0], [2.0, 2.0, 2.0, 2.0, 2.0, 2.0, 2.0, 2.0]]

# Get the length of each phase segment
segment_lengths = [len(seg) for seg in y]

# Split the time list into matching segments
x_split = []
current_position = 0
for length in segment_lengths:
    # Slice the time list from current position to current + segment length
    x_segment = x[current_position:current_position + length]
    x_split.append(x_segment)
    current_position += length

Method 2: Concise Itertools Approach

If you prefer a more compact solution, use itertools.accumulate to generate split indices automatically:

from itertools import accumulate

# Get segment lengths (same as above)
segment_lengths = [len(seg) for seg in y]

# Generate indices to split the time list
split_indices = [0] + list(accumulate(segment_lengths))

# Create the nested time list
x_split = [x[split_indices[i]:split_indices[i+1]] for i in range(len(split_indices)-1)]

Step 2: Plot the Segmented Data #

Now that x_split matches the structure of y, you can plot each segment individually. Matplotlib won't connect lines between separate plot() calls, so each phase segment will be independent:

plt.figure(figsize=(10, 6))

# Loop through each time-phase segment pair and plot
for time_seg, phase_seg in zip(x_split, y):
    plt.plot(time_seg, phase_seg, marker='o', linewidth=2)  # Marker helps visualize segments

plt.xlabel('Time')
plt.ylabel('Phase')
plt.title('Segmented Phase vs Time Plot')
plt.grid(True)
plt.show()

Bonus: Add a Data Validation Check #

To avoid errors if your time list length doesn't match the total length of phase segments, add a quick check at the start:

total_phase_length = sum(segment_lengths)
assert total_phase_length == len(x), f"Time list length ({len(x)}) doesn't match total phase length ({total_phase_length})!"

This will catch mismatches early and save you from debugging weird plotting issues.

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