有没有办法在 Altair 中循环并绘制情节？

如何解决有没有办法在 Altair 中循环并绘制情节？

我正在尝试在 altair 中复制这个 matplotlib 图。最好的方法是什么？

import numpy as np
import matplotlib as plt
N = 10
a = np.random.normal(0,0.2,size=(N))
b = np.random.normal(1,0.5,size=(N))
x_seq = np.linspace(-3,3,50)

for i in range(N):
  plt.plot(x_seq,a[i] + b[i] * x_seq,"k-",alpha=0.3);

解决方法

Matplotlib 有一个低级绘图界面，可促进对数组的循环操作（如您的示例中所示），而 Altair 和 Seaborn 等软件包则具有处理数据框及其列名的高级界面。

因此，您可以在 Altair 中执行以下操作，而不是使用 for 循环：

import altair as alt


dfx = pd.DataFrame([a[i] + b[i] * x_seq for i in range(N)]).T
dfx['x'] = x_seq
dfx = dfx.melt(id_vars='x',var_name='line_group',value_name='y')

alt.Chart(dfx).mark_line().encode(
    x='x',y='y',color='line_group:N') # N is to explicitly indicate a nominal variable

You can read more about the melting step and why long data is perferred in the docs.