Using numpy.delete() or any other function to delete from a list of lists/arrays

Question

I have the following list, let's call it R:

[(array([[1, 2, 3],
         [4, 5, 6],
         [7, 8, 9]]),
  array([100, 101, 102])),
 (array([[10, 11, 12],
         [13, 14, 15],
         [16, 17, 18]]),
  array([103, 104, 105]))]

I want to be able to delete columns of R in a for loop, based on an index i. For example, if i = 3, the 3rd column should be deleted, which should result in the following new, say R1:

    [(array([[1, 2],
             [4, 5],
             [7, 8]]),
      array([100, 101])),
     (array([[10, 11],
             [13, 14],
             [16, 17]]),
      array([103, 104]))]

I have zero experience with handling such multi dimensional arrays, so I am unsure how to use numpy.delete(). My actual list R is pretty big, so I would appreciate if someone can suggest how to go about the loop.

Mahdi F. · Accepted Answer · 2022-10-31 17:05:08Z

2

You can use np.delete with col==2 and axis=-1.

# if your 'list' be like below as you say in the question :
print(lst)
# [
#     array([[1, 2, 3],
#            [4, 5, 6],
#            [7, 8, 9]]), 
#     array([100, 101, 102]), 
#     array([[10, 11, 12],
#            [13, 14, 15],
#            [16, 17, 18]]),
#     array([103, 104, 105])
# ]

for idx, l in enumerate(lst):
    lst[idx] = np.delete(l, 2, axis=-1)

print(lst)

Output:

[
    array([[1, 2],
           [4, 5],
           [7, 8]]), 
    array([100, 101]), 
    array([[10, 11],
           [13, 14],
           [16, 17]]), 
    array([103, 104])
]

Creating input array like in the question:

import numpy as np

lst  = [[[1, 2, 3],
         [4, 5, 6],
         [7, 8, 9]],
        [100, 101, 102],
        [[10, 11, 12],
         [13, 14, 15],
         [16, 17, 18]],
        [103, 104, 105]
       ]

lst = [np.array(l) for l in lst]

Update base comment, If you have a tuple of np.array in your list, you can try like below:

lst = [
    (np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]]), np.array([100, 101, 102])), 
    (np.array([[10, 11, 12], [13, 14, 15], [16, 17, 18]]), np.array([103, 104, 105]))
]

for idx, tpl  in enumerate(lst):
    lst[idx] = tuple(np.delete(l, 2, axis=-1) for l in tpl)
    
print(lst)

Output:

[
    (array([[1, 2],
            [4, 5],
            [7, 8]]), 
     array([100, 101])
    ), 
    (array([[10, 11],
            [13, 14],
            [16, 17]]), 
     array([103, 104]))
]

edited Oct 31, 2022 at 17:05

answered Oct 30, 2022 at 17:46

Mahdi F.

24.1k5 gold badges25 silver badges32 bronze badges

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

7 Comments

S C Over a year ago

Can you please tell me how did you initialize the list in your code? I am getting the value error when I create a list like above and run your code: ValueError: could not broadcast input array from shape (3,3) into shape (3,).

Mahdi F. Over a year ago

@SC, edited answer and add part of how to create input array

S C Over a year ago

But isn't your lst different than the example in my question? I have two arrays in each bracket (), your code only has one.

Mahdi F. Over a year ago

@SC, I will work one it, give me one hour to become free.

Mahdi F. Over a year ago

@SC, edited answer, please read second part : Update base comment

|

Collectives™ on Stack Overflow

Using numpy.delete() or any other function to delete from a list of lists/arrays

1 Answer 1

7 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

7 Comments

Your Answer

Sign up or log in

Post as a guest