Python中如何按指定z值截取配对数组内对应z、T数据？

方案1：纯标准库实现（无第三方依赖） #

适合小数据量场景，不需要额外安装依赖，同时兼容浮点数精度误差问题：

import math
from typing import List, Tuple

# 初始化原始配对数据
z_seq: List[float] = [0, 0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9, 1.0]
T_seq: List[int] = [0, 100, 200, 300, 400, 500, 600, 700, 800, 900, 1000]
data_pairs: List[Tuple[float, int]] = list(zip(z_seq, T_seq))

def get_data_up_to_z(target_z: float) -> List[Tuple[float, int]]:
    """输入指定z值，返回从序列开头到该z值位置的所有配对数据"""
    for idx, z in enumerate(z_seq):
        if math.isclose(z, target_z):
            return data_pairs[:idx+1]
    raise ValueError(f"输入的z值{target_z}不在给定z序列范围内")

def get_data_between_z(z_start: float, z_end: float) -> List[Tuple[float, int]]:
    """输入两个z值作为区间端点，返回区间内所有配对数据（包含两端点）"""
    if z_start > z_end:
        raise ValueError("区间起始值不能大于终止值")
    # 查找两个端点的索引
    start_idx = end_idx = None
    for idx, z in enumerate(z_seq):
        if math.isclose(z, z_start):
            start_idx = idx
        if math.isclose(z, z_end):
            end_idx = idx
    if start_idx is None or end_idx is None:
        raise ValueError("输入的区间端点不在给定z序列范围内")
    return data_pairs[start_idx:end_idx+1]

使用示例 #

• 单z值查询：执行get_data_up_to_z(0.6)，返回结果为[(0, 0), (0.1, 100), (0.2, 200), (0.3, 300), (0.4, 400), (0.5, 500), (0.6, 600)]
• 区间查询：执行get_data_between_z(0.1, 0.7)，返回结果为[(0.1, 100), (0.2, 200), (0.3, 300), (0.4, 400), (0.5, 500), (0.6, 600), (0.7, 700)]

方案2：基于Pandas实现（适合大数据量/后续数据分析场景） #

如果需要处理更大的数据集，或者后续要做统计、导出等操作，用Pandas实现更简洁高效：

import math
import pandas as pd

# 构造结构化数据表
df = pd.DataFrame({
    "z": [0, 0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9, 1.0],
    "T": [0, 100, 200, 300, 400, 500, 600, 700, 800, 900, 1000]
})

def get_df_up_to_z(target_z: float) -> pd.DataFrame:
    target_idx = df[df["z"].apply(lambda x: math.isclose(x, target_z))].index[0]
    return df.iloc[:target_idx+1]

def get_df_between_z(z_start: float, z_end: float) -> pd.DataFrame:
    if z_start > z_end:
        raise ValueError("区间起始值不能大于终止值")
    return df[(df["z"] >= z_start - 1e-9) & (df["z"] <= z_end + 1e-9)]

优势 #

• 原生支持表格导出、数值计算等扩展需求
• 可轻松扩展支持模糊匹配，比如输入不存在的z值时自动返回最近值对应的区间数据

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