plot a vertical line if the date is a weekend in matplotlbi in python

main.py
import matplotlib.pyplot as plt
from datetime import datetime

# Generate a list of dates
dates = ['2021-10-01', '2021-10-02', '2021-10-03', '2021-10-04', '2021-10-05', '2021-10-06']

# Convert string dates to datetime objects
dates = [datetime.strptime(date, '%Y-%m-%d') for date in dates]

# Plot the dates on x-axis
plt.plot(dates, [0]*len(dates))

# Iterate over the dates
for date in dates:
    # Check if the date is a weekend (Saturday or Sunday)
    if date.isoweekday() in [6, 7]:
        # Plot a vertical line at x-coordinate of the date
        plt.axvline(x=date, color='r', linestyle='--')

# Display the plot
plt.show()
635 chars
22 lines