csv文件读取、保存、画热力图

#(1)对csv文件进行读取
data_ = pd.read_csv('D:\Data\data_6.8_overlap.csv')
print(data_.head())
print(data_.describe())
#(2)对DataFrame数据转换为array
data_np = np.array(data_)
#(3)矩阵元素对应相乘
data = np.multiply(data_np,weight_np)
data = pd.DataFrame(data)
print(data.head(5))
#(4)数据转化为DataFrame保存为csv格式文件
data.to_csv('D:/Data/data.csv')
index=False保存时不加入列名
data.to_csv('./data_v1.csv', index=False)
#(5)读取时不读取index
data_v1 = pd.read_csv('./data_v1.csv', index_col=0)
读取后转化为array
data_v1 = np.array(data_v1)
#(6)保存为txt格式文件
np.savetxt('data_sdcn.txt',(data_v1))
加载读取
data_sdcn = np.loadtxt('./data_sdcn.txt')
#(6)画出热力图import seaborn as sns
data = pd.read_csv('./data_v.csv', index_col=0)# 读取数据
data = np.array(data)# 转化为array
#画图保存
plt.figure(dpi= 200)
sns.heatmap(data, cmap= 'Reds')
plt.savefig('./data_headmap.jpg')
plt.show()