微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

python – pandas和seaborn – 没有颜色的热图

我一直在使用seaborn及其热图功能.我想构建一个带有来自pandas dataframe df的带注释值的矩阵:

C,L,N
a,x,10
a,y,2
a,z,4
b,x,1
b,y,22
b,z,11
c,x,3
c,y,1
c,z,0

到目前为止,它适用于:

# Read DataBase
df = pd.read_csv('myfile.csv')

# Save Users/City/Language Matrix
pdf = df.pivot(index='C',columns='L',values='N').fillna(0)

# Set Font Parameters
rc = {'font.size': 8, 'xtick.labelsize': 11, 'ytick.labelsize': 11}

# Set figure
fig = plt.figure(figsize=(20, 9))

# Assign to Seaborn
sns.set(rc=rc)

with sns.axes_style('white'):

    sns.heatmap(pdf,
                cbar=False,
                square=False,
                annot=True,
                cmap='Blues',
                fmt='g',
                linewidths=0.5)

哪个回报:

enter image description here

最后,我有兴趣只保留值并将结构保存为一个简单的表格,丢弃颜色.我尝试设置cmap = None但它不起作用,如果没有cmap,seaborn会为热图分配一个认的cmap.

解决方法:

如果我不理解你的错误,并且你想要的只是忽略色彩映射,你可以使用matplotlib的ListedColormap创建你自己选择的背景颜色的自定义色彩映射:

from matplotlib.colors import ListedColormap

with sns.axes_style('white'):
    sns.heatmap(pdf,
                cbar=False,
                square=False,
                annot=True,
                fmt='g',
                cmap=ListedColormap(['white']),
                linewidths=0.5)

哪个会产生:

enter image description here

将字符串白色替换为STR,HEX或RGB颜色,以设置您选择的背景颜色.

但是,我相信大熊猫提供更好的出口选择.在下面找到所有可能的导出选项的屏幕:

enter image description here

根据您要插入表格的位置,也许to_html,to_json或to_latex比使用seaborn绘图更好.

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。

相关推荐