scikit-learn包中DBSCAN函数的输入格式

Question

Compute DBSCAN

db = DBSCAN(eps=0.3, min_samples=10).fit(X)

以上是使用scikit-learn包计算DBSCAN的示例代码。我自己的输入格式是这样的:[[37.9358， -122.3478]， [33.8312， -117.6053] .

示例输入格式如下:[[0.84022039 1.14802236][-1.15474834 -1.2041171]]

我创建自己的格式的方法是使用追加操作构造list的list。样例格式的结构是什么?如何构造它?我遇到的错误信息是:TypeError: list索引必须是整数，而不是list

谢谢!

Answer 1

这是一个bug:列表的列表应该被DBSCAN.fit接受，但它不是，所以你必须使用NumPy数组(*):

X = np.array([[37.9358, -122.3478], [33.8312, -117.6053]])
db = DBSCAN(eps=0.3, min_samples=10).fit(X)

就可以了。

(*) NumPy数组是首选的数据格式;为方便起见，可以接受列表的列表，但会在内部进行转换。