2017-09-19 2 views
0

Kann mir jemand helfen, mir zu sagen, wie die Wahrscheinlichkeitsdaten aus den Ergebnissen auf Predict_proba SVM konvertieren. Die Ergebnisse sehen aus wie im Beispiel unten, aber ich möchte eine Liste oder ein Array, das nur die zweite Spalte enthält und frei von Klammern ist.Extrahieren Sie die Wahrscheinlichkeit von predict_proba in SVM Python

[[ 0.15941701 0.84058299] 
[ 0.12190033 0.87809967] 
[ 0.06293788 0.93706212] 
... 
[ 0.93175738 0.06824262]] 
<class 'numpy.ndarray'> 

Vielen Dank.

+0

Sehen Sie diese Antwort gibt: [Aufschneiden numpy Spalten] (https://stackoverflow.com/questions/4455076/how-to-access-the-ith -Spalte eines mehrzeiligen multidimensionalen Arrays. Extrahieren der zweiten Spalte = a [: 1]. – blackplant

Antwort

0

Sie müssen nur die numpy Indizierung für mehrdimensionale Arrays verwenden. In diesem Fall möchten Sie die gesamte erste Dimension (Zeilen) und nur das zweite Element in der zweiten Dimension (zweite Spalte). So verwenden Sie [:, 1]

a = np.array([ 
    [0.15941701, 0.84058299], 
    [0.12190033, 0.87809967], 
    [0.06293788, 0.93706212]]) 

a[:,1] 

Welche

[ 0.84058299 0.87809967 0.93706212] 
+0

Danke Jungs. Es ist eigentlich eine einfache Aufgabe. –

Verwandte Themen