基于第三变量

Question

我试图用变量x相对于另一个y进行绘图，并根据另一个变量z的值添加一个颜色图

所以情节应与此类似

我的尝试

import numpy as np
import pandas as pd
import seaborn as sns
import random
import matplotlib.pyplot as plt
import matplotlib.cm as cm

x=np.random.randint(0,20,30)
y=np.random.randint(-5,5,30)
z=np.random.randint(-2,10,30)
df=pd.DataFrame(data={'A':x,'B':y,'C':z})

points = plt.scatter(df['A'],df['B'],cmap="jet")
m = cm.ScalarMappable(cmap=cm.jet)
m.set_array(df['C'])
plt.colorbar(points)
sns.lmplot('A', 'B', data=df, hue='C', fit_reg=False)

TypeError: You must first set_array for mappable

我正在混合matplotlib和seaborn，因为在seaborn我不能使用colormap'jet' 但欢迎使用任何替代方法来获得相同的图表

Answer 1

这个怎么样（在scatter命令中使用c = df.C）：

points = plt.scatter(df.A, df.B, c=df.C,cmap="jet", lw=0)
plt.colorbar(points)