我有一个以下的熊猫 df，我想根据 num 列上的任何数字从列表列中提取元素：
list num
[1,2,3,4,5] 3
[7,2,1,3,4] 4
获得：
list num element
[1,2,3,4,5] 3 4
[7,2,1,3,4] 4 4
我试过了：
df['element'] = df['list'].apply(lambda x: x[df['num'].apply(lambda y: y)])
但我得到了TypeError: list indices must be integers or slices, not Series。
无论如何我可以做到这一点吗？谢谢！
使用DataFrame.apply每行有axis=1：
df['element'] = df.apply(lambda x: x['list'][x['num']], axis=1)
print (df)
list num element
0 [1, 2, 3, 4, 5] 3 4
1 [7, 2, 1, 3, 4] 4 4
或列表理解zip：
df['element'] = [x[y] for x, y in zip(df['list'], df['num'])]
如果可能，某些值与列表不匹配，可以使用以下方法：
def func(a, b):
try:
return a[b]
except Exception:
return np.nan
df['element'] = df.apply(lambda x: func(x['list'], x['num']), axis=1)
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句