pandas读取csv并绘制散点图

读取

loc='checkpoints/mnist_vae2/mu_and_sigma.csv'
import pandas as pd
import numpy as np
X=pd.DataFrame.from_csv(loc)
#X = X.sort_index(by='label')  按照label排序
X = X.values #转换成熟悉的numpy

绘制

import matplotlib.pyplot as plt
plt.scatter(X[890000:, 1], X[890000:, 2], c=X[890000:,5], marker = 'o', s=40, cmap=plt.cm.Spectral)#c--color，s--size,marker点的形状,890000是指对从890000行起的数据绘制散点图；c选项不用明确指定颜色只用区分不同颜色对应的值
plt.show()