[已解决]新人求助关于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

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

复制代码

跳转到最佳答案楼层

逃兵

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

1. b = [2001,2002,2003]
   
2. print(b[0])
   
3. print(b[1])
   
4. print(b[2])
   
5. print(b[0:2])
   
6. print(b[0:3])
   
7. print(b[1:3])
   

复制代码

hrpzcf

1. 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的列表推导式。

1. >>> a
   
2. [1, 2, 3]
   
3. >>> b
   
4. [2001, 2002, 2003]
   
5. >>> [b[a[i] - 1] for i in range(3)]
   
6. [2001, 2002, 2003]
   
7. >>>

复制代码

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

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

复制代码

hrpzcf

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

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