python find_pair.py

Posted 2021-05-09

import numpy as np
import matplotlib.pyplot as plt
def f1(x,y):
    return x**y-y**x
def f2(x,y):
    return x-y
x0=int(raw_input("x0 = ") or '0') 
xlast=int(raw_input("xlast = ") or '100')
xwindow=int(raw_input("xwindow = ") or '1')
y0=int(raw_input("y0 = ") or '0')
ylast=int(raw_input("ylast = ") or '100')
ywindow=int(raw_input("ywindow = ") or '1')
x=np.arange(x0,xlast,xwindow)
y=np.arange(y0,ylast,ywindow)
X,Y=np.meshgrid(x,y)
Z1=f1(X,Y)
Z2=f2(X,Y)
Z=Z2-Z1
a,b=np.where(Z == 0)
result=np.column_stack((a,b))
np.savetxt('result.txt',result,fmt='%d')
plt.contour(X,Y,Z2)
plt.contour(X,Y,Z1)
plt.show()