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

Question

计算 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 ]]

我创建自己格式的方法是用 append 操作构建一个列表的列表。请问示例格式的结构是什么样的，我该如何构建它？我遇到的错误信息是： TypeError: list indices must be integers, not list

谢谢！

Answer 1

这是一个错误：本来应该可以用一个列表的列表来给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数组是更推荐的数据格式；列表的列表虽然可以用，但内部会被转换成其他格式。