时频颜色映射

Question

我想在彩色地图中显示一个点的出现频率，即(1,2)有3个点的频率，同时仍然保持我的‘x轴’(即df'A')

import pandas as pd
import matplotlib.pyplot as plt
df = pd.DataFrame({'A': [1,1,1,1,2,2,3,4,6,7,7],
                   'B': [2,2,2,3,3,4,5,6,7,8,8]})
plt.figure()
plt.scatter(df['A'], df['B'])
plt.show()

这是我当前的图

​

​

我想要保持相同的轴，同时添加颜色映射。希望我说得够清楚了。

Answer 1

您可以使用collections包来计算某个值的频率。

freq_dic = collections.Counter(df["B"])

然后，您需要将这个新列表添加到数据帧中，并将两个新选项添加到散点图中。颜色映射表图例将与plt.colorbar一起显示。这段代码并不完美，因此欢迎进一步的改进。

import pandas as pd
import matplotlib.pyplot as plt
import collections


df = pd.DataFrame({'A': [1,1,1,1,2,2,3,4,6,7,7],
                   'B': [2,2,2,3,3,4,5,6,7,8,8]})

freq_dic = collections.Counter(df["B"])


for index, entry in enumerate(df["B"]):
    df.at[index, 'freq'] = (freq_dic[entry])


plt.figure()
plt.scatter(df['A'], df['B'],
            c=df['freq'],
            cmap='viridis')
plt.colorbar()
plt.show()

​

​