gpt4 book ai didi

python - 可视化 pandas/numpy 中的共现矩阵

转载 作者:行者123 更新时间:2023-11-28 18:30:49 25 4
gpt4 key购买 nike

我计算了一个数组,其中有多少百分比的排列包含所有不同的对组合。现在我想将其可视化为例如热图。我有以下计算代码:

#occurrences matrix
a = np.array(np.array_split(np.random.binomial(1,.5,30),10), dtype='f')

#co-occurrences matrix
acov=np.dot(a.T, a)
acov[np.diag_indices_from(acov)]=0
acov /= acov.sum()

然后我尝试这个可视化:

plt.imshow(acov,interpolation='nearest')
plt.colorbar()
plt.show()

但是我真的不知道我在做什么,我第一次使用热图所以不确定我看到了什么。我想要的是一个与数组形状相同的图形,其中每个方 block 都有一个颜色强度,代表它在 0-1 范围内的大小。

最佳答案

这个呢?

plt.imshow(acov,interpolation='nearest', cmap='Reds')
plt.colorbar()
plt.show()

enter image description here

使用 seaborn.heatmap() :

sns.heatmap(acov)

enter image description here

关于python - 可视化 pandas/numpy 中的共现矩阵,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/37631649/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com