在相同的轴上绘制不同的面色

Question

我是Python和Datascience的新手。

我有一个数据集，其中有两个输入x1和x2和一个输出y：

df=pd.DataFrame({'x1': [1, 2, 2, 1, 0.5, -1, -2, -2, -1, -0.5], 'x2': [1, 1, 2, 2, 0.5, -1, -1, -2, -2, -0.5], 'y': [0, 0, 0, 0, 0, 1, 1, 1, 1, 1]})

    x1   x2  y
0  1.0  1.0  0
1  2.0  1.0  0
2  2.0  2.0  0
3  1.0  2.0  0
4  0.5  0.5  0
5 -1.0 -1.0  1
6 -2.0 -1.0  1
7 -2.0 -2.0  1
8 -1.0 -2.0  1
9 -0.5 -0.5  1

我已经绘制了这个数据集：

plt.scatter(df.x1[df['y']==1], df.x2[df['y']==1], color='red')
plt.scatter(df.x1[df['y']==0], df.x2[df['y']==0], color='blue')
plt.show()

我想做的是在同一个情节中为班级积分设置不同的背景。所以我想要的结果是这样的：

我对Matplotlib并不是很熟悉，我能做到的最好的事情是每个班级都有两个不同的轴，但这不是我真正想要的。

import pandas as pd
import matplotlib.pyplot as plt

df=pd.DataFrame({'x1': [1, 2, 2, 1, 0.5, -1, -2, -2, -1, -0.5], 'x2': [1, 1, 2, 2, 0.5, -1, -1, -2, -2, -0.5], 'y': [0, 0, 0, 0, 0, 1, 1, 1, 1, 1]})
fig, (ax0, ax1) = plt.subplots(nrows=1, ncols=2, figsize=(20, 10))
ax0.scatter(df.x1[df1['y']==1], df.x2[df1['y']==1], color='red')
ax0.set_facecolor('xkcd:blue')
ax1.scatter(df.x1[df1['y']==0], df.x2[df1['y']==0], color='blue')
ax1.set_facecolor('xkcd:red')
plt.show()

我想要这个结果，但是在同一轴上，请问任何解决方案？

Answer 1

我建议为此使用matlab的fill_between方法：

注意：您可以创建一个函数，该函数可以动态构建分隔线的斜率。

# Get X Values
X = {x for x in df['x1']}
X.update({x for x in df['x2']})
X.update({min(X)-0.25, max(X)+0.25})
X = pd.Series(list(X)).sort_values()


fig, ax = plt.subplots(1,1, figsize=(10, 4))

# Plot the backgrounds first to avoid overwriting the colors
ax.fill_between(X,-1*X,X.min(), color='blue', alpha=0.7)
ax.fill_between(-X,X.max(),(X), color='red', alpha=0.7)

ax.scatter(x = 'x1', y='x2', data = df[df['y']==1], color='red')
ax.scatter(x = 'x1', y='x2', data = df[df['y']==0], color='blue')
plt.show()