Creating an array based on a plot of custom function (Python)

Question

I'm trying to use Numpy to create a y vector that will correspond to the following plot:

The x values will run from 0 to 24, the y values should be:

0 to 6 will be 0

6 to 18 will be sort of parabola

18 to 24 will be 0 again

What is a good way to do it? I don't have any practical ideas yet (I thought about some sort of interpolation).

Thank you!

Ferran Parés · Accepted Answer · 2018-05-04 09:52:23Z

1

I have done it assuming that you want a circle shape instead of a parabola (based on your scheme).

import numpy as np

length = 24
radius = 6

x = np.arange(length)
y = np.sqrt(radius**2-(x-(length/2))**2)
y = np.nan_to_num(y)

print(x)
# [ 0  1  2  3  4  5  6  7  8  9 10 11 12 13 14 15 16 17 18 19 20 21 22 23]
print(y)
# [0.         0.         0.         0.         0.         0.
#  0.         3.31662479 4.47213595 5.19615242 5.65685425 5.91607978
#  6.         5.91607978 5.65685425 5.19615242 4.47213595 3.31662479
#  0.         0.         0.         0.         0.         0.        ]

answered May 4, 2018 at 9:52

Ferran Parés

6363 silver badges12 bronze badges

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

3 Comments

ValientProcess Over a year ago

Not exactly the thing I had in mind but it works! :)

Ferran Parés Over a year ago

What did you had in mind? I am just deducing y points from circle mathematical equation: (x-x_o)**2+(y-y_o)**2 = r**2. If you prefer another shape you can just modify the equation.

ValientProcess Over a year ago

Yes it is working. What I had in mind is doing something more general, like part of a sine wave that will peak at the middle and zero padding everywhere else

Collectives™ on Stack Overflow

Creating an array based on a plot of custom function (Python)

1 Answer 1

3 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

3 Comments

Your Answer

Sign up or log in

Post as a guest

Related