Numpy.where на массиве струн с использованием regex ⇐ Python

[Anonymous]

У меня есть массив, как: < /p>

array = ['A0','A1','A2','A3','A4','B0','B1','C0']
< /code>

и хочу получить массив, который является истинным для значений с a, за которым следует число от 0 до 2.selection = np.where ((array == 'A0') | (array == 'A1') | (array == 'A2'), 1, 0)
< /code>

Но есть ли более элегантный способ сделать это, используя, например, регулярное выражение, как: < /p>

selection = np.where (array == 'A[0-1]', 1, 0)


Подробнее здесь: https://stackoverflow.com/questions/585 ... sing-regex