如何在Python中实现Matlab式的向量索引与元素替换？

当然可以实现！不过要注意Python原生列表和Matlab数组在索引规则上有两个关键差异：一是Python用0-based索引（第一个元素索引为0），而Matlab是1-based；二是原生列表不支持直接用整数列表做批量索引赋值，但我们有几种方法可以达到同样的效果。

方法1：使用循环（适合普通列表） #

这是最直观的方式，直接遍历索引列表逐个修改：

# 注意：如果要和你给出的Matlab示例结果一致，这里的a要调整为[0,1,3]（对应Matlab的1,2,4）
a = [0, 1, 3]
b = list(range(1, 5))  # 生成[1,2,3,4]，和Matlab里的b一致
for idx in a:
    b[idx] = 1
print(b)  # 输出: [1, 1, 3, 1]，和Matlab执行结果相同

如果你坚持用原问题里的a=[1,2,4]，那会修改b的第2、3、5个元素（因为Python索引从0开始），得到[1,1,1,4,1,6,7,8,9]（当b是list(range(1,10))时）。

方法2：使用NumPy数组（最接近Matlab风格） #

如果你经常处理数值型数组，推荐用NumPy——它的语法和Matlab非常相似，支持向量化的索引赋值：

import numpy as np

# 同样，注意索引的转换：Matlab的[1,2,4]对应NumPy的[0,1,3]
a = np.array([0, 1, 3])
b = np.array([1,2,3,4])
b[a] = 1
print(b)  # 输出: [1 1 3 1]，完美匹配Matlab的结果

NumPy的数组天然支持整数数组索引，这就和Matlab里的操作逻辑完全一致了，而且效率更高，适合处理大规模数据。

为什么原生列表不能直接这么做？ #

Python原生列表的索引规则只允许：

• 单个整数（比如b[0]）
• 切片（比如b[1:3]）
• 布尔型列表（长度和原列表一致，比如[True, False, True]）

直接用整数列表a去索引b[a]会抛出TypeError，这就是你之前尝试替换失败的原因。

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