我正在尝试从一个中获得两个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)]