根据 Python 函数中的字符串输入创建列输入

Question

所以我正在编写一个自定义函数，我希望将字符串输入转换为数据框，以便模型对其进行预测。现在我被困在编写一段代码，它能够检测输入字符串是否包含特定的单词，如果是，则将其处理到值为 1 或 0 的数据框列。但是，我无法要成功地纠正它，我怎样才能最好地解决这个问题？ 这是我的一段代码：

def function(text_input)
    cols = ['col_foo', 'col_bar', 'col_hello', 'col_world']
    words = ['foo', 'bar', 'hello', 'world']
    data = [text_input.split()]
    for col in cols:
        df[col] = 0
    for word in data:
        if word in words:
           for words in cols:
               df[cols] = 1
    return df

我希望首先使用初始 0 值创建所有列。然后根据text_input中与words列表中的单词匹配的单词，将对应的cols设置为1个值。但是现在，这段代码输出了所有df[cols] = 0

我在这段代码中做错了什么？提前致谢！

Answer 1

我想这就是你想要的：

def text_detecter(text_input): 
     df = pd.DataFrame({'foo':[0],'bar':[0],'hello':[0],'world':[0]}) 
     for col in df.columns: 
     df[col] = df[col].apply(lambda x: 1 if col in text_input else 0) 
     return df 

所以如果你再运行这个：

text_detecter('foo')

这将导致数据框：

     foo  bar  hello  world
0    1    0      0      0