[已解决]新人求助关于list索引问题

wlp1818100227

刚刚从Matlab转Python的萌新，问题如下
a = [1,2,3];
b = [2001,2002,2003];
想通过a把b的值索引出来导入新矩阵中
MATLAB代码如下
i = 1:3
n = b(a(i))
结果
n = [2001,2002,2003]

想用python除了for 循环还有其他办法吗

最佳答案

月排行榜 / 总排行榜

hrpzcf

2021-10-22 09:18:22

n = [b[i - 1] for i in a]

跳转到最佳答案楼层

逃兵

切片就可以了
从0开始，区间前闭后开

b = [2001,2002,2003]
print(b[0])
print(b[1])
print(b[2])
print(b[0:2])
print(b[0:3])
print(b[1:3])

hrpzcf

n = [b[i - 1] for i in a]

wlp1818100227

逃兵 发表于 2021-10-22 09:18
切片就可以了
从0开始，区间前闭后开

谢谢啦，看来还是python可能还是for循环简单一点

wlp1818100227

hrpzcf 发表于 2021-10-22 09:18

用for 好简洁，用MATLAB一直不太喜欢用for

suchocolate

如果按index逐个取的话，python确实需要for循环，或者借助for的列表推导式。

>>> a
[1, 2, 3]
>>> b
[2001, 2002, 2003]
>>> [b[a[i] - 1] for i in range(3)]
[2001, 2002, 2003]
>>> 

单如果是连续取的话，用切片是比较方便的：

>>> b = [1,2,3,4,5]
>>> b[1:4]
[2, 3, 4]
>>> 

hrpzcf

wlp1818100227 发表于 2021-10-22 09:25
用for 好简洁，用MATLAB一直不太喜欢用for

在python中一般不用把索引和值分开储存这种写法
上一级处理出来的索引值直接用于列表取值就可以了，甚至可以把处理索引和取值写在一个语句内，索引值单独储存相当于多了个中间环节