Mean shift
Mean shift is another clustering algorithm that doesn't require an estimate for the number of clusters. It has been successfully applied to image processing. The algorithm tries to iteratively find the maxima of a density function. Before demonstrating mean shift, we will average the rain data on a day-of-the-year basis using a Pandas DataFrame
. Create the DataFrame
and average its data as follows:
df = pd.DataFrame.from_records(x.T, columns=['dates', 'rain']) df = df.groupby('dates').mean() df.plot()
The following plot is the result:
Cluster the data with the mean shift algorithm as follows:
x = np.vstack((np.arange(1, len(df) + 1) , df.as_matrix().ravel())) x = x.T ms = cluster.MeanShift() ms.fit(x) labels = ms.predict(x)
If we visualize the data with different line widths and shading for the three resulting clusters, the following figure is obtained:
As you can see, we have three clusters based on the average rainfall in mm on...