您可以使用seaborn功能来绘制图形。输入dir(sns)以查看所有绘图。这是您在sns.scatterplot中的输出。您可以在此处
检查api文档或者
示例代码和图表。
import seaborn as sns
import pandas as pd
df = pd.DataFrame([[ 1.82716998, -1.75449225],
[ 0.09258069, 0.16245259],
[ 1.09240926, 0.08617436]], columns=["x", "y"])
df["val"] = pd.Series([1, -1, 1]).apply(lambda x: "red" if x==1 else "blue")
sns.scatterplot(df["x"], df["y"], c=df["val"]).plot()
提供
这是你想要的完全的输入输出吗?
你可以使用pyplot来完成,导入seaborn仅会改变pyplot的颜色和绘图方案。
import seaborn as sns
import matplotlib.pyplot as plt
fig, ax = plt.subplots()
df = pd.DataFrame([[ 1.82716998, -1.75449225],
[ 0.09258069, 0.16245259],
[ 1.09240926, 0.08617436]], columns=["x", "y"])
df["val"] = pd.Series([1, -1, 1]).apply(lambda x: "red" if x==1 else "blue")
ax.scatter(x=df["x"], y=df["y"], c=df["val"])
plt.plot()
这里是一个stackoverflow帖子,介绍如何使用sns.lmplot实现相同效果。