列表理解 if 字符串中的关键字

Question

我正在清理数据框中的列，并且想根据某些关键字拆分“描述”列。我写了一个单行代码，但由于某种原因，即使满足条件，仍然返回 np.nan。

def find_and_suck_hashtag_vacuum(string, keyword, output):
    return [output if keyword in string else np.nan]

变量设置为

string_samp = ['Air conditioning,Car park,Interesting wine list,Restaurant offering vegetarian menus,Wheelchair access']

output = find_and_suck_hashtag_vacuum(string_samp, "Air", "AC")

但是在上述情况下该函数仍然返回 np.nan。我错过了什么？

Answer 1

您的输入是字符串列表，而不是字符串。因此代码确定字符串“Air”不是列表中的元素。工作代码是：

string_samp = ['Air conditioning,Car park,Interesting wine list,Restaurant offering vegetarian menus,Wheelchair access']

output = find_and_suck_hashtag_vacuum(string_samp[0], "Air", "AC")