临时绘制两列的总和 python

plot ad hoc the sum of two columns python

我有一个包含 2 列的数据框,即 mean 和 sd。

有什么方法可以临时绘制这两列的总和?

我不想创建第三列,但我想绘制:mean + sd 作为一列。

在使用 ggplot 的 R 世界中,您可以使用 dplyr 并执行以下操作:

df %>% mutate(sumsd = sum+sd) %>% ggplot(.) 等而不将数据保存到列中。

要绘制的数据需要在内存中的某个地方;数据框列是实现这一目标的一种方式。 假设以下数据

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

a = np.random.rand(100,20)
df = pd.DataFrame({"mean" : np.mean(a,1),
                    "std" :  np.std(a,1)})

您现在可以使用 pandas 绘图包装器。只需添加两列。

(df["mean"]+df["std"]).plot()

你也可以对数据帧求和,

df.sum(axis=1).plot()

或者如果您有更多列,

df[["mean","std"]].sum(axis=1).plot()

当然也可以直接使用matplotlib来完成,

# case 1
plt.plot(df.index, (df["mean"]+df["std"]))
# case 2
plt.plot(df.index, df.sum(axis=1))
# case 3
plt.plot(df.index, df[["mean","std"]].sum(axis=1))