python - 从记录 ndarray 中选择行范围

Question

我使用 CSV 文件获取了 NumPy 记录 ndarray

data = matplotlib.mlab.csv2rec('./data.csv', delimiter=b',')

数据集的结构如下:

      date,a0,a1,a2,a3, b0, b1, b2, b3,[...], b9
2012-01-01, 1, 2, 3, 4,0.1,0.2,0.3,0.4,[...],0.9

我想从数组中选择(在 SQL 意义上)列 b0 到 b9，给出结构

 b0, b1, b2, b3,[...], b9
0.1,0.2,0.3,0.4,[...],0.9

问题“How can I use numpy array indexing to select 2 columns out of a 2D array to select unique values from?”类似，但按照建议切片data[:,5:]会抛出IndexError:索引过多并带有记录数组。

Answer 1

鉴于您有一个记录数组，我认为以下方法可行:

data[['b' + str(j) for j in range(10)]]

doc/introduction和 cookbook