gpt4 book ai didi

python - 从 HDF5 获取表索引的最有效方法

转载 作者:太空宇宙 更新时间:2023-11-04 05:44:40 25 4
gpt4 key购买 nike

我有一个包含 pandas Series/DataFrame 表的 HDF5 文件。我需要获取存储在 HDF 键下的表的(pandas)索引,但不一定是整个表:

我可以想到两种(实际上相同的)获取索引的方法:

import pandas as pd

hdfPath = 'c:/example.h5'
hdfKey = 'dfkey'
# way 1:
with pd.HDFStore(hdfPath) as hdf:
index = hdf[hdfKey].index

# way 2:
index = pd.read_hdf(hdfPath, hdfKey)

然而,对于大约 2000 行的 Pandas 系列,这需要 0.6 秒:

%timeit pd.read_hdf(hdfPath, hdfKey).index
1 loops, best of 3: 605 ms per loop

有没有办法只获取 HDF 中表的索引?

最佳答案

HDFStore 对象有一个select_column 方法,可以让您获取索引。请注意,它将返回一个以索引作为值的系列。

with pd.HDFStore(hdfPath) as hdf:
index = hdf.select_column(hdfKey, 'index').values

关于python - 从 HDF5 获取表索引的最有效方法,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/32740083/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com