находят индексы х, у матрицы конкретных значений в питоне

Question

преобразовать список целых чисел в два dimensinal массива, как это:

data = numpy.array(l) 
shape = (10, 30) 
data = data.reshape(shape) 

Я стараюсь, чтобы получить индексы х, у матрицы значения, которые больше определенного порога и ниже некоторого другого порога.

Я попытался сделать следующий, но она дает некоторые ошибки:

data_indices = numpy.where(data<=obj_value_max and data>=obj_value_min) 

Ошибка:

ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

Answer 1

Вы должны изменить свою where линию на что-то вроде:

data_indices = numpy.where((data<=obj_value_max) & (data>=obj_value_min)) 

Обратите внимание на ()s вокруг каждого условного предложения и использования & (что означает «и»). Это работает, потому что в numpy <,<=,>,>=,&,|,... переопределены, то есть они действуют иначе, чем в родном python. and и or нельзя переопределить, и именно поэтому вы получаете сообщение об ошибке, которое вы получаете.

Чтобы получить пар индексов для каждого значения (вместо (массив индексов х, у массива индексов)), сделать

zip(*data_indices)