예제 #1
0
        rz[i] = 1000 * float(input())
        vx[i] = 1000 * float(
            input("inserisci le velocità iniziali vx, vy, vz in km/s: "))
        vy[i] = 1000 * float(input())
        vz[i] = 1000 * float(input())

if data == 'y':
    import solar
    M = solar.massasolare()
    for i in range(num):
        m[i] = solar.masse()[i]
        [rx[i], ry[i], rz[i]] = 1000 * solar.posizioni()[i]
        [vx[i], vy[i], vz[i]] = 1000 * solar.velocità()[i]
b = Planet(M, m, rx, ry, rz, vx, vy, vz, dth, num, N)

px = b.pos()[:, :, 0]
py = b.pos()[:, :, 1]
pz = b.pos()[:, :, 2]
name = input("scrivere il nome del file in cui salvare i dati delle orbite: ")
f = open(name, "w")
f.write(str(b.pos()))
f.close()

s = input("vuoi creare un .mp4? [y/n]: ")
if s == 'n':
    sys.exit()
if s == 'y':
    fig = plt.figure()
    ax = fig.add_subplot(111, projection='3d')
    if num == 1:
        ax.set_xlim3d([min(px[:, 0]) - 1000, max(px[:, 0]) + 1000])