python - pandas 系列列表索引

Question

我有一个 pandas Series 对象 obj 列表和一个索引 idx 列表。我想要的是一个新的系列 out，对于 idx 中的每一行，如果 idx 则包含 obj[idx] 的值code> 不是 255，否则 -1。

下面的代码实现了我想要实现的目标，但我想知道是否有更好的方法来实现这一点，特别是没有首先创建 Python 列表然后将其转换为 pandas 系列的开销。

>>> import pandas as pd
>>> obj = [pd.Series([1, 2, 3]), pd.Series([4, 5, 6]), pd.Series([7, 8, 9])]
>>> idx = pd.Series([0, 255, 2])
>>> out = pd.Series([obj[idx[row]][row] if idx[row] != 255 else -1 for row in range(len(idx))])
>>> out
0    1
1   -1
2    9
dtype: int64
>>>

提前致谢。

Answer 1

使用重新索引 + 查找

pd.Series(pd.concat(obj,1).reindex(idx).lookup(idx,idx.index)).fillna(-1)
Out[822]: 
0    1.0
1   -1.0
2    9.0
dtype: float64