python - 如何为特定集群运行线性回归

Question

我使用了一个 csv 文件，可以在这里找到https://www.kaggle.com/vjchoudhary7/customer-segmentation-tutorial-in-python，并使用 4 个集群运行 kmeans。我制作了一个 dict 列表来查看哪些观察在哪些集群中，结果是一个 array(array 0,1,2,3) 。现在我想对数组 0 中的观察值运行线性回归（以及随后的多项式和指数）。我该怎么做呢？

代码：

 import matplotlib.pyplot as plt
import pandas as pd




ore')
plt.show()

标签： python