Как добавить первый элемент матрицы в список по циклу? - PullRequest
Новая
       21

Как добавить первый элемент матрицы в список по циклу?

0 голосов
/ 16 февраля 2019

У меня есть два цикла, которые выполняются для разных координат x и y, и для каждой разной (x, y) координаты решается линейное уравнение для силы 1 и силы 2 с использованием метода матриц, то есть нахождение инверсии A, если Ax= C. Для каждого цикла он дает ответ в виде матрицы, где первый элемент - сила 1, а 2-й элемент - сила 2 в этих конкретных координатах.Вот мой код: 

import numpy as np
from scipy import linalg
def Force():
    Force1 = np.zeros((160,90))
    Force2 = np.zeros((160,90))
    for x in np.arange(0,16.1,0.1):
        for y in np.arange(1,9.1,0.1):
            l1 = np.hypot(x,y)
            l2 = np.hypot(15-x,y)
            A = np.array([[(x/l1),((x-15)/l2)],[(y/l1),(y/l2)]])
            c = np.array([[0],[70*9.81]])

            F = linalg.solve(A,c)
            Force1[x,y] = F[0]
            Force2[x,y] = F[1]
            print("Force 1 = {} \nForce 2 = {}\n".format(F[0], F[1]))

, поэтому в каждой точке (x, y) решается матрица [[Force 1], [Force 2]].Теперь я хотел бы добавить все Force1 (s) в список Force1 [x, y] и аналогично для Forces2 (s), чтобы я мог сделать 

plt.imshow[Force1]
plt.imshow[Force2]

для построения 2 тепловых карт.Как мне это сделать?

...