python - Python IndexError：列表分配索引超出范围只是试图创建一个相空间

Question

import matplotlib.pyplot as plt
import numpy as np
import scipy as sc
import math


t,theta1=np.loadtxt('Single Small Angle 1.txt',unpack=True,skiprows=2)
t2,theta2=np.loadtxt('Single Small Angle 3.txt',unpack=True,skiprows=2)
theta=[]

omega=np.arange(int(len(theta1)/5)-1)
for x in range (int(len(theta1)/5-1)): 
    omega[x]=(theta1[x*5]-theta1[(x+1)*5])/.005
    theta[x]=theta1[x*5]


plt.plot(theta1,omega)
plt.xlabel("${\Theta}$ [rad]")
plt.ylabel("${\Omega}$ [rad/s]")
plt.title("Small Angle Approximation Phase Space")
plt.show()

Traceback (most recent call last):
    theta[x]=theta1[x*5]
IndexError: list assignment index out of range
[Finished in 0.6s with exit code 1]

我不知道我在做什么，我只想修复错误。我只是想创建一个相位空间，我需要我的 theta1 东西的导数，所以我可以有 d(theta1)/dt。

Answer 1

下面应该工作 - 你的问题是没有用适当的长度初始化 theta。您同样可以插入该行omega=np.arange(int(len(theta1)/5)-1)。这会将两者都创建为列表，然后附加到它们，我认为它读起来更好一些。

theta, omega = [],[]
for x in range (int(len(theta1)/5-1)): 
  omega.append((theta1[x*5]-theta1[(x+1)*5])/.005)
  theta.append(theta1[x*5])

omega = np.array(omega)
theta = np.array(theta)