python - 根据 pandas Dataframe 中的索引从列表中删除元素

Question

如何根据 pandas Dataframe 中的索引范围从列表中删除元素。

假设 DataFrame 是这样的

df:

    values           size
0  [1,2,3,4,5,6,7]    2     #delete first 2 elements from list
1  [1,2,3,4]          3     #delete first 3 elements from list
2  [9,8,7,6,5,4,3]    5     #delete first 5 elements from list

预期输出是

df:

    values           size
0  [3,4,5,6,7]        2
1  [4]                3
2  [4,3]              5

Answer 1

将列表理解与索引结合使用:

df['values'] = [i[j:] for i, j in zip(df['values'], df['size'])]
print (df)
            values  size
0  [3, 4, 5, 6, 7]     2
1              [4]     3
2           [4, 3]     5