python - 值错误 : Number of labels=34866 does not match number of samples=2

Question

我正在尝试运行决策树分类器，但我遇到了这个问题。你能解释一下我该如何解决这个错误吗？我的英语不是很好，但我会努力理解!我才刚刚开始学习程序，所以如果有什么不够好的地方请指出我。谢谢!

import matplotlib.pyplot as plt
from sklearn.tree import DecisionTreeClassifier 
from sklearn import tree
import pandas as pd

sale=pd.read_csv('Online_Sale.csv')
plt.rcParams['font.sans-serif'] = ['Microsoft JhengHei']
sale['回購'] = sale['回購'].apply(lambda x: 1 if x == 'Y' else 0)
sale['單位售價'] = sale['單位售價'].str.replace(',', '').astype(float)
x=sale['年紀'],sale['單位售價']
y=sale['回購']

print(x)
print(y)

clf = DecisionTreeClassifier(random_state=0)
model = clf.fit(x, y)

text_representation = tree.export_text(clf)
print(text_representation)

fig = plt.figure(figsize=(15,12))
tree.plot_tree(clf, 
              filled=True)
fig.savefig("decistion_tree.png")

数据:

我寻找了很多不同的方法，但我没有办法完全理解问题所在......

Answer 1

只有一个小错误:

x=sale['年紀'],sale['單位售價']

这不是选择您想要的列，而是创建列的元组，因此错误消息的结尾 ... does not match number of samples=2

使用所选列创建新的 pd.DataFrame 的一种方法:

x=sale[['年紀', '單位售價']]