是否可以在numpy.genfromtxt输出中添加新字段?

时间:2016-10-21 17:36:41

标签: python numpy genfromtxt

我加载了一个csv文件并使用标题来指定每列的名称。

# Load the Data
data = np.genfromtxt('dat_h.csv',
                     delimiter=',',
                     names=True)

这很棒,因为我可以按名称访问列。例如......

DATES = data['Dates']
Temperature = data['Temp']

假设我有一个与这些测量相匹配的压力观测矢量。我可以使用包含压力变量的新字段附加数据结构吗?

我想做这样的事......

data.append('Pressure',pressure_vector)
# and then be able to access that field like I do the other fields
data['Pressure']

1 个答案:

答案 0 :(得分:2)

查看this answer。以下是docs的重复功能。

主要是我认为你需要的是:

from numpy.lib.recfunctions import append_fields

append_fields(data, 'Pressure', pressure_vector, np.double)