Например, у меня есть 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]])]
Есть ли лучшие пути, вместо того чтобы пройти его?Благодаря.