https://github.com/hansalemaos/numpyslicesplit

Splits a numpy array or a list based on the given indices or ranges and returns the split arrays.
https://github.com/hansalemaos/numpyslicesplit

numpy split

Last synced: 12 days ago
JSON representation

Splits a numpy array or a list based on the given indices or ranges and returns the split arrays.

• Host: GitHub
• URL: https://github.com/hansalemaos/numpyslicesplit
• Owner: hansalemaos
• Created: 2024-02-06T01:18:42.000Z (10 months ago)
• Default Branch: main
• Last Pushed: 2024-02-06T01:18:44.000Z (10 months ago)
• Last Synced: 2024-10-12T00:22:18.714Z (about 1 month ago)
• Topics: numpy, split
• Language: Python
• Homepage: https://pypi.org/project/numpyslicesplit/
• Size: 16.6 KB
• Stars: 0
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.MD

Awesome Lists containing this project

README

        
# Splits a numpy array or a list based on the given indices or ranges and returns the split arrays.

### pip install numpyslicesplit

#### Tested against Windows 10 / Python 3.11 / Anaconda

### How to use it in Python 

```python

Parameters:

a (numpy array/list): The input array/list to be split.

splits (list): The indices or ranges to split the array on.

delete (bool, optional): If True, removes the specified indices from the split arrays,

if not it keeps them and deletes the others

Defaults to True.

Returns:

list: A list of numpy arrays split based on the given indices or ranges.

Example:

from numpyslicesplit import np_slice_split

a = np.arange(1000).reshape(100, 10)[..., 0]

splits = [(3, 5), (9, 14), (24, 30), (41, 43)]

s1 = np_slice_split(a, splits, delete=True)

print(f"{s1=}")

s2 = np_slice_split(a, splits, delete=False)

print(f"{s2=}")

splits = [3, 4, 5, 65, 7, 4, 6, 63, 2, 5, (0, 10)]

s1 = np_slice_split(a.tolist(), splits, delete=True)

print(f"{s1=}")

s2 = np_slice_split(a.tolist(), splits, delete=False)

print(f"{s2=}")

# s1=[array([ 0, 10, 20]), array([50, 60, 70, 80]),

#     array([140, 150, 160, 170, 180, 190, 200, 210, 220, 230]),

#     array([300, 310, 320, 330, 340, 350, 360, 370, 380, 390, 400]),

#     array([430, 440, 450, 460, 470, 480, 490, 500, 510, 520, 530, 540, 550,

#        560, 570, 580, 590, 600, 610, 620, 630, 640, 650, 660, 670, 680,

#        690, 700, 710, 720, 730, 740, 750, 760, 770, 780, 790, 800, 810,

#        820, 830, 840, 850, 860, 870, 880, 890, 900, 910, 920, 930, 940,

#        950, 960, 970, 980, 990])]

# s1=[[100, 110, 120, 130, 140, 150, 160, 170, 180, 190, 200, 210, 220, 230,

# 240, 250, 260, 270, 280, 290, 300, 310, 320, 330, 340, 350, 360, 370, 380

# , 390, 400, 410, 420, 430, 440, 450, 460, 470, 480, 490, 500,

# 510, 520, 530, 540, 550, 560, 570, 580, 590, 600, 610, 620],

#  [640], [660, 670, 680, 690, 700, 710, 720, 730, 740, 750, 760,

#  770, 780, 790, 800, 810, 820, 830, 840, 850, 860, 870, 880, 890,

#  900, 910, 920, 930,

# 940, 950, 960, 970, 980, 990]]

# s2=[[0, 10, 20, 30, 40, 50, 60, 70, 80, 90], [630], [650]]

```