如何使用np.where或列表理解创建子字符串的掩码？

import numpy as np

def create_mask(string, substring):
    mask = np.where(np.char.find(string, substring) >= 0, True, False)
    return mask

string = np.array(['hello', 'world', 'numpy', 'where'])
substring = 'o'

mask = create_mask(string, substring)
print(mask)

[ True False False  True]

def create_mask(string, substring):
    mask = [True if substring in s else False for s in string]
    return mask

string = ['hello', 'world', 'numpy', 'where']
substring = 'o'

mask = create_mask(string, substring)
print(mask)

[True, False, False, True]