Divide array into subarray

Question

I want to create two sub-arrays from this array:

 a = [0, 1, 2, 3, 4, 5, 6]

This array will not always contain the same number of elements because it depends on the user input.

For example, in some occasions it'll be:

a = [0, 5]

or:

a = [5, 6, 4]

I want to divide the array into two subarrays. The first one will contain numbers from 1 to 4 (inclusive) and the second one will contain 0, 5 and 6.

In the first example, it will be:

a = [0, 1, 2, 3, 4, 5, 6]
sub_array1 = [1, 2, 3, 4]
sub_array2 = [0, 5, 6]

In the second:

a = [0, 5] 
sub_array1 = []
sub_array2 = [5]

In the third:

a = [5, 6, 4]
sub_array1 = [4]
sub_array2 = [5, 6]

and so on, depending on the user input.

How can I do this?

xlembouras · Accepted Answer · 2015-10-10 16:58:59Z

3

First thing that comes to mind is Enumerable#partition.

sub_array1, sub_array2 = [0,1,2,3,4,5,6].partition {|x| (1..4).include? x }
=> [[1,2,3,4], [0,5,6]]

if you have two conditions (I mean if 0,5,6 are an actual condition and not the excluded set) I think that a double iteration wouldn't hurt

a = [0,1,2,3,4,5,6]
sub_array1 = a.select { |x| (1..4).include? x }
sub_array2 = a.select { |x| [0,5,6].include? x }

edited Oct 10, 2015 at 16:58

answered Oct 10, 2015 at 15:32

xlembouras

8,3254 gold badges35 silver badges44 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

Cherenkov Over a year ago

Yep! This works! Basically, because later I need to determine the lenght of each subarray. Thanks!

Cary Swoveland Over a year ago

Mr. X, I think you can improve your (good) answer by prepending sub_array1, sub_array_2 = and deleting "but" and all that follows. Also, small typo: you forgot the zero.

the Tin Man · Accepted Answer · 2015-10-10 16:43:55Z

1

You can try something like this:

[0,1,2,3,4,5,6].group_by{|x| [0,5,6].include? x}

The result will be a hash:

{true=>[0, 5, 6], false=>[1, 2, 3, 4]}

In the second case:

[0,5].group_by{|x| [0,5,6].include? x}

The result will be:

{true=>[0, 5]}

edited Oct 10, 2015 at 16:43

the Tin Man

161k44 gold badges222 silver badges308 bronze badges

answered Oct 10, 2015 at 15:26

davidrac

10.8k3 gold badges45 silver badges71 bronze badges

2 Comments

Cherenkov Over a year ago

This is lovely.. but I don't think it suits my case.

Cary Swoveland Over a year ago

I believe you can meet the OP's objection by tacking on .values and prepending with arr1, arr2 =.

Collectives™ on Stack Overflow

Divide array into subarray

2 Answers 2

2 Comments

2 Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

2 Comments

2 Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related