从pandas列中列表的开头移除n个元素，其中n是另一列中的值

import pandas as pd

# 创建示例DataFrame
df = pd.DataFrame({'list': [[1, 2, 3, 4, 5], [6, 7, 8, 9, 10]],
                   'n': [2, 3]})

# 定义移除函数
def remove_elements(row):
    n = row['n']
    return row['list'][n:]

# 应用函数并创建新列
df['new_list'] = df.apply(remove_elements, axis=1)

print(df)

           list  n   new_list
0  [1, 2, 3, 4, 5]  2  [3, 4, 5]
1  [6, 7, 8, 9, 10]  3  [9, 10]