Question

我有一个pandas数据帧，在其中一个列中，列表值出现在某些值中。我需要能够提取列表的第一项，如果它是一个列表，如果它不是一个列表，那么该值将保持不变。我正在努力使用lambda函数实现它：

df1 = pd.DataFrame({'Fruits':['Apple',['Banana',6],'Kiwi','Cheese']})

df1['Fruits'] = df1['Fruits'].apply(lambda(x): x[0] if (type(x) == 'list') else x)

如果我使用上述列，则保持不变。我猜这个必须是lambda函数中的条件语句的问题....

如果有更好的方法在熊猫中实现这一点，我也会感兴趣。

Answer 1

您可以将''从'list'移至list：

df1['Fruits'] = df1['Fruits'].apply(lambda  x : x[0] if type(x) == list else x)
print (df1)
   Fruits
0   Apple
1  Banana
2    Kiwi
3  Cheese

类似的解决方案是使用isinstance：

df1['Fruits'] = df1['Fruits'].apply(lambda x: x[0] if isinstance(x, list) else x)
print (df1)
   Fruits
0   Apple
1  Banana
2    Kiwi
3  Cheese

或者可以使用list comprehension：

df1['Fruits'] = [x[0] if type(x) == list else x for x in df1['Fruits']]
print (df1)
   Fruits
0   Apple
1  Banana
2    Kiwi
3  Cheese

访问Pandas列中不规则出现的第一个列表值项

1 个答案: