使用Json转储2D Python数组

2023年1月27日 198次阅读

我有一个numpy数组,我想与Json一起转储.该数组如下所示：

array([['foo', 'bar', 'something', ...
        'more'],
        ['0.4', '0.7', '0.83', ...
        '0.3', '0.62', '0.51']]

我想将它转储到Json的字符串上,如下所示：

foo: 0.4
bar: 0.7
something: 0.51
...

我尝试过：

import jason
my_string = json.dumps(my_array)

但它抱怨：

"not JSON serializable"

有关如何使用Json将其转储到字符串上的任何想法？

更新：

请注意,我不在乎订购,应按以下顺序打印行：

array[0,0] : array[0,1]
array[1,0] : array[1,1]
array[2,0] : array[2,1]
# etc ...

最佳答案 什么对我有用 – 因为有更大的1024×1002 float64阵列 – 转换为base64.

def Base64Encode(ndarray):
    return json.dumps([str(ndarray.dtype),base64.b64encode(ndarray),ndarray.shape])
def Base64Decode(jsonDump):
    loaded = json.loads(jsonDump)
    dtype = np.dtype(loaded[0])
    arr = np.frombuffer(base64.decodestring(loaded[1]),dtype)
    if len(loaded) > 2:
        return arr.reshape(loaded[2])
    return arr

''' just to compare '''
def SimpleEncode(ndarray):
    return json.dumps(ndarray.tolist())
def SimpleDecode(jsonDump):
    return np.array(json.loads(jsonDump))

ipython％timeit结果非常清楚地指向base64：

arr = np.random.random_sample((1000, 1000))

print 'Simple Convert'
%timeit SimpleDecode(SimpleEncode(arr))
print 'Base64 Encoding'
%timeit Base64Decode(Base64Encode(arr))

结果：

Simple Convert
1 loops, best of 3: 1.42 s per loop
Base64 Encoding
10 loops, best of 3: 171 ms per loop
点赞