时间频率颜色图

Question

我想在颜色图中显示某点的频率，即（1,2）的频率为3点，同时仍保持我的“ xaxis”（即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()