Question

创建h5py dataset后，如何在NxM阵列中添加或删除特定的行或列？

我的问题类似于this one，但我不想盲目截断或扩展数组。删除时，我需要能够指定要删除的确切行或列。

对于添加，我知道在创建初始数据集时我必须指定maxshape=(None, None)，但resize方法似乎不允许您指定在收缩时哪些行或列被截断大小。

Answer 1

h5py实际上并不是为此而设计的。 Pandas可能是一个更好的库，因为它是围绕表格的概念构建的。

话虽如此，这是怎么做的：

In [1]: f = h5py.File('test.h5')

In [2]: arr = rand(4,4)

In [3]: dset = f.create_dataset('foo',data=arr,maxshape=(2000,2000))

In [4]: dset[:]
Out[4]:
array([[ 0.29732874,  0.59310285,  0.61116263,  0.79950116],
       [ 0.4194363 ,  0.4691813 ,  0.95648712,  0.56120731],
       [ 0.76868585,  0.07556214,  0.39854704,  0.73415885],
       [ 0.0919063 ,  0.0420656 ,  0.35082375,  0.62565894]])

In [5]: dset[1:-1,:] = dset[2:,:]

In [6]: dset.resize((3,4))

In [7]: dset[:]
Out[7]:
array([[ 0.29732874,  0.59310285,  0.61116263,  0.79950116],
       [ 0.76868585,  0.07556214,  0.39854704,  0.73415885],
       [ 0.0919063 ,  0.0420656 ,  0.35082375,  0.62565894]])

这会从dset中删除第1列。它通过在将数据集缩小一列之前分别将第2列和第3列分配为1和2来实现。交换下标以删除第1行。如果您将要执行此操作，可以轻松地编写一个包装器。

在h5py数据集中添加或删除特定行或列

1 个答案: