python - 访问 Pandas 数据框中的数据

Question

所以我有一个 pandas 数据框，它是这样的:

这是一段[if语句，for循环]

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

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

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

Answer 1

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']]]

或者比较集合:

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

<小时/>

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