5

我正在尝试加载如下所示的数据集：

Algeria,73.131000,6406.8166213983,0.1
Angola,51.093000,5519.1831786593,2
Argentina,75.901000,15741.0457726686,0.5
Armenia,74.241000,4748.9285847709,0.1

等等。最后，我只需要第 1 列和第 2 列。我不需要国家名称和最后一列。本质上，我需要提取两个维度为 nx1 的矩阵。我知道我需要指定数据类型：

data=np.loadtxt('file.txt',delimiter=',',dtype=[('f0',str),('f1',float),('f2',float),('f3',float)])

但是，这会产生一个元组列表，

array([('', 73.131, 6406.8166213983, 0.1),
   ('', 51.093, 5519.1831786593, 2.0),`

代替

array(['',73.131,6406.8166213983,0.1],
      ['',51.093, 5519.1831786593, 2.0],

错误在哪里？

4

3 回答 3

12

检查NumPy 的文档。

x, y = np.loadtxt(c, delimiter=',', usecols=(1, 2), unpack=True)

该usecols参数应该可以完成您的工作。

于 2013-11-04T12:02:36.760 回答
1

如果您只想要前两列，您可以使用genfromtxt：

import numpy as np
col1 = np.genfromtxt('yourfile.txt',usecols=(1),delimiter=',',dtype=None)
col2 = np.genfromtxt('yourfile.txt',usecols=(2),delimiter=',',dtype=None)

或两者一起：

np.genfromtxt('yourfile.txt',usecols=(1,2),delimiter=',',dtype=None)
于 2013-11-04T12:52:01.053 回答
1

您的“错误”是您设置了自己的 dtype。如果您不想要您设置的 dtype（我看不出您不想要它的原因），您可以使用 np.loadtxt() 的 skiprows 和 usecols 参数来仅加载您想要的列。

您的结果将是一个形状为 (n, 2) 的 NumPy 数组，而不是您认为应该拥有的 (n, 3) （其中 n 是您的行数）。

于 2013-11-04T12:02:04.747 回答