Как организовать массив python с numpy zeros

Question

Работает на простой модели SIR. Мне нужно упорядочить значения, напечатанные на терминале, чтобы они были организованы таким образом, чтобы мой учитель мог легко следовать. Мне сказали, чтобы посмотреть в numpy.zeros, но я новичок в python, и мне нужен кто-то, чтобы помочь.

import matplotlib.pyplot as plt 
import numpy 
beta = 0.24 
gamma = 0.142857 
Tstart = 0 
Tend = 151 
r = 0 
s = (306.8 * 10**6) 
i = (22 * 10**6) 

def compute_next_day(t,R,I,S): 
    R[t] = gamma * I[t - 1] + R[t - 1] 
    I[t] = (beta * I[t-1] * S[t-1]/(r+i+s)) - gamma * I[t-1] + I[t-1] 
    S[t] = - (beta * I[t-1] * S[t-1]/(r+i+s)) + S[t-1] 
    print S[t-1], I[t-1], R[t-1] 


def compute_entire_period(Tstart, Tend, R, I, S): 
    R[Tstart] = r 
    I[Tstart] = i 
    S[Tstart] = s 
    for t in range(Tstart + 1, Tend): 
     compute_next_day(t, R, I, S) 


R = range(Tstart, Tend) 
I = range(Tstart, Tend) 
S = range(Tstart, Tend) 

def graph(R, I, S): 
    plt.plot(R) 
    plt.plot(I) 
    plt.plot(S) 

plt.ylabel("Population") 
plt.xlabel("Time") 
plt.show() 

(compute_entire_period(Tstart, Tend, R, I, S)) 
(graph(R,I,S)) 

значение, напечатанное на терминал являются S, I, значения R ...

what currently appears on terminal want to organize like this

Answer 1

я не достаточно очков, чтобы комментировать, так что я буду «ответ »на ваш вопрос в разделе комментариев вы можете экспортировать значение в csv следующим образом:

import csv 

a = range(2, 10) 
with open("output.csv", "wb") as file: 
    writer = csv.writer(file) 
    writer.writerows(a)