python - 如何从 Pandas DataFrame 的每一行中选择特定列？

Question

我有一个这种格式的 DataFrame:

    a   b   c
0   1   2   3
1   4   5   6
2   7   8   9
3   10  11  12
4   13  14  15

和一个像这样的数组，带有列名:

['a', 'a', 'b', 'c', 'b']

我希望提取一组数据，每行一个值。列名数组指定我想要从每一行中提取哪一列。在这里，结果将是:

[1, 4, 8, 12, 14]

这是否可以作为 Pandas 的单个命令，或者我需要迭代吗？我尝试使用索引

i = pd.Index(['a', 'a', 'b', 'c', 'b'])
i.choose(df)

但我遇到了段错误，由于缺少文档，我无法诊断。

Answer 1

你可以使用 lookup ，例如

>>> i = pd.Series(['a', 'a', 'b', 'c', 'b'])
>>> df.lookup(i.index, i.values)
array([ 1,  4,  8, 12, 14])

如果您愿意，i.index 可以与 range(len(i)) 不同。