Matlab转Python：带条件判断的for循环迭代问题求助

首先明确原Matlab代码的逻辑：保留数组第一个元素，遍历相邻元素对，当后一个元素与前一个的差值超过阈值时保留后一个元素，反之则删除后一个元素，最终得到的结果是[10, 32, 60, 90]。

Python与Matlab的核心差异是索引从0开始，但访问i+1位置元素的逻辑是相通的，以下是两种适合新手的实现方式：

方式1：纯Python列表实现（无需额外库） #

直接通过循环遍历相邻元素，按规则构建新列表：

old_list = [10, 32, 35, 60, 90, 100]
threshold = 20

# 先保留第一个元素
new_list = [old_list[0]]
# 遍历从第0个到倒数第二个元素（对应Matlab的i=1到length(old_list)-1）
for i in range(len(old_list) - 1):
    current_val = old_list[i]
    next_val = old_list[i + 1]
    # 差值超过阈值时，保留后一个元素
    if next_val - current_val > threshold:
        new_list.append(next_val)

print(new_list)  # 输出: [10, 32, 60, 90]

方式2：用Numpy实现（更接近Matlab数组操作风格） #

如果习惯Matlab的数组运算逻辑，可以用Numpy简化代码：

import numpy as np

old_array = np.array([10, 32, 35, 60, 90, 100])
threshold = 20

# 计算相邻元素的差值
diff = np.diff(old_array)
# 找出差值超过阈值的位置，对应的后一个元素索引是这些位置+1，再加上第一个元素的索引0
keep_indices = np.concatenate([[0], np.where(diff > threshold)[0] + 1])
# 按索引筛选得到新数组
new_array = old_array[keep_indices]

print(new_array)  # 输出: [10 32 60 90]

关键说明 #

• Python列表/数组的索引从0开始，原Matlab中i=1对应Python的i=0，遍历范围range(len(old_list)-1)正好覆盖所有相邻元素对。
• 两种方式都实现了与原Matlab代码完全一致的逻辑，无需像Matlab那样先收集要删除的索引再删除，直接构建新列表更符合Python的惯用写法。

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