python - 如何通过 numpy loadtxt 获取标签？

Question

我有一个格式为

的数据文件

Col0   Col1  Col2
2015   1     4
2016   2     3

数据是 float ，我用numpyloadtext做了一个ndarray。但是，我需要跳过标签行和列以获得数据数组。如何在读取标签的同时从数据中生成 ndarray？

import numpy as np
import matplotlib.pyplot as plt

data = np.loadtxt("data.csv", skiprows=1) 
# I need to skip the first row in reading the data but still get the labels.
x= data[:,0]
a= data[:,1]
b= data[:,2]

plt.xlabel(COL0) # Reading the COL0 value from the file.
plt.ylabel(COL1) # Reading the COL1 value from the file.
plt.plot(x,a)

注意:标签(列标题)在脚本中是未知的。该脚本应该是通用的，可以处理具有相同结构的任何输入文件。

Answer 1

使用 genfromtxt 可以获取元组中的名称。您可以查询名称，并且可以使用 dtype.names[n] 将名称输出到变量中，其中 n 是一个索引。

import numpy as np
import matplotlib.pyplot as plt

data = np.genfromtxt('data.csv', names=True)

x = data[data.dtype.names[0]] # In this case this equals data['Col1'].
a = data[data.dtype.names[1]]
b = data[data.dtype.names[2]]

plt.figure()
plt.plot(x, a)
plt.xlabel(data.dtype.names[0])
plt.ylabel(data.dtype.names[1])
plt.show()