python - Pandas loc : Multiindexing and selecting rows with specific index values

Question

我有一个包含 3 列 A B C 的数据框

A B C
1 2 10
2 3 20
3 0 30

我已经为 A、B 列建立了索引并进行了排序

df = ( df.set_index([A,B]).sort_index())

我想选择索引为 A(1,2) 和索引 B (2,3) 的行

df1 = df.loc[[[1,2],[2,3]]]

它抛出一个错误..我做错了什么..我做了一些其他的实验，但无法找到解决方案..

Answer 1

IIUC，您使用元组，因为这就是 MultiIndex 值的散列方式(因为列表无法散列)。

print(df.loc[[(1, 2), (2, 3)]])
      C
A B
1 2  10
2 3  20