将 numpy.recarray 转换为 dict 将条目分配给它们的基本类型以传递给 fastAPI答案

【问题标题】：Convert numpy.recarray to dict assigning entries to their fundamental types for passing to fastAPI将 numpy.recarray 转换为 dict 将条目分配给它们的基本类型以传递给 fastAPI
【发布时间】：2020-06-08 06:18:30
【问题描述】：

将此recarray的内容传递给fastAPI：

import numpy


rec_array =  numpy.recarray(shape = (1, ), 
                            dtype = [('col_a', 'O'), 
                                     ('col_b', '<f8'), 
                                     ('col_c', '<i8')])
rec_array['col_a'][0]  = '0'
rec_array['col_b'][0]  = 1.0
rec_array['col_c'][0]  = 128

此版本有效：

{'col_a':[str(rec_array['col_a'][0])], 
 'col_b':[float(rec_array['col_b'][0])], 
 'col_c':[int(rec_array['col_c'][0])]}

但是这个不是：

{name:[rec_array[name][0]] for name in rec_array.dtype.names}

我想知道为什么。这是我在 windows 下从 fastAPI 获得的错误跟踪：

  File "C:\Users\xor\AppData\Local\Programs\Python\Python38-32\lib\site-packages\fastapi\encoders.py", line 158, in json
able_encoder
    raise ValueError(errors)
ValueError: [TypeError("'numpy.int64' object is not iterable"), TypeError('vars() argument must have __dict__ attribute'
)]

【问题讨论】：

能否请您添加代码的fastapi部分？如何将其传递给 fast-api？

标签： python json numpy fastapi

【解决方案1】：

您面临的问题是由于 int64 (<i8) 类型。在您的第一个代码 sn-p 中，您将其显式转换为常规 int：

'col_c':[int(rec_array['col_c'][0])]
         ===

在第二个中，它仍然是numpy.int64：

d = {name:[rec_array[name][0]] for name in rec_array.dtype.names}
type(d["col_c"][0])
===> numpy.int64

要解决这个问题，您可以执行以下操作：

def make_int(x):
    if isinstance(x, np.int64):
        return int(x)
    return x

{name:[make_int(rec_array[name][0])] for name in rec_array.dtype.names}

这导致您可以发送到快速 API 的管道。

【讨论】：