如何使用python确定两个图中的峰值并绘制两个图中仅有峰值的图

import numpy as np
import matplotlib.pyplot as plt
from scipy.signal import find_peaks

data1 = np.array([1, 2, 3, 2, 1, 4, 5, 2, 1])
data2 = np.array([2, 1, 4, 3, 2, 1, 2, 3, 2])

peaks1, _ = find_peaks(data1)
peaks2, _ = find_peaks(data2)

plt.subplot(3, 1, 1)
plt.plot(data1)
plt.title('Graph 1')

plt.subplot(3, 1, 2)
plt.plot(data2)
plt.title('Graph 2')

plt.subplot(3, 1, 3)
plt.plot(data1)
plt.plot(peaks1, data1[peaks1], 'ro')
plt.plot(data2)
plt.plot(peaks2, data2[peaks2], 'bo')
plt.title('Peaks Only')

plt.tight_layout()
plt.show()