首页 > 解决方案 > 将多个熊猫列附加到列表中

问题描述

我有一个由多列组成的 df。我想获取我的 df 的一个子集并将这些列附加到一个列表中。它们都是字符串。

样本df

A         B       C       D       E       F
word1   word2   word3   word4   word5   word6   
word7   word8   word9   word1   word2   word3 
word4   word5   word6   word7   word8   word9 
word1   word2   word3   word4   word5   word6 
word7   word8   word9   word1   word2   word3

因此,我会以 C、E 和 F 列为例,将它们作为列表附加在一起。

期望的输出

list = [word3, word9, word6, word3, word9, 
        word5, word2, word8, word5, word2, 
        word6, word3, word9, word6, word3]

提前致谢!非常感激!!

标签: pandasstringlistdataframe

解决方案


pandas.melt与 一起使用tolist

df[["C", "E", "F"]].melt()["value"].tolist()

输出:

['word3',
 'word9',
 'word6',
 'word3',
 'word9',
 'word5',
 'word2',
 'word8',
 'word5',
 'word2',
 'word6',
 'word3',
 'word9',
 'word6',
 'word3']

推荐阅读