如何将新元素添加到列表的pandas.DataFrame列? [英] How to add new element to pandas.DataFrame column which is list?

问题描述

我有这样的pandas.DataFrame:

id       count        fruit_list
---------------------------------------
0        42           [orange, apple]
1        57           [lemon]
2        86           [kiwi]
3        33           [pineapple, pear]
4        11           [apple, lemon]
...

我需要在每行中向fruit_list添加新字符串.假设字符串是banana，所以我需要以DataFrame结尾:

I need to add new string to fruit_list in every row. Let's say the string is banana, so I need to end with DataFrame like this:

id       count        fruit_list
---------------------------------------
0        42           [orange, apple, banana]
1        57           [lemon, banana]
2        86           [kiwi, banana]
3        33           [pineapple, pear, banana]
4        11           [apple, lemon, banana]
...

请问我如何最轻松地做到这一点?

How can I do this most easily please?

推荐答案

尝试apply:

df['fruit_list'] = df['fruit_list'].apply(lambda x: x + ['banana'])

示例:

df = pd.DataFrame({'count': [42, 57], 
                   'fruit_list': [['orange', 'apple'], ['lemon']], 
                   'id': [0, 1]}).reindex_axis(['id','count','fruit_list'], 1)
print (df)
   id  count       fruit_list
0   0     42  [orange, apple]
1   1     57          [lemon]

df['fruit_list'] = df['fruit_list'].apply(lambda x: x + ['banana'])
print (df)
   id  count               fruit_list
0   0     42  [orange, apple, banana]
1   1     57          [lemon, banana]

这篇关于如何将新元素添加到列表的pandas.DataFrame列?的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！