介绍

根据数据大小值生成不同色块的颜色值。

代码

`
import numpy as np
import matplotlib.pyplot as plt
import pandas as pd

读取数据文件

excel_path = "xx.csv"
data = pd.read_csv(excel_path)

取result

data_12 = data.iloc[:,3]

取targetID

data_y = data.iloc[:,1]

取inputID

data_x = data.iloc[:,0]

初始化矩阵

fdata = np.zeros((70,199))
i = 0
m = 0
for j in range(0,len(data_x)):

#防止13930数组越界
if  (len(data_x)-2) == j :
    #  print("==xxx==",data_x[j],data_x[j+1])
     fdata[i][m]= data_12[j]
    #  print("m",m)

     break
#判断是同一个x坐标
elif data_x[j] == data_x[j+1]:
     fdata[i][m] = data_12[j]
#判断是一个新x坐标
elif data_x[j] != data_x[j+1]:
     i= i+1
     m = 0
     fdata[i][m] = data_12[j]

# print("im",i,m ,fdata[i][m])

m = m+1 

核对数据

print(fdata[0])

print(fdata[1])

print(fdata[2])

print(fdata[69])

plt.matshow(fdata, cmap=plt.get_cmap('RdBu_r'),alpha=0.9) # , alpha=0.3

颜色条

plt.colorbar()
plt.show()

`

图

Snipaste_2021-04-22_16-11-09.png