seaborn绘制heatmap

代码如下，有用的是前2行和后3行

import matplotlib.pyplot as plt
import numpy as np
import pandas as pd
import seaborn as sns
sns.set()

data=pd.read_csv('falling/CSI_12_1_22_19_34_42.csv', sep=';', parse_dates=True)
data=data.loc[:,data.columns.str.contains('a')].drop(labels=['frame_control','timestamp'],axis=1)[0:200]

data[data>0.5] = 0.5 #做一个bar的操作
data[data<0.1] = 0.1

#dB化
data = data.applymap(lambda x: 10*np.log(x))

ax = sns.heatmap(data.T,cmap="rainbow")
ax.invert_yaxis()#y轴翻转
plt.show()

修改配色
函数详解