大数据散点图添加线条

Question

我需要一个包含 77M 以上行的数据集的散点图，并添加像 plt.axlines 这样的行。

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt

df = pd.DataFrame({ 'x' :np.random.normal(0,1,77000000),
                    'y' : np.random.normal(0,1,77000000) })

intercept_a = (-2, 0.5)
slope_a     = 0.5

intercept_b = (-0.5, -1.5)
slope_b     = 0.1

print(df.shape) # (77000000, 2)

预期结果

# (but w/ 77M rows)

plt.scatter(x = df.loc[1:3000,'x'], 
            y = df.loc[1:3000,'y'])
plt.axline(intercept_a, slope=slope_a, color='red', label='dry')
plt.axline(intercept_b, slope=slope_b, color='blue', label='edge')

我尝试过使用 datashader 和 vaex，但未成功

Datashader

# DATASHADER 
import datashader as ds

cvs = ds.Canvas(plot_width=800, plot_height=500)  # auto range or provide the `bounds` argument
agg = cvs.points(df, 'x', 'y')
img = ds.tf.set_background(ds.tf.shade(agg, cmap=cc.fire), "white").to_pil()
plt.axline(intercept_a, slope=slope_a, color='red', label='dry')
plt.axline(intercept_b, slope=slope_b, color='blue', label='edge')
plt.imshow(img)

Vaex

# VAEX
import vaex
df_vaex = vaex.from_pandas(df=df)

df_vaex.viz.heatmap("x", "y",
               limits = '97%',
               figsize=(15, 7),
               colorbar = False);

plt.axline(intercept_a, slope=slope_a, color='red', label='dry')
plt.axline(intercept_b, slope=slope_b, color='blue', label='edge')

PS 此处是我的真实数据的数据着色器图，我需要背景白色才能很好地可视化饱和点。