49 lines
No EOL
1.1 KiB
Python
49 lines
No EOL
1.1 KiB
Python
import matplotlib.pyplot as plt
|
|
import numpy as np
|
|
import math
|
|
import sys
|
|
|
|
from core.utils.cache import random_cache_path
|
|
from utils import do_something
|
|
from simpleeval import simple_eval
|
|
|
|
do_something()
|
|
|
|
def generate_function(expr):
|
|
x_range = (-10, 10)
|
|
resolution = 100
|
|
x = np.linspace(x_range[0], x_range[1], resolution)
|
|
expression = expr.split()
|
|
|
|
math_functions = {name: getattr(math, name) for name in dir(math) if not name.startswith('_')}
|
|
|
|
for exp in expression:
|
|
if exp.isnumeric():
|
|
y = np.full_like(x, float(exp))
|
|
else:
|
|
y = np.array([simple_eval(exp, functions=math_functions, names={'x': xi}) for xi in x])
|
|
label = f'f(x)={exp}'
|
|
|
|
plt.plot(x, y, label=label)
|
|
|
|
plt.xlabel('x')
|
|
plt.ylabel('y')
|
|
plt.title('')
|
|
plt.grid(True)
|
|
plt.legend()
|
|
|
|
path = random_cache_path() + '.png'
|
|
plt.savefig(path)
|
|
plt.close()
|
|
|
|
return path
|
|
|
|
expr = ''.join(sys.argv[1:])
|
|
|
|
try:
|
|
path = generate_function(expr)
|
|
|
|
sys.stdout.write('Result ' + str(path))
|
|
except Exception as e:
|
|
sys.stdout.write(f'Failed {str(e)}')
|
|
sys.exit() |