Блог

Главная - Вопросы по программированию - Как разделить массив в соответствии с условием в numpy?

Как разделить массив в соответствии с условием в numpy?

#python #numpy

#python #numpy

Вопрос:

Например, у меня есть ndarray то, что:

 a = np.array([1, 3, 5, 7, 2, 4, 6, 8])

Теперь я хочу разделить a на две части: одна — все числа <5, а другая — все>= 5:

 [array([1,3,2,4]), array([5,7,6,8])]

Конечно, я могу пройти a и создать два новых массива. Но я хочу знать, предоставляет ли numpy несколько лучших способов?

Аналогично, для многомерного массива, например

 array([[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9],
       [2, 4, 7]])

Я хочу разделить его в соответствии с первым столбцом <3 и >= 3, результат которого:

 [array([[1, 2, 3],
        [2, 4, 7]]), 
 array([[4, 5, 6],
        [7, 8, 9]])]

Есть ли какие-либо лучшие способы вместо обхода его? Спасибо.

Ответ №1:

 import numpy as np

def split(arr, cond):
  return [arr[cond], arr[~cond]]

a = np.array([1,3,5,7,2,4,6,8])
print split(a, a<5)

a = np.array([[1,2,3],[4,5,6],[7,8,9],[2,4,7]])
print split(a, a[:,0]<3)

Это приводит к следующему результату:

 [array([1, 3, 2, 4]), array([5, 7, 6, 8])]

[array([[1, 2, 3],
       [2, 4, 7]]), array([[4, 5, 6],
       [7, 8, 9]])]

Комментарии:

1. Не arr[cond], arr[~cond] означает ли это, что он проверяет каждый элемент массива на одно и то же условие дважды?

2. @endolith Нет. a[:,0]<3 Будет создан массив bool с тем же размером, a что и , в котором указано, какие элементы попадают в выделение. Это просто отрицание массива, что не дорого.

Ответ №2:

Это может быть быстрым решением 

 a = np.array([1,3,5,7])
b = a >= 3 # variable with condition
a[b] # to slice the array
len(a[b]) # count the elements in sliced array

Ответ №3:

1d массив a = numpy.array([2,3,4,…]) a_new = a[(a < 4)] # чтобы получить элементы меньше 5

2d-массив на основе столбца (учитывайте, что значение столбца i должно быть меньше 5, a = numpy.array([[1,2],[5,6],…] a = a[(a[:,i] < 5)]

если ваше условие основано на нескольких столбцах, вы можете создать новый массив, применив условия к столбцам. Затем вы можете просто сравнить новый массив со значением 5 (согласно моему предположению), чтобы получить индексы и следовать приведенным выше кодам. Обратите внимание, что все, что я написал в () , возвращает индексный массив.