Как разбить массив на два, если один из элементов - это нуль?

Question

Я использую Ruby 2.4. У меня есть массив, который я хочу разделить на две части, в которых первый массив будет либо буква «а» или «б», а второй массив будет все остальное, так что у меня есть эта

arr1, arr2 = data_col.map { |x| a, b, c = x.partition(/^[ab][[:space:]]+/i); [b.strip, a + c] }.transpose 

Однако , это не удается, если один из элементов в массиве равен нулю. Как мне объяснить это и заставить два элемента nil быть в результирующих массивах? То есть, если исходный массив

["A bcd", nil, "B 123"] 

результирующих массивов будет

["A", nil, "B"], ["bcd", nil, "123"] 

?

Answer 1

Явного проверить это

arr1, arr2 = data_col.map do |x| 
    if x 
    a, b, c = x.partition(/^[ab][[:space:]]+/i) 
    [b.strip, a + c] 
    else 
    [nil, nil] 
    end 
end.transpose 

Существует нет умного способа сделать здесь nil распространения.

---

И если ты крут с "" в результате чего элементы, попробуйте это

arr1, arr2 = data_col.map { |x| a, b, c = (x || "").partition(/^[ab][[:space:]]+/i); [b.strip, a + c] }.transpose 

Answer 2

Предполагая, не нулевые элементы всегда должны быть разделена на месте, вы можете избавиться от разделов и регулярных выражений тестирования и просто сделать это:

>> a = ["A bcd", nil, "B 123"] 
>> a.map { |e| e ? [e.split.first, e.split.last] : [nil, nil] }.transpose 
=> [["A", nil, "B"], ["bcd", nil, "123"]]