Skip to content

Instantly share code, notes, and snippets.

@aialenti

aialenti/dbscan.py

Last active February 17, 2021 14:18
Show Gist options
  • Save aialenti/ed0af24821dd08caab9c728b5e06adf8 to your computer and use it in GitHub Desktop.
Save aialenti/ed0af24821dd08caab9c728b5e06adf8 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
dbscan.py
import pandas as pd
from sklearn.cluster import DBSCAN
data = pd.read_csv("data.csv")
data["labels"] = DBSCAN(eps=0.01, min_samples=3).fit(data[["lat","lng"]].values).labels_
print(data["labels"].unique())
data.to_csv("out.csv")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment