how do I use the array first element to do some math in python

Question

I have an

array([[5.1, 3.5, 1.4, 0.2],
       [4.9, 3. , 1.4, 0.2],
       [4.7, 3.2, 1.3, 0.2],
       [4.6, 3.1, 1.5, 0.2])

I want to get the sum of the equation:

(5.1 - 1)

(4.9 - 1)

(4.7 - 1)

(4.6 - 1)

How do I get the every arrays' first element?

Mark · Accepted Answer · 2022-10-11 15:47:50Z

1

Assuming this is a Numpy array, you can just subtract from the first column and let broadcasting do the work. If you want the sum of that result, just use sum():

import numpy as np

arr = np.array([
    [5.1, 3.5, 1.4, 0.2],
    [4.9, 3. , 1.4, 0.2],
    [4.7, 3.2, 1.3, 0.2],
    [4.6, 3.1, 1.5, 0.2]
])

a = arr[:, 0] - 1
#array([4.1, 3.9, 3.7, 3.6])

a.sum()
15.299999999999999

If you are bothered by the inexact sum, make sure you read Is floating point math broken?

edited Oct 11, 2022 at 15:47

answered Oct 11, 2022 at 15:42

Mark

92.7k8 gold badges116 silver badges156 bronze badges

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

Comments

user16836078 · Accepted Answer · 2022-10-11 15:56:27Z

0

There's an axis argument in array.sum that you can set to sum the array vertically.

(arr-1).sum(axis=0)
array([15.3,  8.8,  1.6, -3.2])

answered Oct 11, 2022 at 15:56

user16836078

Collectives™ on Stack Overflow

how do I use the array first element to do some math 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