data_treatment #14
Merged
aurian.jault
merged 5 commits from data_treatment
into master
10 months ago
@ -1,12 +1,23 @@
|
|||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python3
|
||||||
|
|
||||||
import sys
|
import sys
|
||||||
sys.path.append('./src/back/')
|
sys.path.append('./src/back/')
|
||||||
|
|
||||||
import load_csv as l
|
import load_csv as l
|
||||||
|
import show_csv as s
|
||||||
|
import clustering_csv as c
|
||||||
|
|
||||||
df = l.return_csv("./data.csv")
|
df = l.return_csv("./data.csv")
|
||||||
l.csv_value(df)
|
l.csv_value(df)
|
||||||
|
|
||||||
l.csv_value(df)
|
l.csv_value(df)
|
||||||
|
|
||||||
l.csv_stadadisation_Z(df,"Vehicle Year")
|
# l.csv_standardisation_Z(df,"Vehicle Year")
|
||||||
|
|
||||||
|
# l.csv_robust_normalize(df,"Speed Limit")
|
||||||
|
|
||||||
|
# s.histo_col(df,"Speed Limit")
|
||||||
|
|
||||||
|
# s.plotBoxWhisker(df)
|
||||||
|
|
||||||
|
c.launch_cluster(df,['Speed Limit','Vehicle Year'])
|
||||||
|
@ -0,0 +1,83 @@
|
|||||||
|
import numpy as np
|
||||||
|
import matplotlib.pyplot as plt
|
||||||
|
from sklearn.cluster import KMeans, DBSCAN
|
||||||
|
from sklearn.datasets import make_blobs, make_moons
|
||||||
|
from mpl_toolkits.mplot3d import Axes3D
|
||||||
|
|
||||||
|
def visualize_clusters_2d(X, labels, centers=None, title="Clusters"):
|
||||||
|
plt.figure(figsize=(10, 7))
|
||||||
|
plt.scatter(X[:, 0], X[:, 1], c=labels, s=50, cmap='viridis')
|
||||||
|
if centers is not None:
|
||||||
|
plt.scatter(centers[:, 0], centers[:, 1], c='red', s=200, alpha=0.75)
|
||||||
|
plt.title(title)
|
||||||
|
plt.xlabel("Feature 1")
|
||||||
|
plt.ylabel("Feature 2")
|
||||||
|
plt.show()
|
||||||
|
|
||||||
|
def visualize_clusters_3d(X, labels, centers=None, title="Clusters"):
|
||||||
|
fig = plt.figure(figsize=(10, 7))
|
||||||
|
ax = fig.add_subplot(111, projection='3d')
|
||||||
|
ax.scatter(X[:, 0], X[:, 1], X[:, 2], c=labels, s=50, cmap='viridis')
|
||||||
|
if centers is not None:
|
||||||
|
ax.scatter(centers[:, 0], centers[:, 1], centers[:, 2], c='red', s=200, alpha=0.75)
|
||||||
|
ax.set_title(title)
|
||||||
|
ax.set_xlabel("Feature 1")
|
||||||
|
ax.set_ylabel("Feature 2")
|
||||||
|
ax.set_zlabel("Feature 3")
|
||||||
|
plt.show()
|
||||||
|
|
||||||
|
def calculate_cluster_statistics_kmeans(X, labels, centers):
|
||||||
|
unique_labels = np.unique(labels)
|
||||||
|
stats = []
|
||||||
|
for label in unique_labels:
|
||||||
|
cluster_points = X[labels == label]
|
||||||
|
num_points = len(cluster_points)
|
||||||
|
center = centers[label]
|
||||||
|
stats.append({
|
||||||
|
'cluster': label,
|
||||||
|
'num_points': num_points,
|
||||||
|
'center': center
|
||||||
|
})
|
||||||
|
return stats
|
||||||
|
|
||||||
|
def calculate_cluster_statistics_dbscan(X, labels):
|
||||||
|
unique_labels = np.unique(labels)
|
||||||
|
stats = []
|
||||||
|
for label in unique_labels:
|
||||||
|
if label == -1:
|
||||||
|
continue # Ignore noise
|
||||||
|
cluster_points = X[labels == label]
|
||||||
|
num_points = len(cluster_points)
|
||||||
|
density = num_points / (np.max(cluster_points, axis=0) - np.min(cluster_points, axis=0)).prod()
|
||||||
|
stats.append({
|
||||||
|
'cluster': label,
|
||||||
|
'num_points': num_points,
|
||||||
|
'density': density
|
||||||
|
})
|
||||||
|
return stats
|
||||||
|
|
||||||
|
def launch_cluster(df,array_columns):
|
||||||
|
X = df[array_columns].values
|
||||||
|
|
||||||
|
kmeans = KMeans(n_clusters=4, random_state=42)
|
||||||
|
labels_kmeans = kmeans.fit_predict(X)
|
||||||
|
centers_kmeans = kmeans.cluster_centers_
|
||||||
|
|
||||||
|
stats_kmeans = calculate_cluster_statistics_kmeans(X, labels_kmeans, centers_kmeans)
|
||||||
|
# for stat in stats_kmeans:
|
||||||
|
# print(f"Cluster {stat['cluster']}: {stat['num_points']} points, Center: {stat['center']}")
|
||||||
|
|
||||||
|
# Appliquer DBSCAN
|
||||||
|
dbscan = DBSCAN(eps=0.2, min_samples=5)
|
||||||
|
labels_dbscan = dbscan.fit_predict(X)
|
||||||
|
stats_dbscan = calculate_cluster_statistics_dbscan(X, labels_dbscan)
|
||||||
|
# for stat in stats_dbscan:
|
||||||
|
# print(f"Cluster {stat['cluster']}: {stat['num_points']} points, Density: {stat['density']}")
|
||||||
|
if len(array_columns) == 3:
|
||||||
|
visualize_clusters_3d(X, labels_kmeans, centers_kmeans, title="K-Means Clustering 3D")
|
||||||
|
visualize_clusters_3d(X, labels_dbscan, title="DBSCAN Clustering 3D")
|
||||||
|
else:
|
||||||
|
visualize_clusters_2d(X, labels_kmeans, centers_kmeans, title="K-Means Clustering")
|
||||||
|
visualize_clusters_2d(X, labels_dbscan, title="DBSCAN Clustering")
|
||||||
|
return stats_kmeans,stats_dbscan
|
||||||
|
|
@ -0,0 +1,16 @@
|
|||||||
|
import pandas as pd
|
||||||
|
import numpy as np
|
||||||
|
import matplotlib.pyplot as plt
|
||||||
|
|
||||||
|
def histo_col(df,colonne):
|
||||||
|
plt.figure()
|
||||||
|
plt.hist(df[colonne], bins=int(df[colonne].nunique()/4), alpha=0.7, color='blue', edgecolor='black')
|
||||||
|
plt.title(f"Histogramme de la colonne '{colonne}'")
|
||||||
|
plt.xlabel(colonne)
|
||||||
|
plt.ylabel("Fréquence")
|
||||||
|
plt.grid(True)
|
||||||
|
plt.show()
|
||||||
|
|
||||||
|
def plotBoxWhisker(df):
|
||||||
|
df.plot(kind='box', subplots=True, sharex=False, sharey=False)
|
||||||
|
plt.show()
|
Loading…
Reference in new issue