HDF5 数据集名称与组名称冲突问题及解决方案

import h5py

def group_path_ok(file, dset_tag):
    """
    检查给定的路径中是否存在数据集。

    参数:
        file: h5py.File 对象，代表 HDF5 文件。
        dset_tag: 字符串，代表要检查的路径。

    返回:
        布尔值，如果路径中不存在数据集，则返回 True；否则返回 False。
    """
    pset_path = dset_tag.split('/')
    group_path = ''
    for name in pset_path[:-1]:
        group_path += '/' + name if group_path else name
        if group_path in file and isinstance(file[group_path], h5py.Dataset):
            print(f'group name: {group_path} in path is a dataset')
            return False
    return True

import h5py

def create_or_update_dataset(filename, h5_path, data):
    """
    创建或更新 HDF5 文件中的数据集。

    参数:
        filename: 字符串，代表 HDF5 文件名。
        h5_path: 字符串，代表数据集的路径。
        data: 要写入数据集的数据。
    """
    with h5py.File(filename, "a") as file: # 使用 "a" 模式，如果文件不存在则创建
        if group_path_ok(file, h5_path):
            if h5_path in file:
                # 如果数据集已存在，则先删除
                del file[h5_path]
                print("Dataset deleted")

            # 确保父组存在
            path_parts = h5_path.split('/')[:-1]
            current_path = ''
            for part in path_parts:
                current_path += '/' + part if current_path else part
                if current_path not in file:
                    file.create_group(current_path)

            # 创建数据集
            file.create_dataset(h5_path, data=data)
            print(f"Dataset '{h5_path}' created successfully.")
        else:
            print(f"Error: Cannot create dataset '{h5_path}' due to path conflict.")

# 示例代码
filename = "example.h5"
data_set = [1, 2, 3, 4]

# 创建数据集 "my_path"
h5_path1 = "my_path"
create_or_update_dataset(filename, h5_path1, data_set)

# 尝试创建数据集 "my_path/to_another/dest"
h5_path2 = "my_path/to_another/dest"
create_or_update_dataset(filename, h5_path2, data_set)

# 再次创建数据集 "my_path/to_another/dest"，会先删除再创建
h5_path2 = "my_path/to_another/dest"
create_or_update_dataset(filename, h5_path2, data_set)