如何在取列平均值之前更新行值？

Question

我有一个由这些代码行组成的数据文件。

projections = pd.DataFrame({"stock":["A","B","C"],
    "strong":[.39,.30,.06], "moderate":[.17,.15,.14], "weak":[-.05,0,.22]})

projections = projections.set_index("stock")

看上去像这样。

      strong    moderate    weak
stock           
A      0.39        0.17     -0.05
B      0.30        0.15      0.00
C      0.06         0.14     0.22

我想把"B“行乘以-1，然后取每列的平均值。

这是我决定要做的。

(projections.T * [1,-1,1]).mean(axis=1)

但我想要一个更优雅、更通用的解决方案.

我想要一个可以将任何有效列表分配给mylist的解决方案

mylist = ["B"]
pd.merge(
    projections.T.loc[:,projections.T.columns.isin(mylist)],
    projections.T.loc[:,~projections.T.columns.isin(mylist)].mul(-1),
    left_index=True, right_index=True
).mean(axis=1)

但又更优雅了。

Answer 1

你能试一下看看它是否有效吗？

(
    projections.T
    .assign(B=lambda x: x.B.mul(-1))
    .mean(1)
)

如果有多行要乘-1，则可以将所有这些行放到mylist中，然后尝试以下代码。

mylist = ["A", "B"]
(
    projections
    .mul([-1 if e in mylist else 1 for e in projections.index], 0)
    .mean(0)
)