Question

我有一个文件，我需要使用第一列。其余列需要与第一列集成。让我们说我的文件看起来像这样：

100 1.0 1.1 1.2 1.3 0.9
110 1.8 1.9 2.0 2.1 2.2
120 1.8 1.9 2.0 2.1 2.2
130 2.0 2.1 2.3 2.4 2.5

我可以编写一段代码，该代码采用第二列并与第一列和第三列集成，并与第一列集成，依此类推？对于我的代码，我有：

import scipy as sp
first_col=dat[:,0] #first column from data file
cols=dat[:,1:] #other columns from data file

col2 = cols[:,0]  # gets the first column from variable cols
I = sp.integrate.cumtrapz(col2, first_col, initial = 0) #integration step

这仅适用于变量col的第一行，但是，我不想为所有其他列写出来，它会看起来讨论（想到它让我颤抖）。我已经看到了类似的问题，但是我们无法将答案与我的答案联系起来，而那些或多或少相同的答案却含糊不清。有什么想法吗？

Answer 1

函数cumtrapz接受axis参数。例如，假设您将第一列放在x中，其余列放在y中，并且它们具有以下值：

In [61]: x
Out[61]: array([100, 110, 120, 130])

In [62]: y
Out[62]: 
array([[ 1.1,  2.1,  2. ,  1.1,  1.1],
       [ 2. ,  2.1,  1. ,  1.2,  2.1],
       [ 1.2,  1. ,  1.1,  1. ,  1.2],
       [ 2. ,  1.1,  1.2,  2. ,  1.2]])

您可以将y的每一列与x进行整合，如下所示：

In [63]: cumtrapz(y, x=x, axis=0, initial=0)
Out[63]: 
array([[  0. ,   0. ,   0. ,   0. ,   0. ],
       [ 15.5,  21. ,  15. ,  11.5,  16. ],
       [ 31.5,  36.5,  25.5,  22.5,  32.5],
       [ 47.5,  47. ,  37. ,  37.5,  44.5]])

使用数据文件中的多个列

1 个答案: