Как найти первую положительную позицию элемента в каждой строке с учетом двумерного массива без использования цикла ⇐ 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]) Но не смог получить правильный ответ.
Все. Учитывая двумерный массив 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]) Но не смог получить правильный ответ.
-
- Похожие темы
- Ответы
- Просмотры
- Последнее сообщение
-
-
Используйте значения плоского массива в качестве ключей в каждой строке двумерного массива.
Anonymous » » в форуме Php - 0 Ответы
- 13 Просмотры
-
Последнее сообщение Anonymous
-