Python将值行解压缩为两个数组

Question

我正在编写一个 Python 脚本，逐行读取输入文件，每个文件包含 20 个值。我正在尝试将第一个存储在一维数组中，将其他 19 个存储在第二个数组中。

它们被定义为：

x = numpy.zeros((n_lines))
y = numpy.zeros((n_lines,19))

n_lines 输入文件中的行数。 我正在阅读它

for i,line in enumerate(open(infile)):
    x[i],y[i] = line.split(' ')

但我遇到了Too many values to unpack error

我现在将它存储在一个大数组中，然后将其分成两个。我没有使用line.split(' ')[0],line.split(' ')[1:]，因为它看起来并没有针对拆分两次进行优化。

有没有一种简洁的（Pythonista）解包方式？

（如果重要的话，我使用的是 Python 2.7）

Answer 1

你正在做最后一部分（创建一个数组然后对其进行切片），但第一部分应该使用numpy.genfromtxt() 完成，如下所示：

big_array = numpy.genfromtxt(infile) ## probably with more arguments
x, y = numpy.split(big_array, [1])

如果需要更有效的解决方案，替代方案是：

big_array = numpy.empty((nlines, 20))

for i,line in enumerate(open(infile)):
    big_array[i,:] = line.split()

x, y = numpy.split(big_array, [1])

最后，直接从行解包到之前的 x 和 y 数组可以这样完成：

splitted = line.split();
x[i], y[i,:] = splitted[0], splitted[1:]