ПОМОГИТЕ ПОЖАЛУЙСТА!!! ПИТОН

Помогите определить общую массу найденных окаменелостей динозавров. Иногда целого динозавра в музее собирают из костей разных представителей одного вида, поэтому и массу записывают для отдельно найденных частей: голова, корпус, передние лапы, задние лапы и хвост. Поскольку вы - археолог, то массу одной из костей вводите вручную. Все эти показатели хранятся в виде векторов. Ваша первая задача - найти общую массу костей динозавров (стегозавра, тиранозавра и диплодока), если масса отдельных частей задана векторами Stegosaurus, Tyrannosaur и Diplodocus. Масса выражена в сотнях килограмм.

import numpy as np

x=int(input())

Stegosaurus = np.array([2, 12, 3, x, 5])

Tyrannosaur = np.array([5, 23, 4, 18, 15])

Diplodocus = np.array([7, 20, 10, 13, 12])

museum = np.array([[10, 12, 4, 15, 5],[25, 13, 18, 17, 22],

[17, 23, 19, 13, 11],[26, 25, 16, 14, 31]])

new = np.array([[2, 3, 3, 1, 4],[5, 2, 4, 4, 7],[2, 7, 6, 8, 3],[6, 1, 6, 4, 8]])

Sample Input:

6

Sample Output:

155

ответ по ссылке: https://stepik.org/lesson/562364/step/10