如何将数组按照正负分割成多个数组

V2EX = way to explore

V2EX 是一个关于分享和探索的地方

现在注册

已注册用户请登录

这是一个创建于 2375 天前的主题，其中的信息可能已经有所发展或是发生改变。

array([ 0, 1, 2, 3, 2, 1, 0, -2, -3, -5, -7, -9, -8, -6, -5, -3, -1, 2, 3, 5, 7, 8, 6, 5, 7, 9, 11, 12, 14, 16, 13, 11, 9, 8, 7, 5, 4, 2, -1, -3, -6, -8, -11, -13, -16, -18]) 例如把以上数组划分成 4 个数组放进列表里 [array([ 0, 1, 2, 3, 2, 1, 0]),array([ -2, -3, -5, -7, -9, -8, -6, -5, -3, -1]),array([ 2, 3, 5, 7, 8, 6, 5, 7, 9, 11, 12, 14, 16, 13, 11, 9, 8, 7, 5, 4, 2]),array([ -1, -3, -6, -8, -11, -13, -16, -18])]

数组

array

列表里

正负

2 条回复 • 2018-06-22 04:59:15 +08:00

ipwx

2018-06-21 23:33:49 +08:00 via iPhone

np.where

msg7086

2018-06-22 04:59:15 +08:00

a = [0, 1, 2, 3, 2, 1, 0, -2, -3, -5, -7, -9, -8, -6, -5, -3, -1, 2, 3, 5, 7, 8, 6, 5, 7, 9, 11, 12, 14, 16, 13, 11, 9, 8, 7, 5, 4, 2, -1, -3, -6, -8, -11, -13, -16, -18]

r = []
a.each_with_index do |n, i|
r << [] if i == 0 || a[i] >= 0 && a[i-1] < 0 || a[i] < 0 && a[i-1] >= 0
r.last << n
end

r
=> [[0, 1, 2, 3, 2, 1, 0], [-2, -3, -5, -7, -9, -8, -6, -5, -3, -1], [2, 3, 5, 7, 8, 6, 5, 7, 9, 11, 12, 14, 16, 13, 11, 9, 8, 7, 5, 4, 2], [-1, -3, -6, -8, -11, -13, -16, -18]]