我希望使用
matplotlib
绘制带有分类行和列标签的pandas DataFrame的散点图。一个示例DataFrame如下所示:import pandas as pd
df = pd.DataFrame({"a": [1,2], "b": [3,4]}, index=["c","d"])
# a b
#c 1 2
#d 3 4
标记大小是各自数据框架值的函数。到目前为止,我想出了一个笨拙的解决方案,基本上是枚举行和列,绘制数据,然后重构标签:
flat = df.reset_index(drop=True).T.reset_index(drop=True).T.stack().reset_index()
# level_0 level_1 0
#0 0 0 1
#1 0 1 2
#2 1 0 3
#3 1 1 4
flat.plot(kind='scatter', x='level_0', y='level_1', s=100*flat[0])
plt.xticks(range(df.shape[1]), df.columns)
plt.yticks(range(df.shape[0]), df.index)
plt.show()
哪种工作。
![哪种工作](https://istack.dev59.com/e22Ep.webp)
现在的问题是:有没有更直观、更集成的方法来生成这个散点图,最好不需要分离数据和元数据?