gpt4 book ai didi

python - 从 df 列连接列表

转载 作者:太空宇宙 更新时间:2023-11-04 09:37:13 24 4
gpt4 key购买 nike

我有一个如下所示的数据框:

import pandas as pd
df = pd.DataFrame({'id': [1,2,3], 'tokens': [["anna","brings","work","life"], ["apples","bananas"], ["sun","trial","once"]]})

我想做的是简单地将列中列表中的所有字符串连接到一个列表中,以便生成的列表如下所示:

result_list = ["anna","brings","work","life", "apples","bananas","sun","trial","once"]

我尝试遍历每一行,然后遍历列表中的每个元素:

result_list = [item for sublist in df.tokens for item in sublist]

但我得到的是列表的每个符号的列表。我错过了什么?

最佳答案

使用chain.from_iterable :

import pandas as pd

from itertools import chain

df = pd.DataFrame(
{'id': [1, 2, 3], 'tokens': [["anna", "brings", "work", "life"], ["apples", "bananas"], ["sun", "trial", "once"]]})

result = list(chain.from_iterable(df.tokens))
print(result)

关于python - 从 df 列连接列表,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/53227233/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com