Как найти первую положительную позицию элемента в каждой строке с учетом двумерного массива без использования циклаPython

Программы на Python
Ответить Пред. темаСлед. тема
Гость
 Как найти первую положительную позицию элемента в каждой строке с учетом двумерного массива без использования цикла

Сообщение Гость »


Все. Учитывая двумерный массив numpy, как узнать индекс строки и столбца первого положительного элемента в каждой строке без использования цикла. Например, данные

data = np.array([[1.0, -2.0, 3.0,3], [0,0, 1,5, -2,0,5], [-1.0, -2.0, -3.0, -5]]) Как найти индексы каждого положительного элемента в строке. В качестве ответа он ожидает получить 2 индекса 1d: row_idx=[0,1], col_idx=[0,1], что соответствует элементам [1 ,1.5] из данных. Кто-нибудь может пролить свет на этот вопрос, заранее большое спасибо!!!

Я пробовал использовать

импортировать numpy как np распечатать (данные) row_idx = np.any(данные>0, ось=1) col_idx = np.argmax (данные> 0, ось = 1) print('индексы строк:', row_idx) print('индексы столбцов: ', col_idx) печать (данные [row_idx, col_idx]) Но не смог получить правильный ответ.
Реклама
Ответить Пред. темаСлед. тема

Быстрый ответ

Изменение регистра текста: 
Смайлики
:) :( :oops: :roll: :wink: :muza: :clever: :sorry: :angel: :read: *x)
Ещё смайлики…
   
К этому ответу прикреплено по крайней мере одно вложение.

Если вы не хотите добавлять вложения, оставьте поля пустыми.

Максимально разрешённый размер вложения: 15 МБ.

  • Похожие темы
    Ответы
    Просмотры
    Последнее сообщение

Вернуться в «Python»