我进行了搜索,但没有找到对Seaborn库进行升级的答案.我还检查了lmplot()和regplot()的文档,但均未找到.
是否可以扩展和控制回归线的长度?默认情况下,seaborn根据x轴的长度拟合回归线的长度.另一个选择是使用参数truncate = True-这只会将回归线限制在数据范围内.
还有其他选择吗?
在我的示例中,我希望将较低的回归线向下延伸至x = 0.上面的线一直延伸到与下面的线相交.
import matplotlib.pyplot as plt
import pandas as pd
import seaborn as sns
file = 'cobbles.csv'
df = pd.read_csv(file, sep=',')
sns.regplot(x='downward_temp', y='downward_heat', data=df, ci=None)
sns.regplot(x='upward_temp', y='upward_heat', data=df, ci=None, order=2)
plt.xlim([0,25])
plt.ylim([0,100])
plt.show()
解决方法:
如果在绘制之前知道x极限,则可以在调用regplot之前为轴设置set_xlim,然后seaborn将在xlim范围内扩展回归线和CI.
import matplotlib.pyplot as plt
import pandas as pd
import seaborn as sns
file = 'cobbles.csv'
df = pd.read_csv(file, sep=',')
fig, ax = plt.subplots()
xlim = [0,25]
ax.set_xlim(xlim)
sns.regplot(x='downward_temp', y='downward_heat', data=df, ci=None, ax=ax)
sns.regplot(x='upward_temp', y='upward_heat', data=df, ci=None, order=2, ax=ax)
ax.set_ylim([0,100])
plt.show()