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

使用np.where函数可以根据条件创建一个掩码数组，该数组的元素根据条件的满足与否被设置为True或False。列表理解也可以实现相同的功能。

下面是使用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]

上述代码中，np.char.find函数用于在字符串数组中查找子字符串，返回子字符串在每个字符串中的索引位置。然后，使用np.where函数根据索引位置是否大于等于0来创建掩码数组，满足条件的位置被设置为True，不满足条件的位置被设置为False。

如果要使用列表理解来创建子字符串的掩码，可以使用以下代码：

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)

输出结果为：