таблица соответствия массивов - PullRequest
Новая
       7

таблица соответствия массивов

0 голосов
/ 05 января 2019

эй, мне нравится сортировать мой массив следующим образом: по col_index и уникальному значению. Пример: 

import numpy as np
a = np.array([[21,30,105,1],[1,21,5,1],[5,15,20,1],[7,3,31,2]])
  #print:
  #[[ 21  30 105   1]
  #[  1   21 5    1]
  #[  5  15  20   1]
  #[  7   3  31   2]]

мое желание создать новый массив со всеми строками, которые имеют '1' (уникальное значение) в 4-м (например, [0] [3], a [1] [3] и a [2] [3] ) и со значением, большим 20 для 2-го столбца (например, [0] [1] и [1] [1])

для моего примера мое желание получить: 

new_a = np.array([[21,30,105,1],[1,21,5,1]])

1 Ответ

0 голосов
/ 05 января 2019

Использование Индексация логического массива : 

res = a[(a[:, 3] == 1) & (a[:, 1] > 20)]

# array([[ 21,  30, 105,   1],
#        [  1,  21,   5,   1]])
...