时间频率颜色图

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()