处理数据文件中的多个列答案

【问题标题】：Working with multiple columns from a data file处理数据文件中的多个列
【发布时间】：2015-05-05 12:46:01
【问题描述】：

我有一个文件，我需要在其中使用第一列。其余列需要相对于第一列进行整合。假设我的文件如下所示：

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 的第一行，但是，我不想为所有其他列写出它，它看起来像是在讨论（想到它让我颤抖）。我见过类似的问题，但无法将答案与我的答案联系起来，而那些或多或少相同的答案却含糊不清。有什么想法吗？

【问题讨论】：

您可以使用genfromtxt，然后自己执行所有集成作为替代
哪些参数可以让我循环浏览这些列？
如果我理解你想要的正确，你想在多个列上应用相同的集成，所以data = sp.genfromtxt("file.txt") ; for col in data[0]: ; integrate(data[:,0], data[:,col]) 其中integrate 是一个 def 函数。（尽管答案似乎可以满足您的要求，而无需自己编写！）

标签： python arrays scipy multiple-columns data-files

【解决方案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]])

【讨论】：

这非常有效！我应该在自己使用 cumtrapz 后看到这个......