尝试在 h5py 中打开 pandas 创建的 hdf 时缺少列

2024-03-05

这就是我的数据框的样子。第一列是一个整数。第二列是 512 个整数的单个列表。

IndexID Ids
1899317 [0, 47715, 1757, 9, 38994, 230, 12, 241, 12228...
22861131    [0, 48156, 154, 6304, 43611, 11, 9496, 8982, 1...
2163410 [0, 26039, 41156, 227, 860, 3320, 6673, 260, 1...
15760716    [0, 40883, 4086, 11, 5, 18559, 1923, 1494, 4, ...
12244098    [0, 45651, 4128, 227, 5, 10397, 995, 731, 9, 3...

我将其保存到 hdf 并尝试使用打开它

df.to_hdf('test.h5', key='df', data_columns=True)
h3 = h5py.File('test.h5')

当我列出键时,我看到 4 个键

h3['df'].keys()

KeysViewHDF5 ['axis0', 'axis1', 'block0_items', 'block0_values']

Axis1 看到包含第一列的值

h3['df']['axis1'][0:5]

数组([ 1899317, 22861131, 2163410, 15760716, 12244098,

但是,第二列似乎没有数据。确实还有另一列包含其他数据

h3['df']['block0_values'][0][0:5]

但这似乎与第二列中的任何数据都不对应

数组([128, 4, 149, 1, 0], dtype=uint8)

Purpose

我最终尝试创建一个内存映射的数据存储,它使用特定索引检索数据。

所以像

h3['df']['workingIndex'][22861131, 15760716] 

会检索

[0, 48156, 154, 6304, 43611, 11, 9496, 8982, 1...],
[0, 40883, 4086, 11, 5, 18559, 1923, 1494, 4, ...

问题是你正在尝试序列化 Pandas 系列的 Python 列表,但它不是矩形的(它是锯齿状的)。

Pandas 和 HDF5 主要用于矩形(立方体、超立方体等)数据,而不是锯齿状的列表列表。

您致电时是否看到此警告to_hdf()?

PerformanceWarning: 
your performance may suffer as PyTables will pickle object types that it cannot
map directly to c-types [inferred_type->mixed,key->block0_values] [items->['Ids']]

它试图告诉您的是列表列表不以直观、高性能的方式支持。如果您运行 HDF5 可视化工具,例如h5dump在你的输出文件中,你会看到哪里出了问题。索引(表现良好)如下所示:

  DATASET "axis1" {
     DATATYPE  H5T_STD_I64LE
     DATASPACE  SIMPLE { ( 5 ) / ( 5 ) }
     DATA {
     (0): 1899317, 22861131, 2163410, 15760716, 12244098
     }
     ATTRIBUTE "CLASS" {
        DATA {
        (0): "ARRAY"
        }
     }

但值(列表的列表)看起来像这样:

  DATASET "block0_values" {
     DATATYPE  H5T_VLEN { H5T_STD_U8LE}
     DATASPACE  SIMPLE { ( 1 ) / ( H5S_UNLIMITED ) }
     DATA {
     (0): (128, 5, 149, 164, ...)
     }
     ATTRIBUTE "CLASS" {
        DATA {
        (0): "VLARRAY"
        }
     }
     ATTRIBUTE "PSEUDOATOM" {
        DATA {
        (0): "object"
        }
     }

所发生的情况正是 PerformanceWarning 向您发出的警告:

> PyTables will pickle object types that it cannot map directly to c-types

您的列表列表正在被腌制并存储为 H5T_VLEN,这只是一个字节块。

您可以通过以下一些方法来解决此问题:

  1. 将每一行存储在 HDF5 中的单独键下。也就是说,每个列表都将存储为一个数组,并且它们都可以具有不同的长度。这对于 HDF5 来说没有问题,因为它支持一个文件中包含任意数量的密钥。
  2. 将您的数据更改为矩形,例如通过用零填充较短的列表。看:Pandas 将列表的列拆分为多列 https://stackoverflow.com/questions/35491274/pandas-split-column-of-lists-into-multiple-columns
  3. 使用 h5py 以您喜欢的任何格式写入数据。它比 Pandas/PyTables 更灵活,并且创建更简单(但更强大)的 HDF5 文件。这是一个例子(它表明 h5py 实际上可以存储锯齿状数组,尽管它并不漂亮):使用h5py存储多维可变长度数组 https://stackoverflow.com/questions/42658438/storing-multidimensional-variable-length-array-with-h5py
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

尝试在 h5py 中打开 pandas 创建的 hdf 时缺少列 的相关文章

随机推荐