Adding all the rows of an array after formating Numpy Python

Question

How can I add up all the elements of each row of a multidimensional numpy array. I am trying to turn the '--' elements to 0 and then adding all the rows right after. How could I do such a thing? The -- elements of the result of the masked array.

Array = np.array([[--, --, --, --, --, --, --]
 [3, 4, --, --, --, --, --]
 [--, --, 5, 7, 8, 10, --]
 [--, --, --, --, --, --, --]
 [--, --, --, --, --, --, 20]])
np.where(Array != "--", result, 0)
Array.sum(axis=0)

Expected Output:

[0 7 30 0 20]

It was actually after formatting an array it gives out -- for the empty spaces in the Array. — ege Selcuk
– ege Selcuk, Commented Mar 11, 2021 at 21:27
Array = np.array([[--, --, --, --, --, --, --] [3, 4, --, --, --, --, --] [--, --, 5, 7, 8, 10, --] [--, --, --, --, --, --, --] [--, --, --, --, --, --, 20]]) should throw an error. — QWERTYL
– QWERTYL, Commented Mar 11, 2021 at 21:28
Is this a masked array or a mixed of string and numbers array (which is an object array)? It would help if you provide how you get this array. Thank you — Ehsan
– Ehsan, Commented Mar 11, 2021 at 21:29

matheburg · Accepted Answer · 2021-03-11 21:45:39Z

3

import numpy as np

array = np.ma.array(
    [
        [0, 0, 0, 0, 0, 0, 0],
        [3, 4, 0, 0, 0, 0, 0],
        [0, 0, 5, 7, 8, 10, 0],
        [0, 0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0, 20]
    ],
    mask=[
        [1, 1, 1, 1, 1, 1, 1],
        [0, 0, 1, 1, 1, 1, 1],
        [1, 1, 0, 0, 0, 0, 1],
        [1, 1, 1, 1, 1, 1, 1],
        [1, 1, 1, 1, 1, 1, 0]
    ]
)

array = array.filled(0)
print(np.sum(array, axis=1))

Alternatively, if you want to apply the sum on the masked array before filling with zeros:

array.fill_value = 0
print(np.ma.sum(array, axis=1).filled(0))

edited Mar 11, 2021 at 21:45

answered Mar 11, 2021 at 21:28

matheburg

2,2011 gold badge23 silver badges48 bronze badges

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

Comments

Pablo C · Accepted Answer · 2021-03-11 21:29:18Z

0

You can do this

>>> np.where(arr == '--', 0, arr).astype(int).sum(1)
array([ 0,  7, 30,  0, 20])

answered Mar 11, 2021 at 21:29

Pablo C

4,7612 gold badges10 silver badges26 bronze badges

1 Comment

ege Selcuk Over a year ago

This does not work since the -- elements are that of a masked array, sorry updated the issue.

Collectives™ on Stack Overflow

Adding all the rows of an array after formating Numpy Python

2 Answers 2

Comments

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related