2016-11-21 87 views
-1

train.txt有3列。第一列和第二列保存在A中,但我不能將其用於fit_transform。爲什麼?請幫我爲什麼我的數組不適用於fit_traansform()?我能做什麼?

from macpath import split 
from sklearn.feature_extraction import DictVectorizer 

A=[] 
B=[] 
C=[] 
D=[] 

vec = DictVectorizer() 

with open("train.txt") as f: 
    f1=[x.strip() for x in f if x.strip()] 
    for x in f1[0:]: 
     data=[tuple(x.split())] 
     for x in data: 
      A.append(x[0]+" "+x[1]) 
      B.append(x[2]) 

X=vec.fit_transform(A) 

enter image description here

+0

請附上您的代碼,直接在你的問題,並使用相應的格式化選項來渲染它正確。 – Gormador

回答

0

你必須改變以下部分 A.append(x[0]+" "+x[1]) 本:

A.append({x[0],x[1]: 1 }) 
+0

它工作:)謝謝 –

相關問題