Построить прямые в Matplotlib и конформно отобразить их

%matplotlib inline
import matplotlib.pyplot as plt
import math, cmath
 
# построим прямые функции y = a*x, x=[-1,1], a=[0,1]
 
 
x = [-1, -0.5, -0.2, 0.4, 0.7, 1]
y = [xx*0.1 for xx in x]
y2 = [xx*0.3 for xx in x]
y3 = [xx*0.5 for xx in x]
y4 = [xx*0.7 for xx in x]
y5 = [xx*0.9 for xx in x]
y6 = [xx*1 for xx in x];
 
plt.plot(x,y)
plt.plot(x,y2)
plt.plot(x,y3)
plt.plot(x,y4)
plt.plot(x,y5)
plt.plot(x,y6);

y = []
for a in xrange(1,10):
 y.append([xx*a/10 for xx in x])

plt.plot(x,[xx*a/10 for xx in x])

y.append([xx*a/10 for xx in x])

%matplotlib inline
import matplotlib.pyplot as plt
import math, cmath
 
def conform(f, n = 10, m = 100,
           figsize=(6,6), dpi=300):
    '''
    Функция, рисующая комформное отображение с помощью f(z)
    n- число разбиений единичного квадрата
    m - число точек на линии
    figsize - размер рисунка в дюймах 
    dpi - пространственное разрешения в пикселях на дюйм
    '''
    fig = plt.figure(figsize = figsize, dpi = dpi)
    plt.title('конформное отображение')
    
    # рисуб линии по горизонтали
    for i in range (n+1):
        yy = i/n
        # формирую массивы
        x = [j/m for j in range(m+1)]
        y = [yy for j in range(m+1)]
        # массив комплексных чисел
        z = [ complex(x[j], y[j]) for j in range(m+1)]
        # отображение
        w = [f(z[j]) for j in range(m+1)]
        # делаю массивы с действительной и мнимой частью комплексного числа
        u = [w[j].real for j in range(m+1)]
        v = [w[j].imag for j in range(m+1)]
        plt.plot(u,v, 'r-')
        
# вертикальные линии
    for i in range (n+1):
        xx = i/n
        w = [f(complex(xx, j/m)) for j in range(m+1)]
        u = [w[j].real for j in range(m+1)]
        v = [w[j].imag for j in range(m+1)]
        #print(u,v)
        plt.plot(u,v, 'b-')
    
conform(f= lambda z:z**2, m=10) # вызываю функцию

%matplotlib inline
import matplotlib.pyplot as plt
import math, cmath
 
# построим прямые функции y = a*x, x=[-1,1], a=[0,1]
 
# размер холста
fig = plt.figure(figsize = (6, 6), dpi = 96)
 
# оси
plt.plot([-1,1], [0,0], color='black', lw=2) # ось x
plt.plot([0,0], [-1,1], color='black', lw=2) # ось y
 
x = [-1, -0.5, -0.2, 0.4, 0.7, 1]
y = []
for a in range(1, 10):
    plt.plot(x,[xx*a/10 for xx in x], 'r-')

def conform(f, n):
    '''
    Конформное отображение с помощью функции f(z)
    Сторона квадрата делится на n частей
    '''
    # вверх - вправо - вниз - влево
    zeros = [0]*(n+1)
    ones = [1]*(n+1)
    seq1 = [i/n for i in range(n+1)]
    seq2 = [1-i/n for i in range(n+1)]
    
    xx = zeros + seq1 + ones + seq2
    yy = seq1 + ones + seq2 + zeros
    
    plt.plot(xx, yy, 'b-')
    
    
    # конформное отображение
    m = len(xx)
    # комплексные числа из xx+jyy
    zz = [complex(xx[i], yy[i]) for i in range(m)]
    # отображение
    ww = [f(z) for z in zz]
    uu = [w.real for w in ww]
    vv = [w.imag for w in ww]
    plt.plot(uu, vv, 'r-')
    
 
plt.figure(figsize=(4,4))    
conform(lambda z:z**2, 10)