首页 > 解决方案 > 访问 Pandas 数据框中的数据

问题描述

所以我有一个熊猫数据框,它是这样的:

这是一段[if-statement, for-loop]

这是第二段 [for-loop, java]

解释一下,左列用作文本数据,右列对文本数据的内容进行分类。

我只想在第二段访问“java”。如何访问数据框中的列表?

标签: pythonpandas

解决方案


IIUC 需要:

df = pd.DataFrame({'col1':['This is a paragraph','This is a second paragraph'],
                   'col2':[['if-statement', 'for-loop'],['for-loop','java']]})

df = df[df['col2'].apply(lambda x: 'java' in x)]
#alternative solution
#df = df[['java' in x for x in df['col2']]]

或比较sets:

df = df[df['col2'].apply(set) >= set(['java'])]

print (df)
                         col1              col2
1  This is a second paragraph  [for-loop, java]

推荐阅读