我正在尝试从一个中获得两个dataframe。 dataframe有两组单词(中性和非中性),因此我需要将其划分为只有中立单词的数据集和另一个只有非中性单词(mantaining所有行和列)。这些单词位于一个名为PALABRA的列中。

这是变量中的单词的示例(它们的数量多于这些):

neutral_words = ('CAR','CLOUD','SUN')
nonneutral_words = ('ACCIDENT','BUG','BURN')

DF看起来像这样:

   PRESSEDKEY  PALABRA   COLOR KEYCORR    RT CORRECT
90           v      BURN    red       r   496       N
96           v       SUN    red       r  1307       N
102          v       BUG    red       r     0       N
108          v     CLOUD   blue       a   168       N
114          v   ACCIDENT  green      v    73       Y

这样,我需要将Dataframe划分为DF1中的Dataframe与neutral_words和DF2使用nonneutral_words。我怎样才能做到这一点?

分析解答

您可以使用isin

df1 = df.loc[df['PALABRA'].isin(neutral_words)]
df2 = df.loc[df['PALABRA'].isin(nonneutral_words)]