python-3.x - 使用 'for' 循环为 df 中的每一列创建 Plotly 图

Question

我想使用 for 在 df 中制作多个 Plotly Scatter 图(每列一个) Python 中的循环。我还希望能够通过输入列名来显示图。
查看示例代码:

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']