Говорю сразу, я в программировании в целом почти что новичок, и опыт с питоном имею чуть больший, чем сдающие с ним ЕГЭ
Ваша задача для опытных программистов с релевантным опытом.
Библиотека
matplotlib . Библиотека имеет несколько уровней, так вот верхнем уровнем довольно легко пользоваться, у них есть галерея в том числе и 3d графики с готовыми примерами. Но я не уверен что ваш кейс возможно решить на верхнем уровнем библиотеки, а вот использование более низких уровней в matplotlib штука не простая.
