使用' for‘循环为df中的每一列创建Plotly图

Question

我想使用Python语言中的for循环在df中制作多个Plotly散点图(每列一个)。我还希望能够通过输入列名来显示绘图。

请参阅示例代码：

import plotly.express as px
import pandas as pd

df = pd.DataFrame({'A': [3, 1, 2, 3],
                   'B': [5, 6, 7, 8],
                   'C': [2, 1, 6, 3]})
df

     A    B    C
0    3    5    2
1    1    6    1
2    2    7    6
3    3    8    3

我得到的最接近的是：

for i in df.columns:
i = px.scatter(df,
               x="A",
               y=i)

但这无法为每个图赋值。我希望能够通过输入A来显示列A的绘图，通过输入B来显示B的绘图，依此类推。

Answer 1

您可以将图添加到字典中，如下所示

plots = {i: px.scatter(df, x="A", y=i) for i in df.columns}

它等同于，但比

plots = {}
for i in df.columns:
    plots[i] = px.scatter(df, x="A", y=i)

然后，您可以使用plots['A']、plots['B']和plots['C']显示每个图