Как разбить массив в соответствии с условием в NumPy? - PullRequest
16 голосов
/ 05 октября 2011

Например, у меня есть 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]])]

Есть ли лучшие пути, вместо того чтобы пройти его?Благодаря.

Ответы [ 2 ]

28 голосов
/ 05 октября 2011
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]])]
0 голосов
/ 08 августа 2018

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

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
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...