主页 / 问题 / 1363343
Accepted
Trulimon
Asked:2022-05-18 13:56:49 +0000 UTC2022-05-18 13:56:49 +0000 UTC

检查单元格列表中单词的出现

  • 772

我们需要创建一个“测试”列，该列将包含“关键字”单元格内容的一般特征。

如果“关键字”包含“婚姻”一词，则将“情节剧”一词放在“测试”中。

如果'keywords'包含'future'这个词，那么把'fantasy'这个词放在'test'中。

如果 'keywords' 包含单词 'tale'，则将单词 'tale' 放入 'test' 中。

如果'keywords'中没有列出的单词，则将'other'放在'test'中。

表格和链接：https ://drive.google.com/file/d/1ps6nLGq30Or7sMNfQrGrjlIA-kr_PL22/view?usp=sharing

工作表

df['test'] = ["мелодрама" if ("marriage" in x)
                               else("фантастика" if ("future" in x) 
                                    else("сказка" if ("tale" not in x) 
                                         else 'другое')) for x in df['keywords'].str.split(pat='|')]

错误：“TypeError：'float' 类型的参数不可迭代”

python

1 个回答

  1. Best Answer

    试试这样：

    mapping = {"мелодрама": "marriage", "фантастика": "future", "сказка": "tale"}

# регулярные выражения для замены целых строк 
pat = {fr".*\b{v}\b.*": k for k, v in mapping.items()}

# создаем новый столбец при помощи замены строк
df["test"] = df["keywords"].replace(pat, regex=True)

# создаем "boolean mask" для строк из категории "другое"
mask = ~df["test"].isin(mapping.keys())

# заменяем строки по маске на "другое"
df.loc[mask, "test"] = "другое"

    结果：

    In [29]: df
Out[29]:
                                            keywords  score        test
0             avatar|future|marine|native|paraplegic    7.9  фантастика
1  goddess|marriage ceremony|marriage proposal|pi...    7.1   мелодрама
2                bomb|espionage|sequel|spy|terrorist    6.8      другое
3  deception|imprisonment|lawlessness|police offi...    8.5      другое
4                                                NaN    7.1      другое
5  alien|american civil war|male nipple|mars|prin...    6.6      другое
6          sandman|spider man|symbiote|venom|villain    6.2      другое
7  17th century|based on fairy tale|disney|flower...    7.8      сказка
8  artificial intelligence|based on comic book|ca...    7.5      другое
9                   blood|book|love|potion|professor    7.5      другое
    • 1

相关问题