如何用Numpy按指定步长扩充数组Z列，插入多行中间行

核心逻辑 #

先锁定原始数组中Z值的范围，生成包含所有中间步长的Z序列，再将其他列的固定值重复对应次数，最后拼接成完整的扩充数组。

代码实现 #

假设原始数组为arr，步长step=1，具体操作如下：

1. 初始化数据与依赖 #

import numpy as np

# 示例原始数组
arr = np.array([
    [26.4812, 32.0000, -5.0000, 10000.0000, 20000.0000, 2.0000],
    [26.4812, 32.0000, 10.0000, 10000.0000, 20000.0000, 2.0000],
    [26.4812, 32.0000, 0.0000, 10000.0000, 20000.0000, 2.0000]
])

step = 1  # 指定步长

2. 生成完整Z值序列 #

提取Z列的最小、最大值，生成包含所有中间值的序列：

z_min = arr[:, 2].min()
z_max = arr[:, 2].max()
z_sequence = np.arange(z_min, z_max + step, step)

3. 重复非Z列固定值 #

取任意一行的非Z列数据（因示例中其他列数值统一），重复对应次数：

# 提取第一行的非Z列数据
non_z_data = arr[0, [0, 1, 3, 4, 5]]
# 重复len(z_sequence)次，转为二维数组格式
non_z_repeated = np.tile(non_z_data, (len(z_sequence), 1))

4. 拼接成最终扩充数组 #

将重复后的非Z列与Z序列按原结构拼接：

# 将Z序列转为二维列数组
z_col = z_sequence.reshape(-1, 1)
# 按原列顺序拼接
expanded_arr = np.hstack([non_z_repeated[:, :2], z_col, non_z_repeated[:, 2:]])

特殊场景处理 #

• 若原始数组中其他列存在不同数值：需先按非Z列的相同值分组，再对每组单独执行上述流程。
• 若步长为非整数：建议使用np.linspace替代np.arange，通过设置endpoint=True确保包含最大值端点，避免精度误差。

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