主页 / 问题 / 1200557
Accepted
Yuriy Tigiev
Asked:2021-11-06 03:09:43 +0000 UTC2021-11-06 03:09:43 +0000 UTC

seaborn如何将DataFrame值显示为阴影表？（海运）

  • 772

seaborn如何将DataFrame值显示为阴影表？

x 轴显示来自 DataFrame.values 范围的 c1、c2、c3 列的可能值

y 轴以降序显示记录的索引号 df.index

对于每一行（记录），其值在列 c1、c2、c3 中给出的正方形被涂上。

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


sns.set()

d = {5:{"c1":1, "c2":7, "c3":10}, 4:{"c1":1, "c2":6, "c3":8}, 3:{"c1":4, "c2":7, "c3":9}}
df = pd.DataFrame(d).T


sns.relplot(data=df.T)
plt.show()

print(df)


   c1  c2  c3
5   1   7  10
4   1   6   8
3   4   7   9

预期结果。除了颜色之外，表格的格式必须与示例中的相同。

在此处输入图像描述

python

2 个回答

  1. 我想知道是否可以编写一个单行代码，将df问题中给出的框架转换为：

       1   2   3   4   5   6   7   8   9   10
3   0   0   0   1   0   0   1   0   1   0
4   1   0   0   0   0   1   0   1   0   0
5   1   0   0   0   0   0   1   0   0   1

    解决方案：

    r = (df
     .stack()
     .reset_index(name="col")
     .assign(x=1)
     .pivot_table(index="level_0", columns="col", values="x")
     .reindex(range(1,11), axis=1)
     .fillna(0)
     .astype("int8")
     .rename_axis(None)
     .rename_axis(None, axis=1))

    结果：

    In [204]: r
Out[204]: 
   1   2   3   4   5   6   7   8   9   10
3   0   0   0   1   0   0   1   0   1   0
4   1   0   0   0   0   1   0   1   0   0
5   1   0   0   0   0   0   1   0   0   1

    PS @stawdog 已经展示了如何绘制所需的表格。

    • 7
  2. Best Answer

    我已经很久没有这么变态了，但是哦，好吧：

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

sns.set(rc={'figure.facecolor':'white'})

d = {5:{"c1":1, "c2":7, "c3":10}, 4:{"c1":1, "c2":6, "c3":8}, 3:{"c1":4, "c2":7, "c3":9}}

idx = max([max(val.values()) for key, val in d.items()])
cols = [5, 4, 3]
df = pd.DataFrame(index = range(1, idx+1), columns=cols).T.fillna(0)

for k, v in d.items():
    for x in v.keys():
        df.loc[k, v[x]] = 1


sns.heatmap(df, cmap="Blues", cbar=False, square = True, linewidths=1, linecolor="k")
plt.tick_params(labelbottom=False, labeltop=True)
plt.show()

    在此处输入图像描述

    更新

    问题的作者要求为“从 1 到 5 的列......淡绿色和从 6 到 10 淡黄色的列”另外为单元格着色

    为此，将相应的值放入每个单元格并构建热图就足够了，之前已经设置了自己的颜色图：

    my_cmap = ["lightgreen", "darkblue", "lightyellow"]
df[df.iloc[:,5:]==0]=2
sns.heatmap(df, cmap=my_cmap, cbar=False, square = True, linewidths=1, linecolor="k")
plt.tick_params(labelbottom=False, labeltop=True)
plt.show()

    在此处输入图像描述

    • 5

相关问题