如何使用scikit Learn确保测试集和训练集具有相同的功能？

Question

我正在尝试预测再次出现的票证的看不见的数据？我有很多分类变量。这些变量可以相同或重复。我如何确保功能相同？功能可能会因传入音量而异？

y=predict_pi.P_I
X=predict_pi.drop('P_I',1)
X=pd.get_dummies(X)
print(X)
X_train,X_test,y_train,y_test=train_test_split(X,y,test_size=0.3,random_state=42)

logReg=LogisticRegression()
logReg.fit(X_train,y_train)
pred_pi=logReg.predict(X_test)
print(pred_pi)
print(accuracy_score(y_test,pred_pi))
test=pd.get_dummies(test)
print(test)
pred_test=logReg.predict(test)

I get the error: ValueError: X has 159 features per sample; expecting 161

这两列是具有分类值的列，而功能取决于它们。功能可以增加或减少。

AUDITOR REASON_CODE 

任何建议都会有所帮助。

火车和测试数据集中的行和列数如下。

train - (599939, 9) test - (130314, 8)

培训数据为6个月，测试数据为前一个月。问候，仁。

Answer 1

extra_cols_train = [i for i in list(test) if i not in list(X)]
extra_cols_test = [i for i in list(X) if i not in list(test)]

X = X.reindex(columns=X.columns.tolist() + extra_cols_train)
X[extra_cols_train] = 0

test = test.reindex(columns=test.columns.tolist() + extra_cols_test)
test[extra_cols_test] = 0