时间戳

首页 > 解决方案 > 如何在字符串中搜索关键字，提取该字符串并将其放入新列中？

python - 如何在字符串中搜索关键字，提取该字符串并将其放入新列中？

问题描述

我正在使用熊猫。这是我的df：

df = {'Product Name': ['Nike Zoom Pegasus', 'All New Nike Zoom Pegasus 4', 'Metcon 3', 'Nike Metcon 5']}

我想搜索每个字符串值并仅提取产品类别，然后将提取的字符串值放在另一列（“类别”）中。您可能会注意到，产品名称没有正式的命名约定，因此 .split() 不适合使用。

最终结果应如下所示：

df = {'Product Name': ['Nike Zoom Pegasus', 'All New Nike Zoom Pegasus 4', 'Metcon 3', 'Nike Metcon 5'], 'Category': ['Pegasus', 'Pegasus', 'Metcon', 'Metcon]}

我当前的代码是这样的，但出现错误：

def get_category(product):
if df['Product Name'].str.contains('Pegasus') or df['Product Name'].str.contains('Metcon'):
    return product

df['Category'] = df['Product Name'].apply(lambda x: get_category(x))

ValueError: The truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all().

希望你能帮忙。谢谢！

标签： pythonpandas

解决方案

这个解决方案怎么样，当你有一个新类别时，你所要做的就是将新类别添加到猫数组中。

import pandas as pd
import numpy as np

df = pd.DataFrame({'Product Name': ['Nike Zoom Pegasus', 'All New Nike Zoom Pegasus 4', 'Metcon 3', 'Nike Metcon 5']})
cats = ["Pegasus","Metcon"]
df["Category"] = df["Product Name"].apply(lambda x: np.intersect1d(x.split(" "),cats)[0])


output
                  Product Name Category
0            Nike Zoom Pegasus  Pegasus
1  All New Nike Zoom Pegasus 4  Pegasus
2                     Metcon 3   Metcon
3                Nike Metcon 5   Metcon

推荐阅读