import seaborn as sns
iris = sns.load_dataset("iris")
grid = sns.JointGrid(iris.petal_length, iris.petal_width, space=0, size=6, ratio=50)
grid.plot_joint(plt.scatter, color="g")
上面的代码将基于Iris数据集创建散点图。我想在[3,.05]处添加另一个红色的数据点;或将数据集中的最后一点设为红色。我该怎么做呢?
最佳答案
要在自定义x
和y
坐标处添加点,请在坐标中添加matplotlib.pyplot.scatter
:
plt.scatter(x=3, y=0.5, color='r')
为了给最后一点着色,请在数据上使用
.iloc
定位符:plt.scatter(iris.petal_length.iloc[-1], iris.petal_width.iloc[-1], color='r')
请注意,
iloc
定位符来自pandas
,plt.scatter
来自matplotlib.pyplot
。这两个都是seaborn的强制性依赖项,因此,如果您使用的是seaborn,则一定要将它们安装在计算机上。例如:
import seaborn as sns
import matplotlib.pyplot as plt
iris = sns.load_dataset("iris")
grid = sns.JointGrid(iris.petal_length, iris.petal_width, space=0, size=6, ratio=50)
grid.plot_joint(plt.scatter, color="g")
# add your point
plt.scatter(x=3, y=0.5, color='r')
# or
# plt.scatter(iris.petal_length.iloc[-1], iris.petal_width.iloc[-1], color='r')