Python Numpy數組索引

Question

我在使用Numpy索引時遇到了一些小問題。當腳本給出三個不同數組的索引（腳本中的F_fit）時，該腳本只給出最後一個數組的索引。我相信這是一件簡單的事情，但我還沒有弄明白。該3_phases.txt文件中包含這3個行

1 -1 -1 -1 1 1 
1 1 1 -1 1 1 
1 1 -1 -1 -1 1 

下面是代碼：

import numpy as np 
import matplotlib.pyplot as plt 

D = 12.96 
n = np.arange(1,7) 

F0 = 1.0 
x = np.linspace(0.001,4,2000) 
Q = 2*np.pi*np.array([1/D, 2/D, 3/D, 4/D, 5/D, 6/D]) 
I = (11.159, 43.857, 26.302, 2.047, 0.513, 0.998)  
phase = np.genfromtxt('3_phases.txt') 

for row in phase: 

    F = (np.sqrt(np.square(n)*I/sum(I)))*row 
    d = sum(i*(np.sin(x*D/2+np.pi*j)/(x*D/2+np.pi*j))for i,j in zip(F,n)) 
    e = sum(i*(np.sin(x*D/2-np.pi*j)/(x*D/2-np.pi*j))for i,j in zip(F,n)) 
    f_0 = F0*(np.sin(x*D/2)/(x*D/2)) 
    F_cont = np.array(d) + np.array(e) + np.array(f_0) 
    plt.plot(x,F_cont,'r') 
    #plt.show() 
    plt.clf() 

D2 = 12.3 
I2 = (9.4, 38.6, 8.4, 3.25, 0, 0.37) 
Q2 = 2*np.pi*np.array([1/D2, 2/D2, 3/D2, 4/D2, 5/D2, 6/D2]) 
n2 = np.arange(1,7) 

for row in phase: 
    F2 = (np.sqrt(np.square(n2)*I2/sum(I2)))*row 
    plt.plot(Q2,F2,'o') 
    #plt.show() 
    F_data = F2 
    Q_data = Q2 
    I_data = np.around(2000*Q2/(4-0.001)) 
    I_data = np.array(map(int,I_data)) 
    F_fit = F_cont[I_data] 
    print F_fit 
    R2 = (1-(sum(np.square(F_data-F_fit))/sum(np.square(F_data-np.mean(F_data))))) 

任何幫助，將不勝感激。

Answer 1

您每次通過第一個循環時都重新定義了F_cont。當您進入第二個循環（所有的_2值）時，您只能訪問最近的row的F_cont。

要解決這個問題，請將您的_2定義移到第一個循環上方，並且只循環一次，然後您將可以訪問每個F_cont，並且您的打印輸出將有所不同。

以下代碼與您的相同，除了上述重新排列，以及我從上面實施了我的評論（在您的Q中使用n/D）。

import numpy as np 
import matplotlib.pyplot as plt 

D = 12.96 
n = np.arange(1,7) 

F0 = 1.0 
x = np.linspace(0.001,4,2000) 
Q = 2*np.pi*n/D 
I = (11.159, 43.857, 26.302, 2.047, 0.513, 0.998)  
phase = np.genfromtxt('3_phases.txt') 

D2 = 12.3 
I2 = (9.4, 38.6, 8.4, 3.25, 0, 0.37) 
Q2 = 2*np.pi*n/D2 
n2 = np.arange(1,7) 

for row in phase: 

    F = (np.sqrt(np.square(n)*I/sum(I)))*row 
    d = sum(i*(np.sin(x*D/2+np.pi*j)/(x*D/2+np.pi*j))for i,j in zip(F,n)) 
    e = sum(i*(np.sin(x*D/2-np.pi*j)/(x*D/2-np.pi*j))for i,j in zip(F,n)) 
    f_0 = F0*(np.sin(x*D/2)/(x*D/2)) 
    F_cont = np.array(d) + np.array(e) + np.array(f_0) 
    plt.plot(x,F_cont,'r') 
    plt.clf() 

    F2 = (np.sqrt(np.square(n2)*I2/sum(I2)))*row 
    plt.plot(Q2,F2,'o') 
    F_data = F2 
    Q_data = Q2 
    I_data = np.around(2000*Q2/(4-0.001)) 
    I_data = np.array(map(int,I_data)) 
    F_fit = F_cont[I_data] 
    print F_fit 
    R2 = (1-(sum(np.square(F_data-F_fit))/sum(np.square(F_data-np.mean(F_data))))) 

Answer 2

F_fit正在計算從I_data，而這又是從Q2計算。 Q2設置在循環外部，並且不取決於row - 也許您的意思是I_data是F2的函數？