python - 列表正确访问部分元素的列表
问题描述
我正在尝试访问 python 3 中的列表列表。
a = [
[ 1, 2, 3 ],
[ 4, 5, 6 ],
[ 7, 8, 9 ]
]
我需要访问最右边一列中的第二个和第三个元素。我想得到:
[6, 9]
作为一个列表
但,
from numpy import matrix
from numpy import array
a = [
[ 1, 2, 3 ],
[ 4, 5, 6 ],
[ 7, 8, 9 ]
]
ab = matrix(a)
ab[1:3][2]
ab = array(a)
ab[1:3][2]
它们都不起作用。
如何正确访问部分元素?
解决方案
取从行second
开始的first
列:
In [27]: x = np.arange(1, 10).reshape(3,3)
# array([[1, 2, 3],
# [4, 5, 6],
# [7, 8, 9]])
In [31]: x[1:,2]
Out[31]: array([6, 9])