在使用pandas库时,可以使用apply方法来应用自定义函数到DataFrame的列上。如果想要使用一列中的元素返回另一列中的列表中的索引,可以按照以下步骤进行操作:
import pandas as pd
df = pd.DataFrame({'A': [1, 2, 3, 4, 5], 'B': [[1, 2, 3], [4, 5, 6], [7, 8, 9], [10, 11, 12], [13, 14, 15]]})
def get_index(element, lst):
return lst.index(element)
df['C'] = df.apply(lambda row: get_index(row['A'], row['B']), axis=1)
在上述代码中,lambda函数用于将每一行的'A'列的元素和'B'列的列表传递给自定义函数get_index,并将返回的索引值赋值给新的列'C'。
最终,DataFrame将会如下所示:
A B C
0 1 [1, 2, 3] 0
1 2 [4, 5, 6] 1
2 3 [7, 8, 9] 2
3 4 [10, 11, 12] 1
4 5 [13, 14, 15] 0
以上是使用apply方法来实现一列中的元素返回另一列中的列表中的索引的方法。关于pandas的更多信息和使用方法,可以参考腾讯云的相关产品和文档:
领取专属 10元无门槛券
手把手带您无忧上云