PyTables 3中如何基于字符串列条件查询HDF5表数据

Got it, let's work through this query logic step by step. You need to dynamically match either column B or C based on a flag, find rows where the target value exists, then pull the corresponding A column values. Here's a complete, working implementation:

Full Code Implementation #

import tables as tb

# Your existing table creation code
h5file = tb.open_file("ex.h5", "w")
class ex(tb.IsDescription):
    A = tb.StringCol(5, pos=0)
    B = tb.StringCol(5, pos=1)
    C = tb.StringCol(5, pos=2)
table = h5file.create_table('/', 'table', ex, "Passing string as column name")
rows = [ ('abc', 'bcd', 'dse'), ('der', 'fre', 'swr'), ('xsd', 'weq', 'rty') ]
table.append(rows)
table.flush()

# --- Query Logic ---
target_value = 'swr'
condition_flag = 'B'  # Change this to 'C' to test matching column C

# Dynamically select the column to match
match_col = 'B' if condition_flag == 'B' else 'C'

# Build the query condition (note string values need single quotes in PyTables)
query = f"{match_col} == '{target_value}'"

# Execute query and extract A column values
matching_A = [row['A'] for row in table.where(query)]

# Output results
print(f"Matched A values: {matching_A}")

# Clean up: close the HDF5 file
h5file.close()

Key Details Explained #

• Dynamic Column Selection: We use a simple conditional to pick either B or C based on your condition_flag—easy to adjust if you need to add more columns later.
• Query Syntax: PyTables requires string literals to be wrapped in single quotes within the query string, so we use an f-string to safely construct this condition.
• Result Extraction: We use a list comprehension to iterate over matching rows and pull just the A column value, keeping the code concise.

Test Cases #

• If condition_flag = 'B': No rows match (since B column has 'bcd', 'fre', 'weq'), so matching_A will be an empty list.
• If condition_flag = 'C': The second row matches (C column has 'swr'), so matching_A will return ['der'].

内容的提问来源于stack exchange，提问作者ymb