python - 删除 numpy 数组末尾的 dtype

Question

我正在编写一种从数据文件创建数组的方法。该方法如下所示:

import numpy
def readDataFile(fileName):
    try:
        with open(fileName, 'r') as inputs:
            data = None
            for line in inputs:
                line = line.strip()
                items = line.split('\t')
                if data == None:
                    data = numpy.array(items[0:len(items)]) 
                else:
                    data = numpy.vstack((data, items[0:len(items)]))
                return numpy.array(data)
    except IOError as ioerr:
        print 'IOError: ', ioerr
        return None

我的数据文件包含几行数字，每一行之间用制表符分隔，例如:

1 2 3
4 5 6
7 8 9

我希望收到如下数组:

array([[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9]])

但是，结果的末尾包含 dtype:

array([[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9]], dtype='|S9')

因此，我无法对结果执行某些操作，例如如果我尝试使用 result.max(0) 查找每一行的最大值，我将收到一个错误:

TypeError: cannot perform reduce with flexible type.

那么，谁能告诉我我的代码有什么问题以及如何修复它？非常感谢。

Answer 1

Numpy 数组包含一个方法来完成这项工作:

import numpy as np
a = np.array(['A', 'B'])
a
# Returns: array(['A', 'B'],  dtype='|S1')

a.tolist()
# Returns ['A', 'B']

http://docs.scipy.org/doc/numpy/reference/generated/numpy.ndarray.tolist.html#numpy.ndarray.tolist