在 Python 中转换列表输出

Question

我有一个函数，它可以读取 .csv 文件并将其存储在数组中。

def read_csv(self, filename, delimiter = ',', quotechar = '"'):
    reader = csv.reader(open(filename, 'rb'), delimiter = delimiter, quotechar = quotechar)
    # read first line and extract its data 
    self.column_headings = np.array(next(reader))
    # read subsequent lines
    rows = []
    for row in reader:
        rows.append(row)
    self.data = np.array(rows)
    self.m, self.n = self.data.shape

我只是试图读取一个 .tsv 文件，以便它以相同的形式返回。到目前为止我有这个：

traindata = np.array(p.read_table('train.tsv'))[:,2]

但是，当我尝试调用时：

m, n = traindata.data.shape

# Display
print m, n, traindata.column_headings

我得到了错误：

---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
<ipython-input-14-1f877ccb37b5> in <module>()
----> 1 m, n = traindata.data.shape

AttributeError: 'buffer' object has no attribute 'shape'

是什么导致了这个问题，我该如何解决？

Answer 1

您显式创建了list 的traindata：

traindata = list(np.array(p.read_table('train.tsv'))[:,2])
          # ^ here

如果您想将其用作numpy.array，请删除list() 调用：

traindata = np.array(p.read_table('train.tsv'))[:,2]

其次，您想要array 的shape，而不是它的data：

m, n = traindata.shape