0

我已经创建了一个 numpy.recarray 的子类。该类的目的是为记录数组提供漂亮的打印，同时保持记录数组的功能。

这是代码：

import numpy as np
import re

class TableView(np.recarray):

    def __new__(cls,array):
        return np.asarray(array).view(cls)

    def __str__(self):
        return self.__unicode__().encode("UTF-8")

    def __repr__(self):
        return self.__unicode__().encode("UTF-8")

    def __unicode__(self):

        options = np.get_printoptions()
        nrows = len(self)

        print "unicode called"

        if nrows > 2*options['edgeitems']:
            if nrows > options['threshold']:
                nrows = 2*options['edgeitems'] + 1

        ncols = len(self.dtype)

        fields = self.dtype.names

        strdata = np.empty((nrows + 1,ncols),dtype='S32')
        strdata[0] = fields

        np_len = np.vectorize(lambda x: len(x))
        maxcolchars = np.empty(ncols,dtype='i4')

        for i, field in enumerate(fields):
            strdata[1:,i] = re.sub('[\[\]]','',np.array_str(self[field])).split()
            maxcolchars[i] = np.amax(np_len(strdata[:,i]))

        rowformat = ' '.join(["{:>%s}" % maxchars for maxchars in maxcolchars])
        formatrow = lambda row: (rowformat).format(*row)
        strdata = np.apply_along_axis(formatrow,1,strdata)

        return '\n'.join(strdata)

这是它的打印方式：

In [3]: x = np.array([(22, 2, -1000000000.0, 2000.0), (22, 2, 400.0, 2000.0),
   ...:  (22, 2, 500.0, 2000.0), (44, 2, 800.0, 4000.0), (55, 5, 900.0, 5000.0),
   ...:  (55, 5, 1000.0, 5000.0), (55, 5, 8900.0, 5000.0),
   ...:  (55, 5, 11400.0, 5000.0), (33, 3, 14500.0, 3000.0),
   ...:  (33, 3, 40550.0, 3000.0), (33, 3, 40990.0, 3000.0),
   ...:  (33, 3, 44400.01213545, 3000.0)],  
   ...:           dtype=[('subcase', '<i4'), ('id', '<i4'), ('vonmises', '<f4'), ('maxprincipal', '<f4')])

In [6]: TableView(x)
unicode called
Out[6]: 
subcase id        vonmises maxprincipal
     22  2 -1.00000000e+09        2000.
     22  2  4.00000000e+02        2000.
     22  2  5.00000000e+02        2000.
     44  2  8.00000000e+02        4000.
     55  5  9.00000000e+02        5000.
     55  5  1.00000000e+03        5000.
     55  5  8.90000000e+03        5000.
     55  5  1.14000000e+04        5000.
     33  3  1.45000000e+04        3000.
     33  3  4.05500000e+04        3000.
     33  3  4.09900000e+04        3000.
     33  3  4.44000117e+04        3000.

但是当我只打印一行时这不起作用：

In [7]: TableView(x)[0]
Out[7]: (22, 2, -1000000000.0, 2000.0)

它适用于多行：

In [8]: TableView(x)[0:1]
unicode called
Out[8]: 
subcase id        vonmises maxprincipal
     22  2 -1.00000000e+09        2000.

经进一步调查：

In [10]: type(TableView(x)[0])
Out[10]: numpy.record

In [11]: type(TableView(x)[0:1])
Out[11]: __main__.TableView

如何使 TableView 的 numpy.record 具有相同的unicode？

4

1 回答 1

1

你的诊断是对的。此类的单个元素是 a record，而不是Tableview数组。

使用切片或列表进行索引，[0:1]或者[[0]]，是直接的解决方案。

尝试子类化np.record和更改元素Tableview似乎很复杂。

您可以尝试自定义__getitem__方法。这在您索引数组时调用。

它以：

    else:
        # return a single element
        return obj

修改后的版本可能会返回单个元素 Tableview

        return Tableview([obj])

但这可能会产生某种无休止的递归，使您无法将元素作为常规记录访问。

否则，您可能只想使用切片索引。

于 2016-09-06T17:00:55.253 回答