是否可以重新整理.npz文件中的数组顺序?

时间:2018-10-17 10:17:53

标签: python arrays numpy

我有一个数组数组,可以使用保存到.npz文件中

np.savez('test2.npz', *NumpyArrayOfArrays)

我想知道是否有可能改组保存数组的顺序。

我尝试过

np.random.shuffle(测试)

不起作用

2 个答案:

答案 0 :(得分:1)

您尝试过普通的random.shuffle()吗?

import random
np.savez('test2.npz', *random.shuffle(NumpyArrayOfArrays))

答案 1 :(得分:0)

如果我以(4,3)数组开头:

In [92]: arr
Out[92]: 
array([[ 0,  1,  2],
       [ 3,  4,  5],
       [ 6,  7,  8],
       [ 9, 10, 11]])

并按需保存,*arr将其视为一维数组(行)的列表,并将每个数组保存到npz存档中的单独文件中:

In [93]: np.savez('test.npz', *arr)

load给了我这些数组的字典:

In [94]: d = np.load('test.npz')
In [96]: list(d.keys())
Out[96]: ['arr_0', 'arr_1', 'arr_2', 'arr_3']
In [97]: d['arr_1']
Out[97]: array([3, 4, 5])

或使用操作系统unzip列出存档:

In [101]: !unzip -v test.npz
Archive:  test.npz
 Length   Method    Size  Cmpr    Date    Time   CRC-32   Name
--------  ------  ------- ---- ---------- ----- --------  ----
     152  Stored      152   0% 1980-01-01 00:00 ea1240f7  arr_0.npy
     152  Stored      152   0% 1980-01-01 00:00 c2903dd0  arr_1.npy
     152  Stored      152   0% 1980-01-01 00:00 987e678b  arr_2.npy
     152  Stored      152   0% 1980-01-01 00:00 fcae29f8  arr_3.npy
--------          -------  ---                            -------
     608              608   0%                            4 files

除了名称之外,“保存顺序”并不重要。或者至少我不明白你为什么在乎。

或者,我可以使用字典方法设置存档文件名

In [114]: names = ['w','x','y','z']
In [115]: dd = {n:v for n,v in zip(names,arr)}
In [116]: dd
Out[116]: 
{'w': array([0, 1, 2]),
 'x': array([3, 4, 5]),
 'y': array([6, 7, 8]),
 'z': array([ 9, 10, 11])}
In [117]: np.savez('test.npz', **dd)
In [118]: d = np.load('test.npz')
In [119]: list(d.keys())
Out[119]: ['w', 'x', 'y', 'z']
In [120]: d['x']
Out[120]: array([3, 4, 5])

In [121]: !unzip -v test.npz
Archive:  test.npz
 Length   Method    Size  Cmpr    Date    Time   CRC-32   Name
--------  ------  ------- ---- ---------- ----- --------  ----
     152  Stored      152   0% 1980-01-01 00:00 ea1240f7  w.npy
     152  Stored      152   0% 1980-01-01 00:00 c2903dd0  x.npy
     152  Stored      152   0% 1980-01-01 00:00 987e678b  y.npy
     152  Stored      152   0% 1980-01-01 00:00 fcae29f8  z.npy
--------          -------  ---                            -------
     608              608   0%                            4 files

我可以使用类似字典的方法从存档中获取值:

In [122]: list(d.items())
Out[122]: 
[('w', array([0, 1, 2])),
 ('x', array([3, 4, 5])),
 ('y', array([6, 7, 8])),
 ('z', array([ 9, 10, 11]))]
In [123]: list(d.values())
Out[123]: [array([0, 1, 2]), array([3, 4, 5]), array([6, 7, 8]), array([ 9, 10, 11])]
In [124]: np.stack(_)
Out[124]: 
array([[ 0,  1,  2],
       [ 3,  4,  5],
       [ 6,  7,  8],
       [ 9, 10, 11]])