У меня есть два массива numpy. Один предназначен, например, для данных изображения, а второй - для соответствующих меток изображения. Например,
import numpy as np
## image data
a = np.array([[[1,5,9],[7,5,9],[2,5,5]], [[3,8,5],[7,15,10],[1,0,9]], [[8,15,0],[1,1,15],[5,5,0]], [[4,5,10],[10,10,1],[4,5,10]]])
## corresponding label
b = np.array([0,1,2,1])
Я ищу способ получения новых данных изображения, которые содержат только значения для метки изображения 0 и 1. В моем случае я ожидаю новых результатов, как показано ниже,
a_new = np.array([[[1,5,9],[7,5,9],[2,5,5]], [[3,8,5],[7,15,10],[1,0,9]], [[4,5,10],[10,10,1],[4,5,10]]])
b_new = np.array([0,1,1])
Можем ли мы сделать это в numpy без использования цикла for? Спасибо, если кто-то может мне помочь с этим.