用自己的祖源成分数据与其他祖源成分数据做对比

最新推荐文章于 2024-11-04 22:26:34 发布

yhlhhhh

最新推荐文章于 2024-11-04 22:26:34 发布

阅读量141

点赞数

文章标签： python 数据可视化可视化 matplotlib pandas

本文链接：https://blog.csdn.net/yhlhhhhh/article/details/118156498

版权

这篇博客介绍了如何使用Python的pandas、numpy和matplotlib库来读取CSV文件，并生成祖源成分数据的热力图。通过设定x轴和y轴标签，展示不同祖源之间的对比。代码中设置了中文显示，并最终保存了图片，为读者提供了一种清晰的数据可视化方法。

摘要由CSDN通过智能技术生成

用自己的祖源成分数据与其他祖源成分数据做对比

概述

编程语言：python3.8
模块：pandas numpy matplotlib
可选：jupyter
整体思路：读取csv并生成热力图便于对比

步骤：见代码上注释

代码：

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
# 读取csv
df = pd.read_csv('e11.csv', index_col = 0, header = None)
# 设定x轴 y轴标签
ylabels = df.index
xlabels = ['非洲','欧洲','印度','马来','傣族','彝族','华东','日本','鄂伦春','雅库特','美洲']
# 选择所有数据
data = df.iloc[:,:]
# 设置支持中文画图
plt.rcParams['font.sans-serif'] = ['SimHei']
# matplotlib画热力图
fig = plt.figure(dpi=600)
ax = fig.add_subplot(1,1,1)
im = ax.imshow(data)
# 显示色块图例
plt.colorbar(im)
# 设定显示x轴 y轴标签
ax.set_xticks(np.arange(len(xlabels)))
ax.set_yticks(np.arange(len(ylabels)))
ax.set_xticklabels(xlabels)
ax.set_yticklabels(ylabels)
# 使x轴标签名向右倾斜45度以清晰显示
plt.setp(ax.get_xticklabels(), rotation=45, ha="right",
         rotation_mode="anchor")
# 每个色彩块上显示数值
for i in range(len(ylabels)):
    for j in range(len(xlabels)):
        text = ax.text(j, i, data.iloc[i, j],
                       ha="center", va="center", color="w")
# 设置标题
ax.set_title("E11对比图（%）")
# 保存图片
plt.savefig('heatmap.png')
# 显示图片
plt.show()