Python - 从CSV绘制群集

Question

我是Python的初学者。我想从CSV数据中绘制一些二维簇。 CSV由x，y和标签组成。 CSV示例：

0.800717,0.84454,0
0.876446,0.626109,0
0.00504536,0.367249,1
0.0142114,0.794659,2
0.140504,0.240966,1

我正在尝试这样的东西，但它不能正常工作：

#!usr/bin/env

import numpy as np
import matplotlib.pyplot as plt
import csv

with open("kmeans.csv") as filein:
    reader = csv.reader(filein, skipinitialspace = True)
    xs, ys, label = zip(*reader)

plt.scatter(xs, ys)
plt.savefig("output.png")

当前输出示例：

预期的结果将是这样的（忽略质心）：

Answer 1

我建议你使用熊猫，而不是csv阅读器：

import pandas as pd
df = pd.read_csv('kmeans.csv', header=None)
df.plot(kind='scatter', x=0, y=1, c=2, colormap='rainbow', colorbar=None,
        markeredgewidth=0.0) # No circle margin!