Splitting arrays in Python

Question

I have the following problem: I would like to find different "cuts" of the array into two different arrays by adding one element each time, for example:

If I have an array

a = [0,1,2,3]

The following splits are desired:

[0] [1,2,3]

[0,1] [2,3]

[0,1,2] [3]

In the past I had easier tasks so np.split() function was quite enough for me. How should I act in this particular case?

Many thanks in advance and apologies if this question was asked before.

Does this answer your question? Split an array in all possible combinations (not regular splitting). (You can limit to nsplits = 2). — Pierre D
– Pierre D, Commented Nov 25, 2022 at 17:53

Mahdi F. · Accepted Answer · 2022-11-25 17:53:37Z

2

Use slicing, more details : Understanding slicing.

a = [0,1,2,3]

for i in range(len(a)-1):
    print(a[:i+1], a[i+1:])

Output:

[0] [1, 2, 3]
[0, 1] [2, 3]
[0, 1, 2] [3]

answered Nov 25, 2022 at 17:53

Mahdi F.

24.1k5 gold badges25 silver badges32 bronze badges

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

Comments

Adrian Kurzeja · Accepted Answer · 2022-11-25 17:56:27Z

1

Check this out:

a = [0,1,2,3]

result = [(a[:x], a[x:]) for x in range(1, len(a))]

print(result)
# [([0], [1, 2, 3]), ([0, 1], [2, 3]), ([0, 1, 2], [3])]

# you can access result like normal list
print(result[0])
# ([0], [1, 2, 3])

answered Nov 25, 2022 at 17:56

Adrian Kurzeja

8931 gold badge12 silver badges29 bronze badges

Collectives™ on Stack Overflow

Splitting arrays in Python

2 Answers 2

Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related