python基础：list索引

原创于 2019-03-28 10:33:03 发布 · 285 阅读

1 ·

CC 4.0 BY-SA版权

python基础专栏收录该内容

13 篇文章

订阅专栏

本文介绍了使用Python的NumPy库高效获取数组中对角线元素的方法。通过使用numpy.diag函数可以直接提取ndarray的对角线元素，避免了复杂的循环操作，提高了代码的简洁性和执行效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

当时想对numpy的ndarray取任意一条斜线上的元素，想来想去只能写循环，后来发现可以先flatten（），再用定长步长进行取出操作。不过后来发现还可以直接用numpy.diag(arr)直接获得对角线元素。

>>> a = list(range(10))
>>> a[1:3]
[1, 2]
>>> a[1::3]
[1, 4, 7]
>>> a[0::4]
[0, 4, 8]
>>> a[1:7:2]
[1, 3, 5]
>>> [1]*10
[1, 1, 1, 1, 1, 1, 1, 1, 1, 1]
>>> "a"*10
'aaaaaaaaaa'