pandas读取csv并绘制散点图
程序员文章站
2024-01-19 08:09:04
...
读取
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()
上一篇： vue h5移动端禁止缩放