你可以使用apply函数来遍历pandas数据帧列中的列表元素,并返回一个新列中的列表。以下是一个示例代码:
import pandas as pd
# 创建一个示例数据帧
df = pd.DataFrame({'col1': [[1, 2, 3], [4, 5], [6, 7, 8, 9]]})
print('原始数据帧:')
print(df)
# 定义一个函数来处理每个列表元素
def process_list(lst):
return [x*2 for x in lst]
# 使用apply函数遍历列表元素,并返回一个新列中的列表
df['col2'] = df['col1'].apply(process_list)
print('新数据帧:')
print(df)
这个示例代码中,首先我们创建了一个包含一个列的数据帧,该列的每个元素都是一个列表。然后定义了一个process_list
函数,该函数对输入的列表元素进行处理(这里是将每个元素乘以2)。最后使用apply
函数将process_list
函数应用到数据帧的列上,并将结果存储在一个新列中。