Indexerror: zu viele Indizes für Array für

Question

Scikit-Learn Wenn ich

import numpy as np 
from sklearn.cluster import MeanShift, estimate_bandwidth 
estimate_bandwidth(np.array([1,2,3,4,5,6])) 

laufen bekomme ich die Fehler

Traceback (most recent call last): 
    File "<input>", line 1, in <module> 
    File "[..]/anaconda2/envs/reifen/lib/python2.7/site-packages/sklearn/cluster/mean_shift_.py", line 72, in estimate_bandwidth 
    d, _ = nbrs.kneighbors(X[batch, :], return_distance=True) 
IndexError: too many indices for array 

Ich verstehe, dass die Dimensionalität des Arrays nicht richtig ist - aber ich bin nicht sicher, was ich hier falsch gemacht habe - der Code ist nicht so lang. Kann schedule_bandwidth nur mit multivariaten Daten verwendet werden?

Answer 1

Ypu sollte so etwas tun:

import numpy as np 
from sklearn.cluster import MeanShift, estimate_bandwidth 

array = np.array([1,2,3,4,5,6]) 

d = estimate_bandwidth(array.reshape(len(array),1))