我想使用matplotlib
来生成带有分类行和列标签的pandas DataFrame的散点图。示例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
标记大小是各个DataFrame值的函数。到目前为止,我想出了一个尴尬的解决方案,本质上列举的行和列,绘制数据,然后重建标签:
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()
现在,问题:是否有更直观,更集成的方式来产生散点图,理想情况下不需要拆分数据和元数据?
,我不认为我们可以用非数值数据绘制。 AFAIK你将不得不单独设置刻度... – MaxU
我想这个问题转化为*“为什么没有库实现我的自定义绘图的愿望功能呢?”*。 – ImportanceOfBeingErnest