根据索引的子集返回 P​​andas 索引的布尔掩码

Question

我有一个这样的 df：

df = pd.DataFrame(index=["a", "b", "c"], data={"col": [1, 2, 3]})

以及索引的子集：s = ["a", "b"]

我想形成布尔掩码，这样我就可以只更改“a”和“b”行的值，如下所示：

df.loc[m, "col"] = [10, 11]

有没有一种巧妙的方法可以做到这一点？

Answer 1

如果赋值列表与索引列表相同，则使用：

df.loc[s, "col"]  = [10, 11]
print (df)
   col
a   10
b   11
c    3

如果 Index.isin 使用布尔掩码并且列表的顺序不同，例如索引得到不同的输出：

df = pd.DataFrame(index=["a", "b", "c"], data={"col": [1, 2, 3]})
#swapped order
s = ["b", "a"]


m = df.index.isin(s)
df.loc[m, "col"]  = [10, 11]
print (df)
   col
a   10
b   11
c    3

df.loc[s, "col"]  = [10, 11]
print (df)
   col
a   11
b   10
c    3