python - 如何通过一系列列名从数据框中选择值？

Question

我有一个数据框df:

    A   B
0   1   4
1   2   5
2   3   6

和一系列s:

0    A
1    B
2    A

现在我想从 df 中选择列名在 s 中指定的值。预期结果是:

0    1  <- from column A
1    5  <- from column B
2    3  <- from column A

我怎样才能高效地完成这项工作？

Answer 1

使用Index.get_indexer对于 Series 的索引并通过 2d array 中的 numpy 索引选择值:

a = df.to_numpy()
b = a[np.arange(len(df)), df.columns.get_indexer(s)]
print (b)
[1 5 3]

---

s1 = pd.Series(b, s.index)
print (s1)
0    1
1    5
2    3
dtype: int64