Имея пролем с выходной формой KDTree.Непосредственно из документации (http://scikit -learn.org / stable / modules / generate / sklearn.neighbors.KDTree.html ), если запустить этот пример кода:
import numpy as np
np.random.seed(0)
X = np.random.random((10, 3)) # 10 points in 3 dimensions
tree = KDTree(X, leaf_size=2)
dist, ind = tree.query(X[:1], k=3)
print(ind) # indices of 3 closest neighbors
print(dist) # distances to 3 closest neighbors
документация говорит, что она должна вывести:
[0 3 1]
[0. 0.19662693 0.29473397]
Но если я запусту этот точный код в блокноте jupyter, мой вывод будет выглядеть такэто:
[[0 3 1]]
[[0. 0.19662693 0.29473397]]
С дополнительным набором скобок.Почему это?Это вызывает у меня некоторые головные боли при передаче результатов в список.
Заранее спасибо!