用python写一个ai代码

好的，以下是一个简单的Python代码，它使用了Scikit-learn库中的K近邻算法来实现简单的分类AI：

from sklearn.neighbors import KNeighborsClassifier

# 训练数据
X_train = [[1, 1], [2, 2], [3, 3], [4, 4]]
y_train = [0, 0, 1, 1]

# 创建K近邻分类器
clf = KNeighborsClassifier(n_neighbors=3)

# 训练分类器
clf.fit(X_train, y_train)

# 预测新数据
X_new = [[0.5, 0.5], [5, 5]]
y_pred = clf.predict(X_new)

# 输出预测结果
print(y_pred)

这段代码首先导入了Scikit-learn库中的KNeighborsClassifier类，然后定义了训练数据X_train和y_train。接下来，创建了一个K近邻分类器对象clf，并使用fit()方法训练分类器。最后，使用预测数据X_new调用predict()方法进行预测，并输出预测结果y_pred。