python - NumPy 中的数据类型

Question

我在 numpy 中遇到过以下语句:

x=numpy.zeros((2,2),dtype=[('x','i4'),('y','i4')])

输出是这样的:

[[(0,0)(0,0)]    
 [(0,0)(0,0)]]

[('x','i4'),('y','i4')] 是什么意思？请解释。

Answer 1

这就是数组元素被赋予名称和数据类型的方式。

在这种情况下，可以使用 'x' 访问数组中每个条目的第一个元素的名称，可以使用 'y' 访问第二个元素的名称>:

>>> x['x']
array([[0, 0],
       [0, 0]])
>>> x['y']
array([[0, 0],
       [0, 0]])

如果我们更改其中一个条目，这会更清楚:

>>> x['x'] = numpy.array([[1,1],[1,1]])
>>> x
array([[(1, 0), (1, 0)],
       [(1, 0), (1, 0)]], dtype=[('x', 'i4'), ('y', 'i4')])

如您所见，每个条目中的第一个元素都已更改。

'i4' 部分指定元素的数据类型。具体来说:

i 表示有符号整数

4 表示 4 字节大小

参见文档 here