我有一个二维列表，喜欢从列表中获取单个列。
X=[]
for line in lines:
c,xmin,ymin,xmax,ymax = line.split(' ')
xmin_ = float(xmin)
ymin_ = float(ymin)
xmax_ = float(xmax)
ymax_ = float(ymax)
area_ = (xmax_ - xmin_)*(ymax_ - ymin_)
X.append(xmax_-xmin_, ymax_-ymin_, area_)
如果我打印列值，则会出现错误
>>>X[:,1]
*** TypeError: list indices must be integers, not tuple
但是为什么下面的例子有效
>>> a = arange(20).reshape(4,5)
>>> a
array([[ 0, 1, 2, 3, 4],
[ 5, 6, 7, 8, 9],
[10, 11, 12, 13, 14],
[15, 16, 17, 18, 19]])
>>> a[:, 1]
array([ 1, 6, 11, 16])
有什么区别？
您可以将列表更改为 numpy 数组。
import numpy as np
x=[[1,2,3],[4,5,6]]
x[:,1]
Traceback (most recent call last):
File "<input>", line 1, in <module>
TypeError: list indices must be integers, not tuple
xa = np.array(x)
xa
array([[1, 2, 3],
[4, 5, 6]])
xa[:,1]
array([2, 5])
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句